Merge pull request SynapseFI#48 from synapsepay/dev2

stevula · web-flow · commit 17cf984871d0 · 2017-05-04T18:10:52.000-07:00
Update test prompt for client keys and add relative path for test physical doc file
diff --git a/synapse_pay_rest/tests/fixtures/client.py b/synapse_pay_rest/tests/fixtures/client.py
@@ -1,10 +1,13 @@
 import os
 from synapse_pay_rest.client import Client
 
-# CLIENT_ID = os.environ['TEST_CLIENT_ID']
-# CLIENT_SECRET = os.environ['TEST_CLIENT_SECRET']
-CLIENT_ID = input("Please enter client ID")
-CLIENT_SECRET = input("Please enter client secret")
+try:
+  CLIENT_ID = os.environ['TEST_CLIENT_ID']
+  CLIENT_SECRET = os.environ['TEST_CLIENT_SECRET']
+except KeyError:
+  CLIENT_ID = input("Please enter client ID: ")
+  CLIENT_SECRET = input("Please enter client secret: ")
+
 FINGERPRINT = 'test_fp'
 IP_ADDRESS = '127.0.0.1'
 
diff --git a/synapse_pay_rest/tests/models/document_tests.py b/synapse_pay_rest/tests/models/document_tests.py
@@ -8,6 +8,9 @@
 from synapse_pay_rest.models import SocialDocument
 from synapse_pay_rest.models import VirtualDocument
 from synapse_pay_rest.models import Question
+import os
+dir = os.path.dirname(__file__)
+filename = os.path.join(dir, '../fixtures/test.png')
 
 
 class DocumentTestCases(unittest.TestCase):
@@ -36,7 +39,7 @@ def setUp(self):
 
     def test_physical_document_with_bytestream(self):
         type = 'GOVT_ID'
-        file_path = input("Input file test file full path")
+        file_path = filename
         byte_stream = open(file_path, 'rb').read()
         doc = self.base_document.add_physical_document(type=type,
                                                        byte_stream=byte_stream,
@@ -45,7 +48,7 @@ def test_physical_document_with_bytestream(self):
 
     def test_physical_document_with_file_path(self):
         type = 'GOVT_ID'
-        file_path = input("Input file test file full path")
+        file_path = filename
         doc = self.base_document.add_physical_document(type=type,
                                                        file_path=file_path)
         self.assertIsInstance(doc, PhysicalDocument)
