From 575917fb5b21ce01b75dbd7228f656ecd5bdcbd6 Mon Sep 17 00:00:00 2001 From: def Date: Sun, 15 Jan 2023 03:01:04 +0400 Subject: [PATCH] fix tests, add new abstract methods --- mirrortea/repository/abstract_db_repository.py | 8 ++++++++ tests/test_repository.py | 12 ++++++------ 2 files changed, 14 insertions(+), 6 deletions(-) diff --git a/mirrortea/repository/abstract_db_repository.py b/mirrortea/repository/abstract_db_repository.py index 2b98deb..050c960 100644 --- a/mirrortea/repository/abstract_db_repository.py +++ b/mirrortea/repository/abstract_db_repository.py @@ -17,6 +17,10 @@ class AbstractDatabaseRepository(ABC): def get_user_by_id(self) -> User: """""" + @abstractmethod + def update_user_by_id(self) -> User: + """""" + @abstractmethod def delete_user(self) -> None: """""" @@ -29,6 +33,10 @@ class AbstractDatabaseRepository(ABC): def get_matrix_room(self) -> MatrixRoom: """""" + @abstractmethod + def update_matrix_room(self) -> MatrixRoom: + """""" + @abstractmethod def delete_matrix_room(self) -> None: """""" diff --git a/tests/test_repository.py b/tests/test_repository.py index 4cb36c6..2d21205 100644 --- a/tests/test_repository.py +++ b/tests/test_repository.py @@ -13,16 +13,11 @@ def create_repository_object(): def test_create_tables(): - os.system("rm -rf " + PATH) + os.system("rm -f " + PATH) repo = create_repository_object() assert repo.create_tables() is None -def test_get_user_by_id(): - repo = create_repository_object() - assert repo.get_user_by_id(user_id="123") is None - - def test_create_user(): repo = create_repository_object() assert ( @@ -37,3 +32,8 @@ def test_create_user(): ) is None ) + + +def test_get_user_by_id(): + repo = create_repository_object() + assert repo.get_user_by_id(user_id="123") is None