Changeset View
Changeset View
Standalone View
Standalone View
swh/storage/tests/test_storage.py
Show First 20 Lines • Show All 3,758 Lines • ▼ Show 20 Lines | class CommonPropTestStorage: | ||||
def test_add_arbitrary(self, objects): | def test_add_arbitrary(self, objects): | ||||
self.reset_storage() | self.reset_storage() | ||||
for (obj_type, obj) in objects: | for (obj_type, obj) in objects: | ||||
obj = obj.to_dict() | obj = obj.to_dict() | ||||
if obj_type == 'origin_visit': | if obj_type == 'origin_visit': | ||||
origin_id = self.storage.origin_add_one(obj.pop('origin')) | origin_id = self.storage.origin_add_one(obj.pop('origin')) | ||||
if 'visit' in obj: | if 'visit' in obj: | ||||
del obj['visit'] | del obj['visit'] | ||||
self.storage.origin_visit_add(origin_id, **obj) | self.storage.origin_visit_add( | ||||
origin_id, obj['date'], obj['type']) | |||||
else: | else: | ||||
method = getattr(self.storage, obj_type + '_add') | method = getattr(self.storage, obj_type + '_add') | ||||
try: | try: | ||||
method([obj]) | method([obj]) | ||||
except HashCollision: | except HashCollision: | ||||
pass | pass | ||||
▲ Show 20 Lines • Show All 121 Lines • Show Last 20 Lines |