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)