def test_get_fizzbuzz(self):
"""
Ensure we can get a fizzbuzz object.
"""
message = 'Random message: ' + str(randint(0,99999))
fizzbuzz1 = FizzBuzz(message=message)
fizzbuzz1.save()
response = self.client.get('/fizzbuzz/' + str(fizzbuzz1.fizzbuzz_id) + '/')
self.assertEqual(response.status_code, status.HTTP_200_OK)
self.assertEqual(response.data['fizzbuzz_id'], fizzbuzz1.fizzbuzz_id)
self.assertEqual(response.data['message'], fizzbuzz1.message)
def test_list_fizzbuzz(self):
"""
Ensure we can list fizzbuzz objects.
"""
message = 'Random message: ' + str(randint(0,99999))
fizzbuzz1 = FizzBuzz(message=message)
fizzbuzz1.save()
message = 'Random message: ' + str(randint(0,99999))
fizzbuzz2 = FizzBuzz(message=message)
fizzbuzz2.save();
response = self.client.get('/fizzbuzz/')
self.assertEqual(response.status_code, status.HTTP_200_OK)
self.assertEqual(response.data[0].get('fizzbuzz_id'), fizzbuzz1.fizzbuzz_id)
self.assertEqual(response.data[0].get('message'), fizzbuzz1.message)
self.assertEqual(response.data[1].get('fizzbuzz_id'), fizzbuzz2.fizzbuzz_id)
self.assertEqual(response.data[1].get('message'), fizzbuzz2.message)
