From c67014b8591588cad53ade23899cb84fcfe42655 Mon Sep 17 00:00:00 2001
From: Shane Kilkelly <shane.kilkelly@overleaf.com>
Date: Thu, 14 Jan 2021 14:00:48 +0000
Subject: [PATCH] Update SqliteDBStoreTest with swap/compression

---
 .../bridge/db/sqlite/SqliteDBStoreTest.java   | 25 ++++++++++++++-----
 1 file changed, 19 insertions(+), 6 deletions(-)

diff --git a/services/git-bridge/src/test/java/uk/ac/ic/wlgitbridge/bridge/db/sqlite/SqliteDBStoreTest.java b/services/git-bridge/src/test/java/uk/ac/ic/wlgitbridge/bridge/db/sqlite/SqliteDBStoreTest.java
index 02065c7b0c..9c88db88d8 100644
--- a/services/git-bridge/src/test/java/uk/ac/ic/wlgitbridge/bridge/db/sqlite/SqliteDBStoreTest.java
+++ b/services/git-bridge/src/test/java/uk/ac/ic/wlgitbridge/bridge/db/sqlite/SqliteDBStoreTest.java
@@ -69,10 +69,23 @@ public class SqliteDBStoreTest {
         assertEquals("asdf", dbStore.getOldestUnswappedProject());
     }
 
+    @Test
+    public void swapAndRestore() {
+      String projectName = "something";
+      String compression = "bzip2";
+      dbStore.setLatestVersionForProject(projectName, 42);
+      dbStore.swap(projectName, compression);
+      assertNull(dbStore.getOldestUnswappedProject());
+      assertEquals(dbStore.getSwapCompression(projectName), compression);
+      // and restore
+      dbStore.restore(projectName);
+      assertEquals(dbStore.getSwapCompression(projectName), null);
+    }
+
     @Test
     public void noOldestProjectIfAllEvicted() {
         dbStore.setLatestVersionForProject("older", 3);
-        dbStore.setLastAccessedTime("older", null);
+        dbStore.swap("older", "bzip2");
         assertNull(dbStore.getOldestUnswappedProject());
     }
 
@@ -93,7 +106,7 @@ public class SqliteDBStoreTest {
                 )
         );
         assertEquals("older", dbStore.getOldestUnswappedProject());
-        dbStore.setLastAccessedTime("older", null);
+        dbStore.swap("older", "bzip2");
         assertEquals("newer", dbStore.getOldestUnswappedProject());
     }
 
@@ -115,9 +128,9 @@ public class SqliteDBStoreTest {
                 Timestamp.valueOf(LocalDateTime.now())
         );
         assertEquals(1, dbStore.getNumUnswappedProjects());
-        dbStore.setLastAccessedTime(
+        dbStore.swap(
                 "asdf",
-                null
+                "bzip2"
         );
         assertEquals(0, dbStore.getNumUnswappedProjects());
     }
@@ -143,8 +156,8 @@ public class SqliteDBStoreTest {
     @Test
     public void projectStateIsSwappedIfLastAccessedIsNull() {
         dbStore.setLatestVersionForProject("asdf", 1);
-        dbStore.setLastAccessedTime("asdf", null);
+        dbStore.swap("asdf", "bzip2");
         assertEquals(ProjectState.SWAPPED, dbStore.getProjectState("asdf"));
     }
 
-}
\ No newline at end of file
+}