From 32d1fecbb438987e6285c4edbfb861a0d1e651d8 Mon Sep 17 00:00:00 2001
From: Markus Unterwaditzer <markus@unterwaditzer.net>
Date: Sun, 1 Oct 2017 20:37:16 +0200
Subject: [PATCH] New tests for sync status

---
 tests/unit/sync/test_status.py     | 35 +++++++++++++++++++
 tests/unit/{ => sync}/test_sync.py |  0
 vdirsyncer/sync.py                 | 55 ++++++++++++++----------------
 3 files changed, 61 insertions(+), 29 deletions(-)
 create mode 100644 tests/unit/sync/test_status.py
 rename tests/unit/{ => sync}/test_sync.py (100%)

diff --git a/tests/unit/sync/test_status.py b/tests/unit/sync/test_status.py
new file mode 100644
index 0000000..f21e690
--- /dev/null
+++ b/tests/unit/sync/test_status.py
@@ -0,0 +1,35 @@
+import pytest
+
+from hypothesis import assume, given
+import hypothesis.strategies as st
+
+from vdirsyncer.sync import SqliteStatus
+
+
+
+@pytest.fixture(params=[
+    SqliteStatus
+])
+def new_status(request):
+    return request.param
+
+
+@given(status_dict=st.dictionaries(
+    st.text(),
+    st.tuples(*(
+        st.fixed_dictionaries({
+            'href': st.text(),
+            'hash': st.text(),
+            'etag': st.text()
+        }) for _ in range(2)
+    ))
+))
+def test_legacy_status(new_status, status_dict):
+    hrefs_a = {meta_a['href'] for meta_a, meta_b in status_dict.values()}
+    hrefs_b = {meta_b['href'] for meta_a, meta_b in status_dict.values()}
+    assume(len(hrefs_a) == len(status_dict))
+    assume(len(hrefs_b) == len(status_dict))
+
+    status = new_status()
+    status.load_legacy_status(status_dict)
+    assert dict(status.to_legacy_status()) == status_dict
diff --git a/tests/unit/test_sync.py b/tests/unit/sync/test_sync.py
similarity index 100%
rename from tests/unit/test_sync.py
rename to tests/unit/sync/test_sync.py
diff --git a/vdirsyncer/sync.py b/vdirsyncer/sync.py
index 3e8a8d2..c99f82e 100644
--- a/vdirsyncer/sync.py
+++ b/vdirsyncer/sync.py
@@ -116,6 +116,31 @@ class _IdentAlreadyExists(SyncError):
 
 
 class _StatusBase(metaclass=abc.ABCMeta):
+    def load_legacy_status(self, status):
+        with self.transaction():
+            for ident, metadata in status.items():
+                if len(metadata) == 4:
+                    href_a, etag_a, href_b, etag_b = metadata
+                    props_a = _ItemMetadata(href=href_a, hash='UNDEFINED',
+                                            etag=etag_a)
+                    props_b = _ItemMetadata(href=href_b, hash='UNDEFINED',
+                                            etag=etag_b)
+                else:
+                    a, b = metadata
+                    a.setdefault('hash', 'UNDEFINED')
+                    b.setdefault('hash', 'UNDEFINED')
+                    props_a = _ItemMetadata(**a)
+                    props_b = _ItemMetadata(**b)
+
+                self.insert_ident_a(ident, props_a)
+                self.insert_ident_b(ident, props_b)
+
+    def to_legacy_status(self):
+        for ident in self.iter_old():
+            a = self.get_a(ident)
+            b = self.get_b(ident)
+            yield ident, (a.to_status(), b.to_status())
+
     @abc.abstractmethod
     def transaction(self):
         raise NotImplementedError()
@@ -180,41 +205,13 @@ class _StatusBase(metaclass=abc.ABCMeta):
 class SqliteStatus(_StatusBase):
     SCHEMA_VERSION = 1
 
-    def __init__(self, path):
+    def __init__(self, path=':memory:'):
         self._path = path
         self._c = sqlite3.connect(path)
         self._c.isolation_level = None  # turn off idiocy of DB-API
         self._c.row_factory = sqlite3.Row
         self._update_schema()
 
-    def load_legacy_status(self, status):
-        for ident, metadata in status.items():
-            if len(metadata) == 4:
-                href_a, etag_a, href_b, etag_b = metadata
-                params = (ident, href_a, 'UNDEFINED', etag_a, href_b,
-                          'UNDEFINED', etag_b)
-            else:
-                a, b = metadata
-                params = (ident,
-                          a.get('href'), a.get('hash', 'UNDEFINED'),
-                          a.get('etag'),
-                          b.get('href'), b.get('hash', 'UNDEFINED'),
-                          b.get('etag'))
-
-            self._c.execute(
-                'INSERT INTO status'
-                ' (ident, href_a, hash_a, etag_a,'
-                '  href_b, hash_b, etag_b)'
-                ' VALUES (?, ?, ?, ?, ?, ?, ?)',
-                params
-            )
-
-    def to_legacy_status(self):
-        for ident in self.iter_old():
-            a = self.get_a(ident)
-            b = self.get_b(ident)
-            yield ident, (a.to_status(), b.to_status())
-
     def _update_schema(self):
         if self._is_latest_version():
             return