def test_calls_view_if_flag_active(self): """Test that the wrapped view is caleld if the feature flag is active.""" FeatureFlagFactory(code='test-feature-flag', is_active=True) mock = Mock() feature_flagged_view('test-feature-flag')(mock)() mock.assert_called_once()
def test_raises_404_is_flag_inactive(self): """Test that an Http404 is raised if the feature flag is inactive.""" FeatureFlagFactory(code='test-feature-flag', is_active=False) mock = Mock() with pytest.raises(Http404): feature_flagged_view('test-feature-flag')(mock)() mock.assert_not_called()
def test_raises_404_if_flag_does_not_exist(self): """Test that an Http404 is raised if the feature flag does not exist.""" mock = Mock() with pytest.raises(Http404): feature_flagged_view('test-feature-flag')(mock)() mock.assert_not_called()