Fix for my fix

This commit is contained in:
2020-09-17 11:31:32 +01:00
parent dcc0e53062
commit b90be708d0

View File

@@ -1,12 +1,12 @@
import pytz
from reversion import revisions as reversion from reversion import revisions as reversion
from django.conf import settings from django.conf import settings
from django.core.exceptions import ValidationError from django.core.exceptions import ValidationError
from django.test import TestCase from django.test import TestCase
from django.urls import reverse from django.urls import reverse
from RIGS import models from RIGS import models
from assets import models as amodels
from versioning import versioning from versioning import versioning
from datetime import date, timedelta, datetime, time from datetime import date,timedelta, datetime, time
from decimal import * from decimal import *
from PyRIGS.tests.base import create_browser from PyRIGS.tests.base import create_browser
@@ -221,20 +221,20 @@ class TestVersioningViews(TestCase):
cls.events[1].description = "A test description" cls.events[1].description = "A test description"
cls.events[1].save() cls.events[1].save()
working = models.AssetStatus.objects.create(name="Working", should_show=True) working = amodels.AssetStatus.objects.create(name="Working", should_show=True)
broken = models.AssetStatus.objects.create(name="Broken", should_show=False) broken = amodels.AssetStatus.objects.create(name="Broken", should_show=False)
general = models.AssetCategory.objects.create(name="General") general = amodels.AssetCategory.objects.create(name="General")
lighting = models.AssetCategory.objects.create(name="Lighting") lighting = amodels.AssetCategory.objects.create(name="Lighting")
cls.assets = {} cls.assets = {}
with reversion.create_revision(): with reversion.create_revision():
reversion.set_user(cls.profile) reversion.set_user(cls.profile)
cls.assets[1] = models.Asset.objects.create(asset_id="1991", description="Spaceflower", status=broken, category=lighting, date_acquired=datetime.date(1991, 12, 26)) cls.assets[1] = amodels.Asset.objects.create(asset_id="1991", description="Spaceflower", status=broken, category=lighting, date_acquired=date.today())
with reversion.create_revision(): with reversion.create_revision():
reversion.set_user(cls.profile) reversion.set_user(cls.profile)
cls.assets[2] = models.Asset.objects.create(asset_id="0001", description="Virgil", status=working, category=lighting, date_acquired=datetime.date(2015, 1, 1)) cls.assets[2] = amodels.Asset.objects.create(asset_id="0001", description="Virgil", status=working, category=lighting, date_acquired=date.today())
with reversion.create_revision(): with reversion.create_revision():
reversion.set_user(cls.profile) reversion.set_user(cls.profile)
@@ -246,12 +246,17 @@ class TestVersioningViews(TestCase):
self.profile.save() self.profile.save()
self.assertTrue(self.client.login(username=self.profile.username, password='testuser')) self.assertTrue(self.client.login(username=self.profile.username, password='testuser'))
def test_event_history_loads_successfully(self): def test_history_loads_successfully(self):
request_url = reverse('event_history', kwargs={'pk': self.events[1].pk}) request_url = reverse('event_history', kwargs={'pk': self.events[1].pk})
response = self.client.get(request_url, follow=True) response = self.client.get(request_url, follow=True)
self.assertEqual(response.status_code, 200) self.assertEqual(response.status_code, 200)
request_url = reverse('asset_history', kwargs={'pk': self.assets[1].asset_id})
response = self.client.get(request_url, follow=True)
self.assertEqual(response.status_code, 200)
def test_activity_feed_loads_successfully(self): def test_activity_feed_loads_successfully(self):
request_url = reverse('activity_feed') request_url = reverse('activity_feed')
@@ -269,12 +274,6 @@ class TestVersioningViews(TestCase):
response = self.client.get(request_url, follow=True) response = self.client.get(request_url, follow=True)
self.assertEqual(response.status_code, 200) self.assertEqual(response.status_code, 200)
def test_asset_history_loads_successfully(self):
request_url = reverse('asset_history', kwargs={'pk': self.assets[1].asset_id})
response = self.client.get(request_url, follow=True)
self.assertEqual(response.status_code, 200)
# Some edge cases that have caused server errors in the past # Some edge cases that have caused server errors in the past
def test_deleted_event(self): def test_deleted_event(self):
request_url = reverse('activity_feed') request_url = reverse('activity_feed')