From 6ced748cb1d4729688423d06f21da015b43aeba4 Mon Sep 17 00:00:00 2001
From: philippeitis <33013301+philippeitis@users.noreply.github.com>
Date: Fri, 7 Oct 2022 20:38:41 -0700
Subject: [PATCH] Fix tested type

---
 src/generator/lib/__tests__/util_test.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/generator/lib/__tests__/util_test.py b/src/generator/lib/__tests__/util_test.py
index 9e1393aefe..eb34be7e1a 100644
--- a/src/generator/lib/__tests__/util_test.py
+++ b/src/generator/lib/__tests__/util_test.py
@@ -82,7 +82,7 @@ class UtilsTest(unittest.TestCase):
         test_properties = (
             ('Album', 'title', 'String'), # string
             ('Status', 'code', 'i32'), # numeric
-            ('Album', 'mediaItemsCount', 'String'), # numeric via "count" keyword
+            ('Album', 'mediaItemsCount', 'i64'), # numeric via "count" keyword
             ('Album', 'isWriteable', 'bool'), # boolean
             ('Album', 'shareInfo', 'ShareInfo'), # reference type
             ('SearchMediaItemsResponse', 'mediaItems', 'Vec<MediaItem>'), # array
@@ -90,7 +90,7 @@ class UtilsTest(unittest.TestCase):
         for (class_name, property_name, expected) in test_properties:
             property_value = schemas[class_name]['properties'][property_name]
             rust_type = to_rust_type(schemas, class_name, property_name, property_value, allow_optionals=False)
-            self.assertEqual(rust_type, expected)
+            self.assertEqual(rust_type, expected, f"Parsed class: {class_name}, property: {property_name}")
 
         # items reference
         class_name = 'SearchMediaItemsResponse'