diff --git a/lib/python/tests/test_unit_container.py b/lib/python/tests/test_unit_container.py
index 8f3297879ada5fcf416cc8afc508d059cf28c95d..94d18b79fdf3953cba79d41a80bda9dd0d5a58a3 100644
--- a/lib/python/tests/test_unit_container.py
+++ b/lib/python/tests/test_unit_container.py
@@ -7,8 +7,6 @@ from dbrepo.RestClient import RestClient
from dbrepo.api.dto import Container, Image, ContainerBrief, ImageBrief
from dbrepo.api.exceptions import ResponseCodeError, NotExistsError
-from dbrepo.api.dto import ImageDate
-
class ContainerUnitTest(unittest.TestCase):
@@ -69,29 +67,10 @@ class ContainerUnitTest(unittest.TestCase):
dialect="org.hibernate.dialect.MariaDBDialect",
driver_class="org.mariadb.jdbc.Driver",
jdbc_method="mariadb",
- date_formats=[
- ImageDate(id=1,
- example="2024-03-26 10:26:00",
- database_format="%Y-%c-%d %H:%i:%S",
- unix_format="yyyy-MM-dd HH:mm:ss",
- has_time=True,
- created_at=datetime.datetime(2024, 3, 26, 10, 26, 0, 0,
- datetime.timezone.utc)),
- ImageDate(id=2,
- example="2024-03-26",
- database_format="%Y-%c-%d",
- unix_format="yyyy-MM-dd",
- has_time=False,
- created_at=datetime.datetime(2024, 3, 26, 0, 0, 0, 0,
- datetime.timezone.utc)),
- ImageDate(id=3,
- example="10:25:01",
- database_format="%Y-%c-%d",
- unix_format="yyyy-MM-dd",
- has_time=False,
- created_at=datetime.datetime(2024, 3, 26, 0, 0, 0, 0,
- datetime.timezone.utc)),
- ]),
+ data_types=[
+ DataType(display_name="SERIAL", value="serial",
+ documentation="https://mariadb.com/kb/en/bigint/",
+ is_quoted=False, is_buildable=True)]),
hash="f829dd8a884182d0da846f365dee1221fd16610a14c81b8f9f295ff162749e50")
# mock
mock.get('/api/container/1', json=exp.model_dump())
diff --git a/lib/python/tests/test_unit_database.py b/lib/python/tests/test_unit_database.py
index dea15691e22a990434772c433683eb6eefc0b253..e7cefa6d5b7e829f2f30dd1befff453d6d40a9de 100644
--- a/lib/python/tests/test_unit_database.py
+++ b/lib/python/tests/test_unit_database.py
@@ -10,8 +10,6 @@ from dbrepo.api.dto import Database, User, Container, Image, UserAttributes, Dat
UserBrief
from dbrepo.api.exceptions import ResponseCodeError, NotExistsError, ForbiddenError, MalformedError, AuthenticationError
-from dbrepo.api.dto import ImageDate
-
class DatabaseUnitTest(unittest.TestCase):
@@ -72,7 +70,11 @@ class DatabaseUnitTest(unittest.TestCase):
dialect='org.hibernate.dialect.MariaDBDialect',
driver_class='org.mariadb.jdbc.Driver',
jdbc_method='mariadb',
- default_port=3306
+ default_port=3306,
+ data_types=[
+ DataType(display_name="SERIAL", value="serial",
+ documentation="https://mariadb.com/kb/en/bigint/",
+ is_quoted=False, is_buildable=True)]
)
)
)