From f2cb0542faac028dc3f90a61b46c0949bb35339f Mon Sep 17 00:00:00 2001 From: Étienne Loks Date: Wed, 4 Sep 2019 12:03:28 +0200 Subject: Serializer: manage queryset filter --- ishtar_common/tests.py | 17 +++++++++++++---- 1 file changed, 13 insertions(+), 4 deletions(-) (limited to 'ishtar_common/tests.py') diff --git a/ishtar_common/tests.py b/ishtar_common/tests.py index 7bb3ce098..68c206865 100644 --- a/ishtar_common/tests.py +++ b/ishtar_common/tests.py @@ -599,8 +599,12 @@ class CacheTest(TestCase): class GenericSerializationTest: - def generic_serialization_test(self, serialize): - json_result = serialize() + def generic_serialization_test(self, serialize, no_test=False, kwargs=None): + if not kwargs: + kwargs = {} + json_result = serialize(**kwargs) + if no_test: + return json_result for key in json_result.keys(): __, k = key module_name, model_name = k.split("__") @@ -615,12 +619,17 @@ class GenericSerializationTest: msg="Serialization for model {}.{} failed. {} serialized {} " "expected".format(module.__name__, model_name, serialization_count, current_count)) + return json_result - def generic_restore_test_genzip(self, model_list, serialization): + def generic_restore_test_genzip(self, model_list, serialization, + kwargs=None): current_number = {} for model in model_list: current_number[model.__name__] = model.objects.count() - zip_filename = serialization(archive=True) + if not kwargs: + kwargs = {} + kwargs["archive"] = True + zip_filename = serialization(**kwargs) return current_number, zip_filename def generic_restore_test(self, zip_filename, current_number, model_list): -- cgit v1.2.3