def test_add_badge_to_token(self): result = handlers.handle_badges(event={ 'token': self.good_empty_token, 'key': 'emptyoke' }, context={}) self.assertEqual(result['badges'], ['k'])
def test_get_with_no_token(self): no_token_result = handlers.handle_discography(event={}, context={}) self.assertIn('discography', no_token_result) self.assertEqual(3, len(no_token_result['discography'])) self.assertEqual(no_token_result['badges'], []) self.assertIn('token', no_token_result) self.assertIsNotNone(no_token_result['token']) verify = handlers.handle_badges( event={'token': no_token_result['token']}, context={})
def test_get_with_karaoke(self): karaoke_token_result = handlers.handle_discography( event={'token': self.good_token_with_karaoke}, context={}) self.assertIn('discography', karaoke_token_result) self.assertEqual(4, len(karaoke_token_result['discography'])) self.assertEqual(karaoke_token_result['badges'], ['k']) self.assertIn('token', karaoke_token_result) self.assertIsNotNone(karaoke_token_result['token']) verify = handlers.handle_badges( event={'token': karaoke_token_result['token']}, context={})
def test_get_with_good_empty_token(self): empty_token_result = handlers.handle_discography( event={'token': self.good_empty_token}, context={}) self.assertIn('discography', empty_token_result) self.assertEqual(3, len(empty_token_result['discography'])) self.assertEqual(empty_token_result['badges'], []) self.assertIn('token', empty_token_result) self.assertIsNotNone(empty_token_result['token']) verify = handlers.handle_badges( event={'token': empty_token_result['token']}, context={})
def test_validate_bad_token(self): with self.assertRaises(HandlerBase.InvalidTokenError): handlers.handle_badges(event={'token': self.bad_token}, context={})
def test_validate_good_token(self): handlers.handle_badges(event={'token': self.good_empty_token}, context={})
def test_create_new_token(self): result = handlers.handle_badges(event={}, context={}) self.assertEqual(result['badges'], []) self.assertIn('token', result) self.assertIsNotNone(result['token']) handlers.handle_badges(event={'token': result['token']}, context={})