fixes import and adds invocation test

alexgolec · May 14, 2024 · af0138f · af0138f
1 parent 85241ed
commit af0138f
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 1 deletion.
diff --git a/bin/schwab-order-codegen.py b/bin/schwab-order-codegen.py
@@ -1,5 +1,5 @@
 #!/usr/bin/env python
-from tda.scripts.orders_codegen import latest_order_main
+from schwab.scripts.orders_codegen import latest_order_main
 
 if __name__ == '__main__':
     import sys

diff --git a/tests/scripts/orders_codegen_test.py b/tests/scripts/orders_codegen_test.py
@@ -1,3 +1,4 @@
+import subprocess
 import unittest
 from unittest.mock import call, MagicMock, patch
 
@@ -203,3 +204,14 @@ def test_account_id_error(
         mock_construct_repeat_order.assert_not_called
         mock_print.assert_called_once_with(
                 AnyStringWith('TDA returned error: "invalid"'))
+
+
+class ScriptInvocationTest(unittest.TestCase):
+
+    def test_get_help(self):
+        res = subprocess.run(
+                'schwab-order-codegen.py --help',
+                shell=True, check=True, text=True, capture_output=True)
+        print(res.stdout)
+        self.assertIn('Utilities for generating code from historical orders',
+                      res.stdout)