production test fix

TexasCoding · Jun 25, 2024 · 0ab9c45 · 0ab9c45
1 parent 16819e1
commit 0ab9c45
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/.github/workflows/test-package.yaml b/.github/workflows/test-package.yaml
@@ -11,6 +11,7 @@ env:
   SLACK_ACCESS_TOKEN: ${{ secrets.SLACK_ACCESS_TOKEN }}
   SLACK_USERNAME: ${{ secrets.SLACK_USERNAME }}
   OPENAI_API_KEY: ${{ secrets.OPENAI_API_KEY }}
+  PRODUCTION: ${{ secrets.PRODUCTION }}
 
 jobs:
   build:

diff --git a/tests/test_global_functions.py b/tests/test_global_functions.py
@@ -14,8 +14,8 @@ def test_send_position_messages_sell(capsys):
     ]
     pos_type = "sell"
     result = send_position_messages(positions, pos_type)
-    printed_message = capsys.readouterr().out
     if not production:
+        printed_message = capsys.readouterr().out
         assert (
             printed_message
             == """Message: Successfully sold the following positions:
@@ -32,8 +32,8 @@ def test_send_position_messages_buy(capsys):
     ]
     pos_type = "buy"
     result = send_position_messages(positions, pos_type)
-    printed_message = capsys.readouterr().out
     if not production:
+        printed_message = capsys.readouterr().out
         assert (
             printed_message
             == """Message: Successfully bought the following positions:
@@ -50,8 +50,8 @@ def test_send_position_messages_liquidate(capsys):
     ]
     pos_type = "liquidate"
     result = send_position_messages(positions, pos_type)
-    printed_message = capsys.readouterr().out
     if not production:
+        printed_message = capsys.readouterr().out
         assert (
             printed_message
             == """Message: Successfully liquidated the following positions: