def test_obtain_reward(self):
wifi = Reward(MOCK_REWARD_WIFI['name'], MOCK_REWARD_WIFI['type'])
reward_list = RewardList()
reward_list.add_pending_reward(wifi)
self.assertEqual(len(reward_list.pending_rewards), 1)
self.assertEqual(len(reward_list.obtained_rewards), 0)
self.assertEqual(reward_list.is_obtained(MOCK_REWARD_WIFI['name']), False)
reward_list.obtain_reward(MOCK_REWARD_WIFI['name'])
self.assertEqual(len(reward_list.pending_rewards), 0)
self.assertEqual(len(reward_list.obtained_rewards), 1)
self.assertEqual(reward_list.is_obtained(MOCK_REWARD_WIFI['name']), True)
def test_reward_name_listing(self):
wifi = Reward(MOCK_REWARD_WIFI['name'], MOCK_REWARD_WIFI['type'], MOCK_REWARD_WIFI['stats'], 1,
MOCK_REWARD_WIFI['rewards'])
reward_list = RewardList()
self.assertEqual(reward_list.list_reward_names(), [])
reward_list.add_pending_reward(wifi)
self.assertEqual(reward_list.list_reward_names(), [MOCK_REWARD_WIFI['name']])
self.assertEqual(reward_list.list_obtained_reward_names(), [])
self.assertEqual(reward_list.list_pending_reward_names(), [MOCK_REWARD_WIFI['name']])
reward_list.obtain_reward(MOCK_REWARD_WIFI['name'])
self.assertEqual(reward_list.list_obtained_reward_names(), [MOCK_REWARD_WIFI['name']])
self.assertEqual(reward_list.list_pending_reward_names(), [])
self.assertEqual(reward_list.list_reward_names(), [MOCK_REWARD_WIFI['name']])
def test_reward_requirement(self):
wifi = Reward(MOCK_REWARD_WIFI['name'], MOCK_REWARD_WIFI['type'], MOCK_REWARD_WIFI['stats'], MOCK_REWARD_WIFI['materials'],
MOCK_REWARD_WIFI['rewards'])
satellite = Reward(MOCK_REWARD_SATELLITE['name'], MOCK_REWARD_SATELLITE['type'], MOCK_REWARD_SATELLITE['stats'],
MOCK_REWARD_SATELLITE['materials'], MOCK_REWARD_SATELLITE['rewards'])
reward_list = RewardList()
reward_list.add_pending_reward(wifi)
reward_list.add_pending_reward(satellite)
self.assertEqual(len(reward_list.pending_rewards), 2)
self.assertEqual(len(reward_list.obtained_rewards), 0)
self.assertEqual(len(reward_list.possible_rewards(stats=Stats(MAXED_STATS))), 1)
reward_list.obtain_reward(MOCK_REWARD_WIFI['name'])
self.assertEqual(len(reward_list.pending_rewards), 1)
self.assertEqual(len(reward_list.obtained_rewards), 1)
self.assertEqual(len(reward_list.possible_rewards(stats=Stats(MAXED_STATS))), 1)