Add test for comopnent with custom exception in init

tbrlpld · Feb 20, 2024 · b63b9e5 · b63b9e5
1 parent 26b507f
commit b63b9e5
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 0 deletions.
diff --git a/laces/test/example/components.py b/laces/test/example/components.py
@@ -228,3 +228,13 @@ def get_context_data(
 class ServableIntAdderComponent(Component):
     def __init__(self, number: int) -> None:
         self.number = 0 + number
+
+
+class CustomException(Exception):
+    pass
+
+
+@register_servable("with-custom-exception-init")
+class ServableWithCustomExceptionInitializerComponent(Component):
+    def __init__(self) -> None:
+        raise CustomException
diff --git a/laces/test/tests/test_views.py b/laces/test/tests/test_views.py
@@ -164,3 +164,8 @@ def test_get_component_with_non_string_argument(self) -> None:
         response = self.client.get("/components/int-adder/?number=2")
 
         self.assertEqual(response.status_code, HTTPStatus.BAD_REQUEST)
+
+    def test_get_component_with_custom_exception(self) -> None:
+        response = self.client.get("/components/with-custom-exception-init/")
+
+        self.assertEqual(response.status_code, HTTPStatus.BAD_REQUEST)