def test_team_bots_return_team_bots_list_without_token(self):
data = MagicMock()
'''Mocked outputs'''
user = User(user_id=0)
user.team_id = 0
bot1 = Bot(bot_id=1,
name="Bot-Test1",
callback=None,
token="Test-Token")
bot2 = Bot(bot_id=2,
name="Bot-Test2",
callback=None,
token="Test-Token")
bots = [bot1, bot2]
sys.modules[
"models.authentication"].Authenticator.authenticate_team.return_value = user
sys.modules[
"daos.bots"].BotDatabaseClient.get_team_bots.return_value = bots
response = BotService.team_bots(data)
self.assertEqual(UserResponseStatus.LIST.value,
response.json().get("status"))
self.assertEqual(1, response.json().get("bots")[0].get("id"))
self.assertEqual(2, response.json().get("bots")[1].get("id"))
self.assertEqual("Bot-Test1",
response.json().get("bots")[0].get("name"))
self.assertEqual("Bot-Test2",
response.json().get("bots")[1].get("name"))
self.assertIsNone(response.json().get("bots")[0].get("callback"))
self.assertIsNone(response.json().get("bots")[1].get("callback"))
self.assertFalse("token" in response.json().get("bots")[0])
self.assertFalse("token" in response.json().get("bots")[1])
self.assertIsInstance(response, SuccessfulBotListResponse)
def get_team_bots(team_id):
logger.info(f"Attempting to get team #{team_id}'s bots.")
req = ClientRequest(request)
response = BotService.team_bots(req.team_authentication_data(team_id))
return jsonify(response.json()), response.status_code()