Mock state_machine

fix `AttributeError: 'NoneType' object has no attribute 'reporter'`
melexis · Sep 10, 2024 · 7fae11a · 7fae11a
1 parent 41dc79c
commit 7fae11a
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tests/test_process_options.py b/tests/test_process_options.py
@@ -49,7 +49,7 @@ def test_with_empty_options(self):
                                        content_offset=0,
                                        block_text='',
                                        state=None,
-                                       state_machine=None)
+                                       state_machine=Mock())
 
         self.assertNotIn('commalist_arg', fake_node)
         self.assertNotIn('string_arg', fake_node)
@@ -77,7 +77,7 @@ def test_with_existing_options(self):
                                        content_offset=0,
                                        block_text='',
                                        state=None,
-                                       state_machine=None)
+                                       state_machine=Mock())
 
         self.assertNotIn('commalist_arg', fake_node)
         self.assertNotIn('string_arg', fake_node)
@@ -102,7 +102,7 @@ def test_with_just_one_existing_option(self):
                                        content_offset=0,
                                        block_text='',
                                        state=None,
-                                       state_machine=None)
+                                       state_machine=Mock())
 
         self.assertNotIn('commalist_arg', fake_node)
         self.assertNotIn('string_arg', fake_node)