Skip to content

Commit 80f5b33

Browse files
authored
Add --testing option to waffle_flag mgmt command (#494)
1 parent 21d7fad commit 80f5b33

File tree

2 files changed

+9
-1
lines changed

2 files changed

+9
-1
lines changed

waffle/management/commands/waffle_flag.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -83,6 +83,13 @@ def add_arguments(self, parser: CommandParser) -> None:
8383
dest='rollout',
8484
default=False,
8585
help='Turn on rollout mode.')
86+
parser.add_argument(
87+
'--testing', '-t',
88+
action='store_true',
89+
dest='testing',
90+
default=False,
91+
help='Turn on testing mode, allowing the flag to be specified via '
92+
'a querystring parameter.')
8693
parser.add_argument(
8794
'--create',
8895
action='store_true',

waffle/tests/test_management.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -17,13 +17,14 @@ def test_create(self):
1717
name = 'test'
1818
percent = 20
1919
Group.objects.create(name='waffle_group')
20-
call_command('waffle_flag', name, percent=percent,
20+
call_command('waffle_flag', name, percent=percent, testing=True,
2121
superusers=True, staff=True, authenticated=True,
2222
rollout=True, create=True, group=['waffle_group'])
2323

2424
flag = get_waffle_flag_model().objects.get(name=name)
2525
self.assertEqual(flag.percent, percent)
2626
self.assertIsNone(flag.everyone)
27+
self.assertTrue(flag.testing)
2728
self.assertTrue(flag.superusers)
2829
self.assertTrue(flag.staff)
2930
self.assertTrue(flag.authenticated)

0 commit comments

Comments
 (0)