diff --git a/tests/conftest.py b/tests/conftest.py index 2cb9a49..5f76e7b 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -41,6 +41,8 @@ def seed(session): [ Collaborator(name="Col1", email="aa", phone=1, team_id=1), Collaborator(name="Col2", email="bb", phone=2, team_id=2), + Collaborator(name="Col3", email="cc", phone=3, team_id=3), + Collaborator(name="Col4", email="dd", phone=4, team_id=2), ] ) # session.add_all( diff --git a/tests/tools/test_tools.py b/tests/tools/test_tools.py index bb7925e..3ef4b32 100644 --- a/tests/tools/test_tools.py +++ b/tests/tools/test_tools.py @@ -8,7 +8,7 @@ class TestCustomerTools: def test_db_should_be_populated(self, seed, session): test = session.query(Customer).all() - # print(test) + assert test != None def test_list_should_return_all_customers(self, seed, session): tools = CustomerTools(session) @@ -31,7 +31,18 @@ class TestCollaboratorTools: assert reply == 1 def test_should_reply_id_by_team_id(self, seed, session): - pass + tool = CollaboratorTools(session) + reply = tool.get_by_team_id(1) + assert len(reply) == 1 + reply = tool.get_by_team_id(2) + assert len(reply) == 2 + + def test_should_reply_team_id_by_name(self, seed, session): + tool = CollaboratorTools(session) + reply = tool.get_team_by_name("Col1") + reply2 = tool.get_team_by_name("Col2") + assert reply == 1 + assert reply2 == 2 class TestPasswordTools: