diff options
Diffstat (limited to 'ishtar_common/tests.py')
| -rw-r--r-- | ishtar_common/tests.py | 17 |
1 files changed, 13 insertions, 4 deletions
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): |