add id: Serial annotations

piccolo-orm · Nov 30, 2023 · c05bb5d · c05bb5d
1 parent b42f891
commit c05bb5d
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/tests/table/test_insert.py b/tests/table/test_insert.py
@@ -3,7 +3,7 @@
 
 import pytest
 
-from piccolo.columns import Integer, Varchar
+from piccolo.columns import Integer, Serial, Varchar
 from piccolo.query.methods.insert import OnConflictAction
 from piccolo.table import Table
 from piccolo.utils.lazy_loader import LazyLoader
@@ -98,6 +98,7 @@ def test_insert_returning_alias(self):
 )
 class TestOnConflict(TestCase):
     class Band(Table):
+        id: Serial
         name = Varchar(unique=True)
         popularity = Integer()
 

diff --git a/tests/table/test_join_on.py b/tests/table/test_join_on.py
@@ -1,26 +1,28 @@
 from unittest import TestCase
 
-from piccolo.columns import Varchar
+from piccolo.columns import Serial, Varchar
 from piccolo.table import Table
 
 
 class Manager(Table):
+    id: Serial
     name = Varchar(unique=True)
     email = Varchar(unique=True)
 
 
 class Band(Table):
+    id: Serial
     name = Varchar(unique=True)
     manager_name = Varchar()
 
 
 class Concert(Table):
+    id: Serial
     title = Varchar()
     band_name = Varchar()
 
 
 class TestJoinOn(TestCase):
-
     tables = [Manager, Band, Concert]
 
     def setUp(self):