def test_start_game(self):
"""Tests starting new games"""
random.seed(42)
fail = False
print(f"{BColors.OKBLUE}\t[i]\tStarting 20 games and ending them...{BColors.ENDC}")
for _ in range(20):
try:
# Create players
difficulty = random.choice(config.DIFFICULTY_LEVELS)
players = list()
num_players = 1
for _ in range(num_players):
name = random.choice(string.ascii_letters)
players.append("ID" + str(name))
players = ','.join(players)
# Start a new game
url = "/game_board/llist_api/start_game/" + difficulty + '/' + players + '/LLIST'
response = self.client.get(url)
self.assertEqual(response.status_code, 200,
msg=f'{BColors.FAIL}\t[-]\tResponse was not 200!{BColors.ENDC}')
self.assertIn('game_id', response.data.keys(),
msg=f'{BColors.FAIL}\t[-]\tGame ID was not returned!{BColors.ENDC}')
# Remove the test game from the database
sleep(0.2)
db.remove_game(response.data['game_id'])
except Exception as err:
print(f"{BColors.FAIL}\t[-]\tFail creating games: {BColors.ENDC}", str(err))
fail = True
if not fail:
print(f"{BColors.OKGREEN}\t[start_game]\tPass generating games.{BColors.ENDC}")
def test_game_board_state(self):
"""Tests if the game configured as requested. Also tests load"""
# create a new game
created_game = self.client.get('/game_board/llist_api/start_game/Easy/ID1lltest/LLIST')
# load the game
response = self.client.get('/game_board/llist_api/board/' + str(created_game.data['game_id']))
board = response.data
self.assertEqual(board['difficulty'], 'Easy',
msg=f'{BColors.FAIL}\t[-]\tDifficulty does not match!{BColors.ENDC}')
print(f"{BColors.OKGREEN}\t[load_board]\tPass choosing the difficulty level.{BColors.ENDC}")
self.assertEqual(board['curr_data_structure'], 'LLIST',
msg=f'{BColors.FAIL}\t[-]\tCurrent data structure is invalid!{BColors.ENDC}')
print(f"{BColors.OKGREEN}\t[load_board]\tPass setting the data structure.{BColors.ENDC}")
self.assertEqual(board['player_ids'], ['ID1lltest'],
msg=f'{BColors.FAIL}\t[-]\tIncorrect user ID!{BColors.ENDC}')
print(f"{BColors.OKGREEN}\t[load_board]\tPass choosing user ID.{BColors.ENDC}")
self.assertEqual(board['total_food'], 6, msg=f'{BColors.FAIL}\t[-]\tStarting food is not 6!{BColors.ENDC}')
print(f"{BColors.OKGREEN}\t[load_board]\tPass setting the starting food amount.{BColors.ENDC}")
# remove the created game
sleep(0.2)
db.remove_game(created_game.data['game_id'])
def test_move_food(self):
# create a new game
created_game = self.client.get('/game_board/llist_api/start_game/Easy/ID1lltest/LLIST')
# load the game
response = self.client.get('/game_board/llist_api/board/' + str(created_game.data['game_id']))
# call spawn ant function
response = self.client.get('/game_board/llist_api/spawn_ant/' + str(response.data['game_id']))
# call move ant function, from surface -> chamber1 so that we can moving food from chamber1 to chamber 2
response = self.client.get(
'/game_board/llist_api/move_ant/' + str(response.data['game_id']) + '/surface/chamber1')
# call dig tunnel
response = self.client.get(
'/game_board/llist_api/dig_tunnel/' + str(response.data['game_id']) + '/chamber1/None')
# call dig chamber function for tunnel that was just made
response = self.client.get(
'/game_board/llist_api/dig_chamber/' + str(response.data['game_id']) + '/chamber1/no/None')
# call move food function
response = self.client.get(
'/game_board/llist_api/move_food/' + str(response.data['game_id']) + '/chamber1/chamber2')
# make sure there was no error since we can now forage
thing = ' '
if response.status_code == 400:
thing = response.data['invalid_action']
self.assertNotEqual(response.status_code, 400,
msg=f'{BColors.FAIL}\t[-]\tResponse was 400! {thing}{BColors.ENDC}')
self.assertNotEqual(response.status_code, 500, msg=f'{BColors.FAIL}\t[-]\tResponse was 500!{BColors.ENDC}')
self.assertNotEqual(response.status_code, 404, msg=f'{BColors.FAIL}\t[-]\tResponse was 404!{BColors.ENDC}')
print(f"{BColors.OKGREEN}\t[move_food]\tPass returning the correct response code.{BColors.ENDC}")
board = response.data
# make sure time track was updated
self.assertEqual(board['time_tracks']['move/forage'], 34,
msg=f'{BColors.FAIL}\t[-]\tTimetrack was not decremented!{BColors.ENDC}')
print(f"{BColors.OKGREEN}\t[move_food]\tPass updating time tracks.{BColors.ENDC}")
# Make sure chamber1 is updated correctly
self.assertEqual(board['graph']['chambers']['chamber1']['food']['berry'], 0,
msg=f'{BColors.FAIL}\t[-]\tBerry still in chamber1!{BColors.ENDC}')
self.assertEqual(board['graph']['chambers']['chamber1']['food']['donut'], 0,
msg=f'{BColors.FAIL}\t[-]\tChamber1 total not updated!{BColors.ENDC}')
print(f"{BColors.OKGREEN}\t[move_food]\tPass moving food from chamber1.{BColors.ENDC}")
# Make sure chamber2 is updated correctly
self.assertEqual(board['graph']['chambers']['chamber2']['food']['berry'], 1,
msg=f'{BColors.FAIL}\t[-]\tBerry still in chamber1!{BColors.ENDC}')
self.assertEqual(board['graph']['chambers']['chamber2']['food']['total'], 2,
msg=f'{BColors.FAIL}\t[-]\tChamber2 total not updated!{BColors.ENDC}')
print(f"{BColors.OKGREEN}\t[move_food]\tPass moving food from chamber1.{BColors.ENDC}")
# make sure total food didn't change nor total food types.
self.assertEqual(board['total_food'], 3, msg=f'{BColors.FAIL}\t[-]\tTotal Food Changed!{BColors.ENDC}')
print(f"{BColors.OKGREEN}\t[move_food]\tPass not changing total food.{BColors.ENDC}")
# remove the created game
sleep(0.2)
db.remove_game(created_game.data['game_id'])
def test_dig_chamber(self):
# create a new game
created_game = self.client.get('/game_board/llist_api/start_game/Easy/ID1lltest/LLIST')
# load the game
response = self.client.get('/game_board/llist_api/board/' + str(created_game.data['game_id']))
# call spawn ant function
response = self.client.get('/game_board/llist_api/spawn_ant/' + str(response.data['game_id']))
# call move ant function, from surface -> chamber1 so that we can test digging a tunnel after chamber1
response = self.client.get(
'/game_board/llist_api/move_ant/' + str(response.data['game_id']) + '/surface/chamber1')
# call dig tunnel
response = self.client.get(
'/game_board/llist_api/dig_tunnel/' + str(response.data['game_id']) + '/chamber1/None')
# call dig chamber function for tunnel that was just made
response = self.client.get(
'/game_board/llist_api/dig_chamber/' + str(response.data['game_id']) + '/chamber1/no/None')
# Make sure that there was no error in function call
self.assertNotEqual(response.status_code, 400, msg=f'{BColors.FAIL}\t[-]\tResponse was 400!{BColors.ENDC}')
self.assertNotEqual(response.status_code, 500, msg=f'{BColors.FAIL}\t[-]\tResponse was 500!{BColors.ENDC}')
self.assertNotEqual(response.status_code, 404, msg=f'{BColors.FAIL}\t[-]\tResponse was 404!{BColors.ENDC}')
print(f"{BColors.OKGREEN}\t[dig_chamber]\tPass returning the correct response code.{BColors.ENDC}")
board = response.data
# make sure time track was updated
self.assertEqual(board['time_tracks']['dig/fill_chamber'], 35,
msg=f'{BColors.FAIL}\t[-]\tTimetrack was not decremented!{BColors.ENDC}')
print(f"{BColors.OKGREEN}\t[dig_chamber]\tPass updating time tracks.{BColors.ENDC}")
# Make sure the new chamber was put in the chamber list
self.assertIn('chamber2', board['graph']['chambers'],
msg=f'{BColors.FAIL}\t[-]\tChamber2 was not put in chamber list!{BColors.ENDC}')
print(f"{BColors.OKGREEN}\t[dig_chamber]\tPass putting chamber2 in chamber list.{BColors.ENDC}")
# Make sure chamber2 is connected to chamber1
self.assertEqual(board['graph']['chambers']['chamber2']['tunnels']['prev'], 'chamber1',
msg=f'{BColors.FAIL}\t[-]\tChamber2 is not connected to chamber1!{BColors.ENDC}')
print(f"{BColors.OKGREEN}\t[dig_chamber]\tPass connecting chambers with the tunnel.{BColors.ENDC}")
# Make sure ant didn't leave chamber1
self.assertEqual(board['graph']['chambers']['chamber2']['num_ants'], 0,
msg=f'{BColors.FAIL}\t[-]\tAnt was placed in chamber2!{BColors.ENDC}')
self.assertEqual(board['graph']['chambers']['chamber1']['num_ants'], 1,
msg=f'{BColors.FAIL}\t[-]\tAnt was move from chamber1!{BColors.ENDC}')
print(f"{BColors.OKGREEN}\t[dig_chamber]\tPass not moving the ant.{BColors.ENDC}")
# Make sure total number of chambers was updated
self.assertEqual(board['total_chambers'], 2,
msg=f'{BColors.FAIL}\t[-]\tTotal number of chambers was not updated!{BColors.ENDC}')
print(f"{BColors.OKGREEN}\t[dig_chamber]\tPass updating number of chambers.{BColors.ENDC}")
# remove the created game
sleep(0.2)
db.remove_game(created_game.data['game_id'])
def test_delete_correct(self):
""" The document was deleted in the database """
deleted_game = mongo.remove_game( "60afce36-085a-11eb-b6ab-acde48001122" )
self.assertEqual( deleted_game, 1,
msg=f'{BColors.FAIL}\t[-]\tDatabase was not able to delete the game!{BColors.ENDC}')
print(f"{BColors.OKGREEN}\t[+]\tPass gameboard database delete.{BColors.ENDC}")
def test_delete_nonexist(self):
""" The document should not be deleted from the database """
deleted_game = mongo.remove_game(
"This name should'nt exist in the database, and if it does, it deserves to be deleted" )
self.assertEqual( deleted_game, self.fail_phrase,
msg=f'{BColors.FAIL}\t[-]\tIncorrect return for non-existant game!{BColors.ENDC}')
print(f"{BColors.OKGREEN}\t[+]\tPass gameboard database delete nonexistant.{BColors.ENDC}")
def test_fill_chamber(self):
# create a new game
created_game = self.client.get('/game_board/llist_api/start_game/Easy/ID1lltest/LLIST')
# load the game
response = self.client.get('/game_board/llist_api/board/' + str(created_game.data['game_id']))
# call spawn ant function
response = self.client.get('/game_board/llist_api/spawn_ant/' + str(response.data['game_id']))
# call move ant function, from surface -> chamber1 so that we can test digging a tunnel after chamber1
response = self.client.get('/game_board/llist_api/move_ant/' + str(response.data['game_id']) + '/surface/chamber1')
self.assertNotEqual(response.status_code, 400, msg=f'{BColors.FAIL}\t[-]\tResponse was 400!{BColors.ENDC}')
# call dig tunnel
response = self.client.get('/game_board/llist_api/dig_tunnel/' + str(response.data['game_id']) + '/chamber1/None')
# call dig chamber function for tunnel that was just made
response = self.client.get('/game_board/llist_api/dig_chamber/' + str(response.data['game_id']) + '/chamber1/no/None')
response = self.client.get('/game_board/llist_api/fill_chamber/' + str(response.data['game_id']) + '/node1')
# make sure there was an error because selected node does not exist
self.assertEqual(response.status_code, 400, msg=f'{BColors.FAIL}\t[-]\tResponse was not 400!{BColors.ENDC}')
print(f"{BColors.OKGREEN}\t[fill_chamber]\tPass returning the correct response code.{BColors.ENDC}")
response = self.client.get('/game_board/llist_api/fill_chamber/' + str(created_game.data['game_id']) + '/chamber2')
reason = ' '
if response.status_code == 400:
reason = response.data['invalid_action']
self.assertNotEqual(response.status_code, 400, msg=f'{BColors.FAIL}\t[-]\tResponse was 400! {reason}{BColors.ENDC}')
self.assertNotEqual(response.status_code, 500, msg=f'{BColors.FAIL}\t[-]\tResponse was 500!{BColors.ENDC}')
self.assertNotEqual(response.status_code, 404, msg=f'{BColors.FAIL}\t[-]\tResponse was 404!{BColors.ENDC}')
board = response.data
self.assertNotIn('chamber2', board['graph']['chambers'],
msg=f'{BColors.FAIL}\t[-]\tFailed to properly yeet chamber{BColors.ENDC}')
self.assertEqual(board['graph']['chambers']['chamber1']['tunnels']['exit'], False,
msg=f'{BColors.FAIL}\t[-]\tChamber 1 exit still exists{BColors.ENDC}')
self.assertIsNone(board['graph']['chambers']['chamber1']['tunnels']['next'],
msg=f'{BColors.FAIL}\t[-]\tChamber 1 next still exists{BColors.ENDC}')
print(f"{BColors.OKGREEN}\t[fill_chamber]\tPass removing chamber{BColors.ENDC}")
# remove the created game
sleep(0.2)
db.remove_game(created_game.data['game_id'])
def setUp(self):
""" creates game data needed for tests """
self.board = {"game_id": "60afce36-085a-11eb-b6ab-acde48001122",
"graph": {"nodes": "node4(node2(node3)(node1))(node6(node5))",
"node_points": {"node1": 1,"node2": 2,"node3": 3,"node4": 4,"node5": 5,"node6": 6},
"gold_node": "node5","balanced": True},"player_ids": ["id2","id3","id4","id5"],
"player_names": ["naomi","kulsoom","nick","ryan"],
"player_points": {"id2": 2,"id3": 2,"id4": 3,"id5": 10},"turn": "id2",
"cards": {"id2": ["card1","card2","card3"],"id3": ["card1","card2","card3"],
"id4":["card1","card2","card3"],"id5": ["card1","card2","card3"]},
"gold_node": False,"difficulty": "Medium","num_players": 4,"online": True,
"curr_data_structure": "AVL","selected_data_structures": ["AVL","Stack"],
"timed_game": False,"seconds_until_next_ds": 60,"time_created": "07/10/2020 00:05:47",
"end_game": False}
self.fail_phrase = 'nah bro idk about it'
#Make sure the game is cleared out
mongo.remove_game(self.board["game_id"])
def test_move_ant(self):
# create a new game
created_game = self.client.get('/game_board/llist_api/start_game/Easy/ID1lltest/LLIST')
# load the game
response = self.client.get('/game_board/llist_api/board/' + str(created_game.data['game_id']))
# call spawn ant function so that we can test moving it
response = self.client.get('/game_board/llist_api/spawn_ant/' + str(response.data['game_id']))
# call move ant function, from surface -> chamber1
response = self.client.get(
'/game_board/llist_api/move_ant/' + str(response.data['game_id']) + '/surface/chamber1')
board = response.data
# make sure time track was updated
self.assertEqual(board['time_tracks']['move/forage'], 35,
msg=f'{BColors.FAIL}\t[-]\tTimetrack was not decremented!{BColors.ENDC}')
print(f"{BColors.OKGREEN}\t[move_ant]\tPass updating time tracks.{BColors.ENDC}")
# make sure no error, bc you should be able to move the ant from surface to the chamber1
self.assertNotEqual(response.status_code, 400, msg=f'{BColors.FAIL}\t[-]\tResponse was not 400!{BColors.ENDC}')
self.assertNotEqual(response.status_code, 500, msg=f'{BColors.FAIL}\t[-]\tResponse was not 500!{BColors.ENDC}')
print(f"{BColors.OKGREEN}\t[move_ant]\tPass returning the correct response code.{BColors.ENDC}")
# make sure number of surface ants were updated (only queen on surface)
self.assertEqual(board['total_surface_ants'], 1,
msg=f'{BColors.FAIL}\t[-]\tAnt still on surface!{BColors.ENDC}')
self.assertEqual(board['graph']['chambers']['surface']['num_ants'], 1,
msg=f'{BColors.FAIL}\t[-]\tnum_ants.surface not updated!{BColors.ENDC}')
print(f"{BColors.OKGREEN}\t[move_ant]\tPass taking ant off of surface.{BColors.ENDC}")
# make sure ant was placed in chamber1
self.assertEqual(board['graph']['chambers']['chamber1']['num_ants'], 1,
msg=f'{BColors.FAIL}\t[-]\tAnt was not placed in chamber1!{BColors.ENDC}')
print(f"{BColors.OKGREEN}\t[move_ant]\tPass placing ant in the cmahber.{BColors.ENDC}")
# remove the created game
sleep(0.2)
db.remove_game(created_game.data['game_id'])
def test_spawn(self):
# create a new game
created_game = self.client.get('/game_board/llist_api/start_game/Easy/ID1lltest/LLIST')
# load the game
response = self.client.get('/game_board/llist_api/board/' + str(created_game.data['game_id']))
# call spawn ant function
response = self.client.get('/game_board/llist_api/spawn_ant/' + str(response.data['game_id']))
board = response.data
# make sure there is no error
self.assertNotEqual(response.status_code, 400, msg=f'{BColors.FAIL}\t[-]\tCould not spawn ant!{BColors.ENDC}')
print(f"{BColors.OKGREEN}\t[spawn_ant]\tPass spawn ant api.{BColors.ENDC}")
# make sure food was taken
self.assertEqual(board['total_food'], 3, msg=f'{BColors.FAIL}\t[-]\tTotal food is off!{BColors.ENDC}')
print(f"{BColors.OKGREEN}\t[spawn_ant]\tPass taking food after spawn.{BColors.ENDC}")
# make sure ant total was updated
self.assertEqual(board['total_ants'], 2, msg=f'{BColors.FAIL}\t[-]\tTotal ants are off!{BColors.ENDC}')
print(f"{BColors.OKGREEN}\t[spawn_ant]\tPass creating a new ant.{BColors.ENDC}")
# make sure ant was placed on surface
self.assertEqual(board['total_surface_ants'], 2,
msg=f'{BColors.FAIL}\t[-]\tNew ant not on surface!{BColors.ENDC}')
self.assertEqual(board['graph']['chambers']['surface']['num_ants'], 2,
msg=f'{BColors.FAIL}\t[-]\tnum_ants.surface not updated!{BColors.ENDC}')
print(f"{BColors.OKGREEN}\t[spawn_ant]\tPass placing ant on surface.{BColors.ENDC}")
# make sure donut was taken
self.assertEqual(board['total_food_types']['donut'], 0,
msg=f'{BColors.FAIL}\t[-]\tDonut not taken!{BColors.ENDC}')
print(f"{BColors.OKGREEN}\t[spawn_ant]\tPass taking a donut.{BColors.ENDC}")
# remove the created game
sleep(0.2)
db.remove_game(created_game.data['game_id'])
def test_dig_tunnel(self):
# create a new game
created_game = self.client.get('/game_board/llist_api/start_game/Easy/ID1lltest/LLIST')
# load the game
response = self.client.get('/game_board/llist_api/board/' + str(created_game.data['game_id']))
# call spawn ant function
response = self.client.get('/game_board/llist_api/spawn_ant/' + str(response.data['game_id']))
# call move ant function, from surface -> chamber1 so that we can test digging a tunnel after chamber1
response = self.client.get(
'/game_board/llist_api/move_ant/' + str(response.data['game_id']) + '/surface/chamber1')
# call dig tunnel
response = self.client.get(
'/game_board/llist_api/dig_tunnel/' + str(response.data['game_id']) + '/chamber1/None')
# make sure there was no error since we can now dig a tunnel
self.assertNotEqual(response.status_code, 400, msg=f'{BColors.FAIL}\t[-]\tResponse was 400!{BColors.ENDC}')
self.assertNotEqual(response.status_code, 500, msg=f'{BColors.FAIL}\t[-]\tResponse was 500!{BColors.ENDC}')
self.assertNotEqual(response.status_code, 404, msg=f'{BColors.FAIL}\t[-]\tResponse was 404!{BColors.ENDC}')
print(f"{BColors.OKGREEN}\t[dig_tunnel]\tPass returning the correct response code.{BColors.ENDC}")
board = response.data
# make sure time track was updated
self.assertEqual(board['time_tracks']['dig_tunnel_track'], 35,
msg=f'{BColors.FAIL}\t[-]\tTimetrack was not decremented!{BColors.ENDC}')
print(f"{BColors.OKGREEN}\t[dig_tunnel]\tPass updating dig tunnel time track.{BColors.ENDC}")
# make sure chamber1 no longer has a tunnel going to the surface, but has one going no where.
self.assertIsNone(board['graph']['chambers']['chamber1']['tunnels']['next'],
msg=f'{BColors.FAIL}\t[-]\tchamber1\'s next tunnel not None!{BColors.ENDC}')
print(
f"{BColors.OKGREEN}\t[dig_tunnel]\tPass changing tunnel from surface to tunnel leading no where.{BColors.ENDC}")
# remove the created game
sleep(0.2)
db.remove_game(created_game.data['game_id'])
def update_board_db(board, user_id='-1', token='-1'):
"""
Update the game board in the database with the new state.
Returns the board itself unless the game ended.
If the game ended, then it changes the <end_game> field,
and deletes the game from database. Otherwise, changes
turn to the next player.
:param board: game board
:param user_id: username
:param token: authentication token
:return result, {'error': bool, 'reason': string, 'game_board': dict}:
"""
result = {'error': False, 'reason': '', 'game_board': board}
try:
if board['curr_data_structure'] == 'AVL':
# Game ended
if (board['graph']['root_node'] == board['graph']['gold_node'] or
len(board['deck']) == 0):
# update the board
board['end_game'] = True
board['turn'] = max(board['player_points'], key=board['player_points'].get) # get player w/ max points
result['game_board'] = board
# if user is authenticated
if user_id not in ['-1', -1, ''] and token not in ['-1', -1, '']:
# check if the game is being loaded from profile page
if 'profile_load' in list(board.keys()) and board['profile_load'] is False:
# Here check if user_id matches the token with the database
if profile_db.check_user(user_id, token):
# Currently error return if user is not authenticated is disabled
# It just not updates the score
#result['error'] = True
#result['reason'] = "User is not authenticated"
#return result
# if the user is in part of the players (paranoid check)
if str(user_id) in board['player_ids']:
# if not negative points
if board['player_points'][str(user_id)] > 0:
# get user's current points
curr_points = profile_db.get_points(str(user_id))
# get the target points
if user_id == board['turn']:
# if the user is the winner, double the points
target_points = curr_points + (math.log(board['player_points'][str(user_id)]) * 2)
else:
# don't double the points if the user is not the winner
target_points = curr_points + math.log(board['player_points'][str(user_id)])
# set the new points
profile_db.set_points(str(user_id), target_points)
# remove the game from the database
db.remove_game(board['game_id'])
# Game continues
else:
_ = db.update_game(board['game_id'], board)
# hide the UID used by data structure backend from user
# del board['graph']['uid']
# Update
result['game_board'] = board
else:
_ = db.update_game(board['game_id'], board)
# hide the UID used by data structure backend from user
# del board['graph']['uid']
# Update
result['game_board'] = board
except Exception as err:
result['error'] = True
result['reason'] = str(err)
return result
return result
def tearDown(self):
""" Removes game data needed in tests """
mongo.remove_game(self.board["game_id"])
def tearDown(self):
""" Removes game data needed in tests """
mongo.remove_game(self.board["game_id"])
# Called in case purge fails
mongo.remove_game(self.board2["game_id"])
def test_forage(self):
# create a new game
created_game = self.client.get('/game_board/llist_api/start_game/Easy/ID1lltest/LLIST')
# load the game
response = self.client.get('/game_board/llist_api/board/' + str(created_game.data['game_id']))
# call spawn ant function so that we can test foraging with it
response = self.client.get('/game_board/llist_api/spawn_ant/' + str(response.data['game_id']))
# call spawn ant function THIS WILL FAIL UNTIL I MEET WITH DAVID
response = self.client.get('/game_board/llist_api/forage/' + str(response.data['game_id']) + '/Easy/chamber1')
# make sure there was no error since we can now forage
self.assertNotEqual(response.status_code, 400, msg=f'{BColors.FAIL}\t[-]\tResponse was 400!{BColors.ENDC}')
self.assertNotEqual(response.status_code, 500, msg=f'{BColors.FAIL}\t[-]\tResponse was 500!{BColors.ENDC}')
self.assertNotEqual(response.status_code, 404, msg=f'{BColors.FAIL}\t[-]\tResponse was 404!{BColors.ENDC}')
print(f"{BColors.OKGREEN}\t[forage]\tPass returning the correct response code.{BColors.ENDC}")
board = response.data
# make sure time track was updated
self.assertEqual(board['time_tracks']['move/forage'], 35,
msg=f'{BColors.FAIL}\t[-]\tTimetrack was not decremented!{BColors.ENDC}')
print(f"{BColors.OKGREEN}\t[forage]\tPass updating time tracks.{BColors.ENDC}")
# make sure number of surface ants were updated (only queen should on surface)
self.assertEqual(board['total_surface_ants'], 1,
msg=f'{BColors.FAIL}\t[-]\tAnt still on surface!{BColors.ENDC}')
self.assertEqual(board['graph']['chambers']['surface']['num_ants'], 1,
msg=f'{BColors.FAIL}\t[-]\tnum_ants.surface not updated!{BColors.ENDC}')
print(f"{BColors.OKGREEN}\t[forage]\tPass taking ant off of surface.{BColors.ENDC}")
# make sure ant was placed in chamber1
self.assertEqual(board['graph']['chambers']['surface']['num_ants'], 1,
msg=f'{BColors.FAIL}\t[-]\tAnt was not placed in chamber1!{BColors.ENDC}')
print(f"{BColors.OKGREEN}\t[forage]\tPass placing ant in chamber1.{BColors.ENDC}")
# make sure that we can assert the correct things if under attack
if board['graph']['chambers']['chamber1']['under_attack']:
# make sure number of surface ants were updated (only queen should on surface)
self.assertTrue(board['graph']['chambers']['chamber1']['under_attack'],
msg=f'{BColors.FAIL}\t[-]\tUnder attack was not updated!{BColors.ENDC}')
print(f"{BColors.OKGREEN}\t[forage]\tPass being able to come under attack.{BColors.ENDC}")
elif board['graph']['chambers']['chamber1']['food']['crumb'] >= 2:
# make sure number of surface ants were updated (only queen should on surface)
self.assertEqual(board['graph']['chambers']['chamber1']['food']['crumb'], 2,
msg=f'{BColors.FAIL}\t[-]\tcrumb was placed in chamber but too many crumbs!{BColors.ENDC}')
self.assertEqual(board['total_food_types']['crumb'], 2,
msg=f'{BColors.FAIL}\t[-]\ttotal_food_types was not updated!{BColors.ENDC}')
print(f"{BColors.OKGREEN}\t[forage]\tPass being able to forage for a crumb.{BColors.ENDC}")
# make sure total_food was updated
self.assertEqual(board['total_food'], 4, msg=f'{BColors.FAIL}\t[-]\ttotal_food is off!{BColors.ENDC}')
print(f"{BColors.OKGREEN}\t[forage]\tPass updating total_food.{BColors.ENDC}")
elif board['graph']['chambers']['chamber1']['food']['berry'] >= 2:
# make sure number of surface ants were updated (only queen should on surface)
self.assertEqual(board['graph']['chambers']['chamber1']['food']['berry'], 2,
msg=f'{BColors.FAIL}\t[-]\tberry was placed in chamber but too many berries!{BColors.ENDC}')
self.assertEqual(board['total_food_types']['berry'], 2,
msg=f'{BColors.FAIL}\t[-]\ttotal_food_types was not updated!{BColors.ENDC}')
print(f"{BColors.OKGREEN}\t[forage]\tPass being able to forage for a berry.{BColors.ENDC}")
# make sure total_food was updated
self.assertEqual(board['total_food'], 5, msg=f'{BColors.FAIL}\t[-]\ttotal_food is off!{BColors.ENDC}')
print(f"{BColors.OKGREEN}\t[forage]\tPasss updating total_food.{BColors.ENDC}")
elif board['graph']['chambers']['chamber1']['food']['donut'] >= 2:
# make sure number of surface ants were updated (only queen should on surface)
self.assertEqual(board['graph']['chambers']['chamber1']['food']['donut'], 2,
msg=f'{BColors.FAIL}\t[-]\tdonut was placed in chamber but too many crumbs!{BColors.ENDC}')
self.assertEqual(board['total_food_types']['donut'], 2,
msg=f'{BColors.FAIL}\t[-]\ttotal_food_types was not updated!{BColors.ENDC}')
print(f"{BColors.OKGREEN}\t[forage]\tPass being able to forage for a donut.{BColors.ENDC}")
# make sure total_food was updated
self.assertEqual(board['total_food'], 6, msg=f'{BColors.FAIL}\t[-]\ttotal_food is off!{BColors.ENDC}')
print(f"{BColors.OKGREEN}\t[forage]\tPasss updating total_food.{BColors.ENDC}")
# remove the created game
sleep(0.2)
db.remove_game(created_game.data['game_id'])