refactor(tests): dedupe mock input yes/no

Author: thekaveman

tests/commands/user/test_alumni.py

@@ -11,17 +11,13 @@ def mock_commands_reset(mock_commands_reset):
 
 
 @pytest.fixture
-def mock_input_yes(mock_input):
-    fix = mock_input(MODULE)
-    fix.return_value = "y"
-    return fix
+def mock_input_yes(mock_input_yes):
+    return mock_input_yes(MODULE)
 
 
 @pytest.fixture
-def mock_input_no(mock_input):
-    fix = mock_input(MODULE)
-    fix.return_value = "n"
-    return fix
+def mock_input_no(mock_input_no):
+    return mock_input_no(MODULE)
 
 
 @pytest.fixture

tests/commands/user/test_delete.py

@@ -5,17 +5,13 @@
 
 
 @pytest.fixture
-def mock_input_yes(mock_input):
-    fix = mock_input(MODULE)
-    fix.return_value = "y"
-    return fix
+def mock_input_yes(mock_input_yes):
+    return mock_input_yes(MODULE)
 
 
 @pytest.fixture
-def mock_input_no(mock_input):
-    fix = mock_input(MODULE)
-    fix.return_value = "n"
-    return fix
+def mock_input_no(mock_input_no):
+    return mock_input_no(MODULE)
 
 
 @pytest.fixture

tests/commands/user/test_offboard.py

@@ -5,17 +5,13 @@
 
 
 @pytest.fixture
-def mock_input_yes(mock_input):
-    fix = mock_input(MODULE)
-    fix.return_value = "y"
-    return fix
+def mock_input_yes(mock_input_yes):
+    return mock_input_yes(MODULE)
 
 
 @pytest.fixture
-def mock_input_no(mock_input):
-    fix = mock_input(MODULE)
-    fix.return_value = "n"
-    return fix
+def mock_input_no(mock_input_no):
+    return mock_input_no(MODULE)
 
 
 @pytest.fixture

tests/commands/user/test_reset.py

@@ -6,17 +6,13 @@
 
 
 @pytest.fixture
-def mock_input_yes(mock_input):
-    fix = mock_input(MODULE)
-    fix.return_value = "y"
-    return fix
+def mock_input_yes(mock_input_yes):
+    return mock_input_yes(MODULE)
 
 
 @pytest.fixture
-def mock_input_no(mock_input):
-    fix = mock_input(MODULE)
-    fix.return_value = "n"
-    return fix
+def mock_input_no(mock_input_no):
+    return mock_input_no(MODULE)
 
 
 @pytest.fixture

tests/commands/user/test_signout.py

@@ -5,17 +5,13 @@
 
 
 @pytest.fixture
-def mock_input_yes(mock_input):
-    fix = mock_input(MODULE)
-    fix.return_value = "y"
-    return fix
+def mock_input_yes(mock_input_yes):
+    return mock_input_yes(MODULE)
 
 
 @pytest.fixture
-def mock_input_no(mock_input):
-    fix = mock_input(MODULE)
-    fix.return_value = "n"
-    return fix
+def mock_input_no(mock_input_no):
+    return mock_input_no(MODULE)
 
 
 @pytest.fixture

tests/conftest.py

@@ -36,6 +36,30 @@ def mock_input(mock_module_name):
     return mock_module_name("input")
 
 
+@pytest.fixture
+def mock_input_yes(mock_input):
+    """Fixture returns a function that patches the input return value to Yes in a give module."""
+
+    def _mock_input_yes(module):
+        fix = mock_input(module)
+        fix.return_value = "y"
+        return fix
+
+    return _mock_input_yes
+
+
+@pytest.fixture
+def mock_input_no(mock_input):
+    """Fixture returns a function that patches the input return value to No in a give module."""
+
+    def _mock_input_no(module):
+        fix = mock_input(module)
+        fix.return_value = "n"
+        return fix
+
+    return _mock_input_no
+
+
 @click.command
 def dummy_command(**kwargs):
     return RESULT_SUCCESS