Avoid failing with no-key in tests

Author: koen-serry

tests/test_document.py

@@ -12,7 +12,7 @@ class TestDocument(unittest.TestCase):
     ct = 1
 
     def setUp(self):
-        key = os.environ["TWIKEY_API_KEY"]
+        key = os.getenv("TWIKEY_API_KEY")
         if key is None:
             self.skipTest("No TWIKEY_API_KEY set")
 

tests/test_invoices.py

@@ -13,7 +13,7 @@ class TestInvoices(unittest.TestCase):
     _twikey = None
 
     def setUp(self):
-        key = os.environ["TWIKEY_API_KEY"]
+        key = os.getenv("TWIKEY_API_KEY")
         if key is None:
             self.skipTest("No TWIKEY_API_KEY set")
 

tests/test_paylinks.py

@@ -12,7 +12,7 @@ class TestPaylinks(unittest.TestCase):
 
     @unittest.skipIf("TWIKEY_API_KEY" not in os.environ, "No TWIKEY_API_KEY set")
     def setUp(self):
-        key = os.environ["TWIKEY_API_KEY"]
+        key = os.getenv("TWIKEY_API_KEY")
         base_url = "https://test.beta.twikey.com/api/creditor"
         if "TWIKEY_API_URL" in os.environ:
             base_url = os.environ["TWIKEY_API_URL"]

tests/test_refunds.py

@@ -14,7 +14,7 @@ class TestRefunds(unittest.TestCase):
 
     @unittest.skipIf("TWIKEY_API_KEY" not in os.environ, "No TWIKEY_API_KEY set")
     def setUp(self):
-        key = os.environ["TWIKEY_API_KEY"]
+        key = os.getenv("TWIKEY_API_KEY")
         base_url = "https://test.beta.twikey.com/api/creditor"
         if "TWIKEY_API_URL" in os.environ:
             base_url = os.environ["TWIKEY_API_URL"]

tests/test_transaction.py

@@ -15,7 +15,7 @@ class TestTransaction(unittest.TestCase):
 
     @unittest.skipIf("TWIKEY_API_KEY" not in os.environ, "No TWIKEY_API_KEY set")
     def setUp(self):
-        key = os.environ["TWIKEY_API_KEY"]
+        key = os.getenv("TWIKEY_API_KEY")
         base_url = "https://test.beta.twikey.com/api/creditor"
         if "TWIKEY_API_URL" in os.environ:
             base_url = os.environ["TWIKEY_API_URL"]