1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
|
from django.urls import reverse
from waffle import (
get_waffle_flag_model,
get_waffle_sample_model,
get_waffle_switch_model,
)
from waffle.tests.base import TestCase
class WaffleViewTests(TestCase):
def test_wafflejs(self):
response = self.client.get(reverse('wafflejs'))
self.assertEqual(200, response.status_code)
self.assertEqual('application/x-javascript', response['content-type'])
cache_control = [control.strip()
for control in response['cache-control'].split(',')]
self.assertIn('max-age=0', cache_control)
def test_waffle_status(self):
response = self.client.get(reverse('waffle_status'))
self.assertEqual(200, response.status_code)
self.assertEqual('application/json', response['content-type'])
cache_control = [control.strip()
for control in response['cache-control'].split(',')]
self.assertIn('max-age=0', cache_control)
def test_waffle_status_response(self):
get_waffle_flag_model().objects.create(name='test_flag_active', everyone=True)
get_waffle_flag_model().objects.create(name='test_flag_inactive', everyone=False)
get_waffle_switch_model().objects.create(name='test_switch_active', active=True)
get_waffle_switch_model().objects.create(name='test_switch_inactive', active=False)
get_waffle_sample_model().objects.create(name='test_sample_active', percent=100)
get_waffle_sample_model().objects.create(name='test_sample_inactive', percent=0)
response = self.client.get(reverse('waffle_status'))
self.assertEqual(200, response.status_code)
content = response.json()
assert 'test_flag_active' in content['flags'].keys()
assert content['flags']['test_flag_active']['is_active']
assert 'test_flag_inactive' in content['flags'].keys()
assert not content['flags']['test_flag_inactive']['is_active']
assert 'test_switch_active' in content['switches'].keys()
assert content['switches']['test_switch_active']['is_active']
assert 'test_switch_inactive' in content['switches'].keys()
assert not content['switches']['test_switch_inactive']['is_active']
assert 'test_sample_active' in content['samples'].keys()
assert content['samples']['test_sample_active']['is_active']
assert 'test_sample_inactive' in content['samples'].keys()
assert not content['samples']['test_sample_inactive']['is_active']
def test_flush_all_flags(self):
"""Test the 'FLAGS_ALL' list gets invalidated correctly."""
get_waffle_flag_model().objects.create(name='myflag1', everyone=True)
response = self.client.get(reverse('wafflejs'))
self.assertEqual(200, response.status_code)
assert ('myflag1', True) in response.context['flags']
get_waffle_flag_model().objects.create(name='myflag2', everyone=True)
response = self.client.get(reverse('wafflejs'))
self.assertEqual(200, response.status_code)
assert ('myflag2', True) in response.context['flags']
def test_flush_all_switches(self):
"""Test the 'SWITCHES_ALL' list gets invalidated correctly."""
switch = get_waffle_switch_model().objects.create(name='myswitch', active=True)
response = self.client.get(reverse('wafflejs'))
self.assertEqual(200, response.status_code)
assert ('myswitch', True) in response.context['switches']
switch.active = False
switch.save()
response = self.client.get(reverse('wafflejs'))
self.assertEqual(200, response.status_code)
assert ('myswitch', False) in response.context['switches']
def test_flush_all_samples(self):
"""Test the 'SAMPLES_ALL' list gets invalidated correctly."""
get_waffle_sample_model().objects.create(name='sample1', percent='100.0')
response = self.client.get(reverse('wafflejs'))
self.assertEqual(200, response.status_code)
assert ('sample1', True) in response.context['samples']
get_waffle_sample_model().objects.create(name='sample2', percent='100.0')
response = self.client.get(reverse('wafflejs'))
self.assertEqual(200, response.status_code)
assert ('sample2', True) in response.context['samples']
|
