1 files changed, 13 insertions, 13 deletions
diff --git a/tests/test_sqlite_storage.py b/tests/test_sqlite_storage.py
index 384158c..6a93561 100755
--- a/tests/test_sqlite_storage.py
+++ b/tests/test_sqlite_storage.py
@@ -6,7 +6,7 @@ import tempfile
 import aiosqlite
 import string
 import random
-from pypaste.server import Paste
+from pypaste.server import Paste, Key, keygen
 from pypaste.server.sqlite import Sqlite
 from datetime import datetime
 from pathlib import Path
@@ -24,11 +24,11 @@ def generate_key() -> str:
 
 
 async def test_exists_but_not_in_our_table(storage: Sqlite) -> None:
-    key = generate_key()
+    key = keygen(6)
 
     await storage.connection.execute(
-        "insert into pastes values(?, ?, ?, ?)",
-        (key, datetime.now().isoformat(), None, bytes()),
+        "insert into pastes values(?, ?, ?, ?, ?)",
+        (key.data, key.length, datetime.now().isoformat(), None, bytes()),
     )
 
     assert not await storage.exists(key)
@@ -36,20 +36,20 @@ async def test_exists_but_not_in_our_table(storage: Sqlite) -> None:
 
 async def test_exists(storage: Sqlite) -> None:
     dt = datetime.now()
-    key = generate_key()
+    key = keygen(6)
 
-    await storage.insert(Paste(key, dt, "test", "hello world"))
+    await storage.insert(Paste(dt, "test", "hello world"), key)
 
     assert await storage.exists(key)
 
-    assert not await storage.exists(generate_key())
+    assert not await storage.exists(keygen(6))
 
 
 async def test_delete(storage: Sqlite) -> None:
     dt = datetime.now()
-    key = generate_key()
+    key = keygen(6)
 
-    await storage.insert(Paste(key, dt, "test", "hello world"))
+    await storage.insert(Paste(dt, "test", "hello world"), key)
 
     assert await storage.exists(key)
 
@@ -60,14 +60,13 @@ async def test_delete(storage: Sqlite) -> None:
 
 async def test_insert_retrieve(storage: Sqlite) -> None:
     dt = datetime.now()
-    key = generate_key()
+    key = keygen(6)
 
-    await storage.insert(Paste(key, dt, "test", "hello world"))
+    await storage.insert(Paste(dt, "test", "hello world"), key)
 
     paste = await storage.retrieve(key)
 
     assert paste is not None
-    assert paste.key == key
     assert paste.dt == dt
     assert paste.syntax == "test"
     assert paste.text == "hello world"
@@ -81,7 +80,8 @@ async def main() -> int:
             await connection.execute(
                 (
                     "create table pastes("
-                    "key text,"
+                    "key blob,"
+                    "key_length int,"
                     "datetime text,"
                     "size int,"
                     "syntax text"