Fixed tests with Waffle switches.

This commit is contained in:
2013-04-04 22:55:16 +01:00
parent c79dc3b320
commit 42057c3433

View File

@@ -1,6 +1,7 @@
from django.test import TestCase
from django.core.urlresolvers import reverse
from stores.models import Chain, Store
from waffle import Switch
class StoreViewsTestCase(TestCase):
fixtures = ['test_stores']
@@ -9,6 +10,9 @@ class StoreViewsTestCase(TestCase):
self.store = Store.objects.get(pk=3)
self.chain = Chain.objects.get(pk=1)
# Enable claim support for the tests
Switch.objects.create(name='claim_support', active=True)
def test_map_index(self):
resp = self.client.get(reverse('map'))
self.assertEqual(resp.status_code, 200)
@@ -37,7 +41,7 @@ class StoreViewsTestCase(TestCase):
resp = self.client.get(reverse('store-claim', args=[self.store.slug]))
self.assertEqual(resp.status_code, 200)
resp = self.client.get(reverse('store-claim', args=['test-invalid']))
self.assertEqual(resp.status_code, 200)
self.assertEqual(resp.status_code, 404)
def test_store_update(self):
resp = self.client.get(reverse('store-update', args=[self.store.slug]))