Improve custom exception example

quiqueporta · May 24, 2024 · 476c5a0 · 476c5a0
1 parent 2ef7873
commit 476c5a0
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/README.md b/README.md
@@ -156,7 +156,7 @@ class Point(ValueObject):
     x: int
     y: int
 
-    @invariant(MyException)
+    @invariant(exception_type=MyException)
     def inside_first_quadrant(self):
         return self.x > 0 and self.y > 0, "You must be inside the first quadrant"
 

diff --git a/specs/value_object_spec.py b/specs/value_object_spec.py
@@ -265,7 +265,7 @@ class Foo(ValueObject):
 
                 @invariant(exception_type=MyCustomException)
                 def bar(self):
-                    return False
+                    return self.any == "foo"
 
             expect(lambda: Foo("buzz")).to(raise_error(MyCustomException))
 
@@ -279,7 +279,7 @@ class Foo(ValueObject):
 
                 @invariant(exception_type=MyCustomException)
                 def bar(self):
-                    return False, "This is a custom message"
+                    return self.any == "foo", "This is a custom message"
 
             expect(lambda: Foo("buzz")).to(
                 raise_error(MyCustomException, "This is a custom message")