init

1 files changed, 115 insertions, 0 deletions

diff --git a/test_pytest/test_backend.py b/test_pytest/test_backend.py
new file mode 100644
index 0000000..539f5ce
--- /dev/null
+++ b/test_pytest/test_backend.py
@@ -0,0 +1,115 @@
+import pytest
+
+import restlog.backend
+
+
+pytestmark = pytest.mark.asyncio
+
+
+@pytest.fixture
+def db_path(tmp_path):
+    return tmp_path / 'restlog.db'
+
+
+async def test_create(db_path):
+    assert not db_path.exists()
+
+    backend = await restlog.backend.create(db_path, 100)
+    assert db_path.exists()
+
+    await backend.async_close()
+    assert db_path.exists()
+
+
+async def test_register(db_path):
+    backend = await restlog.backend.create(db_path, 100)
+
+    result = await backend.get_entries()
+    assert result == {'entries': [],
+                      'more': False}
+
+    entry = await backend.register(timestamp=123,
+                                   address='address',
+                                   source='source',
+                                   type='type',
+                                   data={'abc': 123})
+    assert entry['entry_id'] is not None
+    assert entry['timestamp'] == 123
+    assert entry['address'] == 'address'
+    assert entry['source'] == 'source'
+    assert entry['type'] == 'type'
+    assert entry['data'] == {'abc': 123}
+
+    result = await backend.get_entry(entry['entry_id'])
+    assert entry == result
+
+    result = await backend.get_entries()
+    assert result == {'entries': [entry],
+                      'more': False}
+
+    await backend.async_close()
+
+    backend = await restlog.backend.create(db_path, 100)
+    result = await backend.get_entry(entry['entry_id'])
+    assert entry == result
+    await backend.async_close()
+
+
+@pytest.mark.parametrize('entries_count', [1, 2, 10])
+async def test_get_entry(db_path, entries_count):
+    backend = await restlog.backend.create(db_path, 100)
+
+    result = await backend.get_entry(123)
+    assert result is None
+
+    entry_ids = []
+    for _ in range(entries_count):
+        entry = await backend.register(timestamp=123,
+                                       address='address',
+                                       source='source',
+                                       type='type',
+                                       data={'abc': 123})
+        entry_ids.append(entry['entry_id'])
+
+    for entry_id in entry_ids:
+        result = await backend.get_entry(entry_id)
+        assert result['entry_id'] == entry_id
+
+    await backend.async_close()
+
+
+@pytest.mark.parametrize('entries_count', [1, 2, 10])
+async def test_get_entries(db_path, entries_count):
+    backend = await restlog.backend.create(db_path, 100)
+
+    result = await backend.get_entry(123)
+    assert result is None
+
+    entry_ids = []
+    for i in range(entries_count):
+        entry = await backend.register(timestamp=123,
+                                       address='address',
+                                       source=f'source{i}',
+                                       type=f'type{i}',
+                                       data={'abc': 123})
+        entry_ids.append(entry['entry_id'])
+
+    assert entry_ids == sorted(entry_ids)
+
+    result = await backend.get_entries()
+    assert ([i['entry_id'] for i in result['entries']] ==
+            sorted(entry_ids, reverse=True))
+
+    for i in range(entries_count):
+        result = await backend.get_entries(source=f'source{i}')
+        assert len(result['entries']) == 1
+        assert result['entries'][0]['entry_id'] == entry_ids[i]
+
+        result = await backend.get_entries(type=f'type{i}')
+        assert len(result['entries']) == 1
+        assert result['entries'][0]['entry_id'] == entry_ids[i]
+
+        result = await backend.get_entries(last_entry_id=entry_ids[i])
+        assert len(result['entries']) == 1 + i
+
+    await backend.async_close()