@@ -600,7 +600,6 @@ def _test_should_continue_without_redirect_for_two_factor_auth(
600600 }
601601
602602 def test_addon_developer_should_redirect_for_two_factor_auth (self ):
603- self .create_flag ('2fa-enforcement-for-developers-and-special-users' )
604603 self .user = user_factory ()
605604 # They have developed a theme, but also an extension, so they will need
606605 # 2FA.
@@ -609,37 +608,17 @@ def test_addon_developer_should_redirect_for_two_factor_auth(self):
609608 self ._test_should_redirect_for_two_factor_auth ()
610609
611610 def test_special_user_should_redirect_for_two_factor_auth (self ):
612- self .create_flag ('2fa-enforcement-for-developers-and-special-users' )
613611 self .user = user_factory ()
614612 # User isn't a developer but is part of a group.
615613 self .grant_permission (self .user , 'Some:Thing' )
616614 self ._test_should_redirect_for_two_factor_auth ()
617615
618- def test_user_should_redirect_for_two_factor_auth_flag_user_specific (self ):
619- self .user = user_factory ()
620- flag = self .create_flag (
621- '2fa-enforcement-for-developers-and-special-users' , everyone = False
622- )
623- flag .users .add (self .user )
624- # User isn't a developer but is part of a group.
625- self .grant_permission (self .user , 'Some:Thing' )
626- # The flag is enabled for that user.
627- self ._test_should_redirect_for_two_factor_auth ()
628-
629- # Others should be unaffected as the flag is only set for a specific
630- # user.
631- self .user = user_factory ()
632- self .grant_permission (self .user , 'Some:Thing' )
633- self ._test_should_continue_without_redirect_for_two_factor_auth ()
634-
635616 def test_theme_developer_should_not_redirect_for_two_factor_auth (self ):
636- self .create_flag ('2fa-enforcement-for-developers-and-special-users' )
637617 self .user = user_factory ()
638618 addon_factory (users = [self .user ], type = amo .ADDON_STATICTHEME )
639619 self ._test_should_continue_without_redirect_for_two_factor_auth ()
640620
641621 def test_addon_developer_already_using_two_factor_should_continue (self ):
642- self .create_flag ('2fa-enforcement-for-developers-and-special-users' )
643622 self .user = user_factory ()
644623 addon_factory (users = [self .user ])
645624 identity = {
@@ -652,7 +631,6 @@ def test_addon_developer_already_using_two_factor_should_continue(self):
652631 )
653632
654633 def test_2fa_enforced_already_using_two_factor_should_continue (self ):
655- self .create_flag ('2fa-enforcement-for-developers-and-special-users' )
656634 self .user = user_factory ()
657635 identity = {
658636 'uid' : '1234' ,
@@ -668,27 +646,10 @@ def test_2fa_enforced_already_using_two_factor_should_continue(self):
668646 assert 'enforce_2fa' not in self .request .session
669647
670648 def test_2fa_enforced_on_this_view_should_redirect_for_two_factor_auth (self ):
671- self .create_flag ('2fa-enforcement-for-developers-and-special-users' )
672649 self .user = user_factory ()
673650 self .request .session ['enforce_2fa' ] = True
674651 self ._test_should_redirect_for_two_factor_auth ()
675652
676- def test_waffle_flag_off_developer_without_2fa_should_continue (self ):
677- self .create_flag (
678- '2fa-enforcement-for-developers-and-special-users' , everyone = False
679- )
680- self .user = user_factory ()
681- addon_factory (users = [self .user ])
682- self ._test_should_continue_without_redirect_for_two_factor_auth ()
683-
684- def test_waffle_flag_off_enforced_2fa_should_have_no_effect (self ):
685- self .create_flag (
686- '2fa-enforcement-for-developers-and-special-users' , everyone = False
687- )
688- self .user = user_factory ()
689- self .request .session ['enforce_2fa' ] = True
690- self ._test_should_continue_without_redirect_for_two_factor_auth ()
691-
692653 @override_settings (FXA_CONFIG = {'default' : {'client_id' : '' }})
693654 def test_fake_fxa_auth (self ):
694655 self .user = user_factory ()
0 commit comments