Fixes for Travis Tests

Signed-off-by: Geoffrey Wiseman <geoffrey.wiseman@codiform.com>
geoffreywiseman · Oct 2, 2020 · 944094f · 944094f
1 parent 8ac6ce8
commit 944094f
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/tests/test_commands.py b/tests/test_commands.py
@@ -20,20 +20,22 @@ def region_fixture():
     region = 'ca-central-1'
     os.environ[AWS_DEFAULT_REGION] = region
     yield region
-    del os.environ[AWS_DEFAULT_REGION]
+    if AWS_DEFAULT_REGION in os.environ:
+        del os.environ[AWS_DEFAULT_REGION]
 
 
 @pytest.fixture(name='security_group', scope='function')
 def security_group_fixture():
-    mock_ec2().start()
+    mock = mock_ec2()
+    mock.start()
     ec2 = boto3.resource('ec2')
     sg = ec2.create_security_group(
         Description='Security Group for SSH Whitelisting',
         GroupName='SSH Whitelist',
         VpcId='vpc-123'
     )
     yield sg
-    mock_ec2().stop()
+    mock.stop()
 
 
 @patch('sys.stdout', new_callable=io.StringIO)