test env var fix

Author: ryan-williams

tests/test_start.py

@@ -4,6 +4,7 @@
 from unittest.mock import call, patch, mock_open, Mock
 from ec2_gha.start import StartAWS
 from botocore.exceptions import WaiterError, ClientError
+import os
 
 
 @pytest.fixture(scope="function")
@@ -114,6 +115,12 @@ def complete_params():
     yield params
 
 
+@patch.dict('os.environ', {
+    'GITHUB_REPOSITORY': 'Open-Athena/ec2-gha',
+    'GITHUB_WORKFLOW': 'CI',
+    'GITHUB_SERVER_URL': 'https://github.com',
+    'GITHUB_RUN_ID': '16725250800'
+})
 def test_build_aws_params(complete_params):
     user_data_params = {
         "token": "test",
@@ -148,6 +155,9 @@ def test_build_aws_params(complete_params):
             "Tags": [
                 {"Key": "Name", "Value": "test"},
                 {"Key": "Owner", "Value": "test"},
+                {"Key": "repository", "Value": "Open-Athena/ec2-gha"},
+                {"Key": "workflow", "Value": "CI"},
+                {"Key": "gha_url", "Value": "https://github.com/Open-Athena/ec2-gha/actions/runs/16725250800"},
             ],
         }
     ]