Skip to content

Commit ccfe062

Browse files
committed
add test for dblog storage
1 parent 66808e3 commit ccfe062

File tree

1 file changed

+16
-1
lines changed

1 file changed

+16
-1
lines changed

tests/test_dblog.py

Lines changed: 16 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -68,7 +68,22 @@ def test_log_request(self):
6868
assert running == 1
6969
assert stored == 0
7070

71-
71+
def test_storage(self):
72+
fake_storage = ns(
73+
uuid="ebf3cd00-0102-11ed-8421-e4b97ac7e02e",
74+
pretty_filename = "pretty_filename.txt",
75+
mimetype="text/plain",
76+
dump=lambda: b'somedata'
77+
)
78+
79+
dblog.update_storage_record(fake_storage)
80+
81+
s = dblog.get_storage_record(fake_storage.uuid)
82+
83+
assert s.uuid == fake_storage.uuid
84+
assert s.pretty_filename == fake_storage.pretty_filename
85+
assert s.mimetype == fake_storage.mimetype
86+
assert s.data == fake_storage.dump()
7287

7388

7489
def load_tests(loader=None, tests=None, pattern=None):

0 commit comments

Comments
 (0)