test when order fetching fails with an account ID set

alexgolec · May 14, 2024 · 232dfdd · 232dfdd
1 parent 880aef7
commit 232dfdd
Showing 1 changed file with 36 additions and 1 deletion.
diff --git a/tests/scripts/orders_codegen_test.py b/tests/scripts/orders_codegen_test.py
@@ -332,7 +332,7 @@ def test_success_account_hash(
     @patch('schwab.scripts.orders_codegen.client_from_token_file')
     @patch('schwab.scripts.orders_codegen.construct_repeat_order')
     @patch('schwab.scripts.orders_codegen.code_for_builder')
-    def test_order_fetching_fails(
+    def test_order_fetching_fails_no_account_id(
             self,
             mock_code_for_builder,
             mock_construct_repeat_order,
@@ -356,6 +356,41 @@ def test_order_fetching_fails(
                 AnyStringWith('Returned HTTP status code 400'))
 
 
+    @no_duplicates
+    @patch('builtins.print')
+    @patch('schwab.scripts.orders_codegen.client_from_token_file')
+    @patch('schwab.scripts.orders_codegen.construct_repeat_order')
+    @patch('schwab.scripts.orders_codegen.code_for_builder')
+    def test_order_fetching_fails_account_id(
+            self,
+            mock_code_for_builder,
+            mock_construct_repeat_order,
+            mock_client_from_token_file,
+            mock_print):
+        self.add_arg('--token_file')
+        self.add_arg('filename.json')
+        self.add_arg('--api_key')
+        self.add_arg('api-key')
+        self.add_arg('--app_secret')
+        self.add_arg('app-secret')
+        self.add_arg('--account_id')
+        self.add_arg('123456')
+
+        mock_client = MagicMock()
+        mock_client_from_token_file.return_value = mock_client
+        mock_client.get_account_numbers.return_value = httpx.Response(
+                200,
+                json=[{
+                    'accountNumber': '123456',
+                    'hashValue': 'hash-value',
+                }])
+        mock_client.get_orders_for_account.return_value = httpx.Response(400)
+
+        self.assertEqual(self.main(), -1)
+
+        mock_print.assert_called_once_with(
+                AnyStringWith('Returned HTTP status code 400'))
+
 
 class ScriptInvocationTest(unittest.TestCase):