def save(self, *args, **kwargs):
kwargs = {
'created_by':
self.created_by,
'total_cows':
ReportStats.get_total_cows(self.client, self.year, self.month),
'aged_cows':
ReportStats.get_aged_cows(self.client, self.year, self.month),
'pregnant_cows':
ReportStats.get_pregnant_cows(self.client, self.year, self.month),
'ill_cows':
ReportStats.get_ill_cows(self.client, self.year, self.month),
'injured_cows':
ReportStats.get_injured_cows(self.client, self.year, self.month),
'gallons_milk':
ReportStats.get_gallons_milk(self.client, self.year, self.month)
}
try:
self.instance = Monthly.objects.get(client=self.client,
year=self.year,
month=self.month)
Monthly.objects.filter(pk=self.instance.pk).update(**kwargs)
return
except Monthly.DoesNotExist:
super(Monthly, self).save()
view_name = 'summary:monthly-client-year-month'
if self.month < 10:
month = '0{}'.format(self.month)
kwargs.update({
'client':
self.client,
'link':
django_reverse(view_name,
kwargs={
'pk': self.pk,
'year': self.year,
'month': month
})
})
else:
kwargs.update({
'client':
self.client,
'link':
django_reverse(view_name,
kwargs={
'pk': self.pk,
'year': self.year,
'month': self.month
})
})
Monthly.objects.filter(pk=self.pk).update(**kwargs)
return
def save(self, *args, **kwargs):
super(Cow, self).save(*args, **kwargs)
kwargs = {
'link': django_reverse('assets:cow-detail', kwargs={'pk': self.pk})
}
Cow.objects.filter(pk=self.pk).update(**kwargs)
return
def save(self, *args, **kwargs):
kwargs = {
'created_by': self.created_by,
'total_cows': ReportStats.get_total_cows(self.client, self.year),
'aged_cows': ReportStats.get_aged_cows(self.client, self.year),
'pregnant_cows':
ReportStats.get_pregnant_cows(self.client, self.year),
'ill_cows': ReportStats.get_ill_cows(self.client, self.year),
'injured_cows':
ReportStats.get_injured_cows(self.client, self.year),
'gallons_milk':
ReportStats.get_gallons_milk(self.client, self.year)
}
try:
self.instance = Annual.objects.get(client=self.client,
year=self.year)
Annual.objects.filter(pk=self.instance.pk).update(**kwargs)
return
except Annual.DoesNotExist:
super(Annual, self).save()
view_name = 'summary:monthly-client-year'
kwargs.update({
'client':
self.client,
'link':
django_reverse(view_name,
kwargs={
'pk': self.pk,
'year': self.year
})
})
Annual.objects.filter(pk=self.pk).update(**kwargs)
return
def save(self, *args, **kwargs):
super(Exercise, self).save(*args, **kwargs)
kwargs = {
'link':
django_reverse('assets:exercise-detail', kwargs={'pk': self.pk})
}
Exercise.objects.filter(pk=self.pk).update(**kwargs)
return
def test_02_annual_client_year(self):
expected = '/summary/api/annual/client/1/year/2015/'
self.assertEqual(
expected,
django_reverse('summary:annual-client-year',
kwargs={
'pk': 1,
'year': 2015
}))
def save(self, *args, **kwargs):
super(HealthRecord, self).save(*args, **kwargs)
kwargs = {
'link':
django_reverse('assets:healthrecord-detail',
kwargs={'pk': self.pk})
}
HealthRecord.objects.filter(pk=self.pk).update(**kwargs)
return
def test_01_monthly_client_year(self):
expected = '/summary/api/monthly/client/2/year/2018/'
self.assertEqual(
expected,
django_reverse('summary:monthly-client-year',
kwargs={
'pk': 2,
'year': 2018
}))
def setUp(self):
self.user = User.objects.get(username=TestData.get_random_username())
self.data = {
'created_by': self.user.username,
'year': TestTime.get_random_year(),
'month': TestTime.get_random_month()
}
self.factory = APIRequestFactory(enforce_csrf_checks=True)
self.url = django_reverse('summary:monthly-client-year-month',
kwargs={
'pk': 1,
'year': self.data['year'],
'month': self.data['month']
})
def setUp(self):
self.client = Client.objects.get(name=TestData.get_random_client())
self.user = User.objects.get(username=TestData.get_random_username())
self.data = {
'client': self.client.name,
'created_by': self.user.username,
'year': TestTime.get_random_year()
}
self.pk = 1
self.factory = APIRequestFactory(enforce_csrf_checks=True)
self.url = django_reverse('summary:annual-client-year',
kwargs={
'pk': self.pk,
'year': self.data['year']
})
def test_01_annual_client(self):
expected = '/summary/api/annual/client/1/'
self.assertEqual(
expected, django_reverse('summary:annual-client', kwargs={'pk':
1}))
def reverse(name):
return django_reverse(f"api:v1:{name}", request=request)
def setUp(self):
self.factory = APIRequestFactory(enforce_csrf_checks=True)
self.request = None
self.url = django_reverse('assets:asset-index')
self.user = User.objects.get(username=TestData.get_random_user())
def test_01_cow_list(self):
expected = '/assets/api/cows/'
self.assertEqual(expected, django_reverse('assets:cow-list'))
def test_01_seed_list(self):
expected = '/assets/api/seeds/'
self.assertEqual(expected, django_reverse('assets:seed-list'))
def test_02_exercise_detail(self):
expected = '/assets/api/exercises/1/'
self.assertEqual(expected,
django_reverse('assets:exercise-detail', args=(1, )))
def test_01_exercise_list(self):
expected = '/assets/api/exercises/'
self.assertEqual(expected, django_reverse('assets:exercise-list'))
def test_01_event_list(self):
expected = '/assets/api/events/'
self.assertEqual(expected, django_reverse('assets:event-list'))
def test_04_cow_detail(self):
expected = '/assets/api/cows/1/'
self.assertEqual(expected,
django_reverse('assets:cow-detail', args=(1, )))
def test_03_cow_list_year(self):
expected = '/assets/api/cows/year/2018/'
self.assertEqual(
expected, django_reverse('assets:cow-list-year', args=('2018', )))
def test_02_cow_list_month(self):
expected = '/assets/api/cows/year/2018/month/03/'
self.assertEqual(
expected,
django_reverse('assets:cow-list-month', args=('2018', '03')))
def test_01_index(self):
expected = '/assets/'
self.assertEqual(expected, django_reverse('assets:asset-index'))
def test_02_milk_detail(self):
expected = '/assets/api/milk/1/'
self.assertEqual(expected,
django_reverse('assets:milk-detail', args=(1, )))
def test_01_healthrecord_list(self):
expected = '/assets/api/healthrecords/'
self.assertEqual(expected, django_reverse('assets:healthrecord-list'))
def test_02_seed_detail(self):
expected = '/assets/api/seeds/1/'
self.assertEqual(expected,
django_reverse('assets:seed-detail', args=(1, )))
def test_02_healthrecord_detail(self):
expected = '/assets/api/healthrecords/1/'
self.assertEqual(
expected, django_reverse('assets:healthrecord-detail', args=(1, )))
def test_01_milk_list(self):
expected = '/assets/api/milk/'
self.assertEqual(expected, django_reverse('assets:milk-list'))
