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={})