Fix test to support NoFeatureFlagFound.

rachelsanders · Jul 30, 2014 · 8a42958 · 8a42958
1 parent f0ee02a
commit 8a42958
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/test_edge_cases.py b/tests/test_edge_cases.py
@@ -36,4 +36,5 @@ def test_running_default_handler_on_app_that_was_never_set_up_returns_false(self
     test_app = Flask(__name__)
 
     with test_app.test_request_context("/"):
-      self.assertFalse(feature_flags.AppConfigFlagHandler("BOGUS_FEATURE_FLAG"))
+      self.assertRaises(feature_flags.NoFeatureFlagFound,
+                        feature_flags.AppConfigFlagHandler, "BOGUS_FEATURE_FLAG")