Add test bot client with registered user

Author: leddcode

app/routers/telegram.py

@@ -26,6 +26,7 @@ def bot_client(req: dict = Body(...), session=Depends(get_db)):
 
     # Check if current chatter is registered to use the bot
     user = session.query(User).filter_by(telegram_id=chat.user_id).first()
+    print(user)
     if user is None:
         return reply_unknown_user(chat)
 

tests/client_fixture.py

@@ -57,7 +57,7 @@ def get_test_placeholder_user():
         email='fake@mail.fake',
         password='123456fake',
         full_name='FakeName',
-        telegram_id=''
+        telegram_id='666666'
     )
 
 

tests/test_telegram.py

@@ -188,14 +188,25 @@ def test_reply_unknown_user():
 '''
 
 
-def test_telegram_router(profile_test_client):
-    req = profile_test_client.get('/telegram')
-    assert req.ok
-    assert b"Start using PyLander telegram bot!" in req.content
+class TestBotClient:
 
+    @staticmethod
+    def test_telegram_router(profile_test_client):
+        req = profile_test_client.get('/telegram')
+        assert req.ok
+        assert b"Start using PyLander telegram bot!" in req.content
+
+    @staticmethod
+    def test_user_not_registered(telegram_client):
+        req = telegram_client.post('/telegram/', json=gen_message('/start'))
+        assert req.ok
+        assert b'Hello, Moshe!' in req.content
+        assert b'To use PyLander Bot you have to register' in req.content
 
-def test_bot_client(telegram_client):
-    req = telegram_client.post('/telegram/', json=gen_message('/start'))
-    assert req.ok
-    assert b'Hello, Moshe!' in req.content
-    assert b'To use PyLander Bot you have to register' in req.content
+    @staticmethod
+    def test_user_registered(telegram_client, session):
+        session.add(get_test_placeholder_user())
+        session.commit()
+        req = telegram_client.post('/telegram/', json=gen_message('/start'))
+        assert req.ok
+        assert b'Welcome to Pylander telegram client!' in req.content