Tuuli/based
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

[DB] metadata CD; D from CRUD

Browse source
This commit is contained in:
Andrew 2023-03-11 23:32:50 +07:00
parent 0522918fdd
commit a533dddf1a
2 changed files with 124 additions and 24 deletions
Download patch file Download diff file Expand all files Collapse all files

34
db_test.py
View file

@ -1,6 +1,6 @@
import unittest
import uuid
from db import DBConnector
from db import DBConnector, ColumnUpdate, ColumnCondition
from columns import *
conninfo = "postgresql://postgres:asarch6122@localhost"
@ -133,12 +133,42 @@ class TestDBConnector(unittest.TestCase):
rows = connector.selectFromTable("test_table", ["name"])
self.assertEqual(rows[0]["name"], "John Doe")
connector.updateDataInTable(
"test_table", {"name": "John"}, {"name": "John Doe"}
"test_table",
[
ColumnUpdate("name", "John"),
],
[
ColumnCondition("name", "John Doe"),
],
)
rows = connector.selectFromTable("test_table", ["name"])
self.assertEqual(rows[0]["name"], "John")
connector.dropTable("test_table")
def test_tableDeleteFrom(self):
connector.dropTable("test_table")
connector.createTable(
"test_table",
[
PrimarySerialColumnDefinition("id"),
TextColumnDefinition("name", "John Doe"),
],
)
connector.insertIntoTable("test_table", {"name": "John Doe"})
connector.insertIntoTable("test_table", {"name": "Jane Doe"})
connector.insertIntoTable("test_table", {"name": "Mikhail Prokopenko"})
rows = connector.selectFromTable("test_table", ["name"])
self.assertEqual(len(rows), 3)
self.assertEqual(rows[0]["name"], "John Doe")
self.assertEqual(rows[1]["name"], "Jane Doe")
self.assertEqual(rows[2]["name"], "Mikhail Prokopenko")
connector.deleteFromTable("test_table", [ColumnCondition("name", "John Doe")])
rows = connector.selectFromTable("test_table", ["name"])
self.assertEqual(len(rows), 2)
self.assertEqual(rows[0]["name"], "Jane Doe")
self.assertEqual(rows[1]["name"], "Mikhail Prokopenko")
connector.dropTable("test_table")
if __name__ == "__main__":
unittest.main()