class CredentialTests(unittest.TestCase):
credentials_file_path = os.path.join(PROJECT_DIR, "secrets", "credentials_file")
current_dir = os.path.dirname(os.path.realpath(__file__))
test_file = os.path.join(current_dir, "test_file")
def setUp(self):
"""
Instantiate GoogleAuth object
"""
self.auth = GoogleAuth()
open(self.test_file, "w+").close()
def tearDown(self):
"""
Remove auth object after tests
"""
del self.auth
os.remove(self.test_file)
def test_check_credentials_from_browser(self):
"""
Basic check if browser returns credentials
"""
# local_webserver_auth() returns credentials and saves them into file
self.credentials = self.auth.local_webserver_auth()
self.assertIsNotNone(self.credentials)
sleep(10)
def test_check_saved_credentials(self):
"""
Function to save credentials to file and check if it's not empty
"""
self.credentials = self.auth.local_webserver_auth()
self.auth.save_credentials_to_file(credentials_file=self.test_file)
with open(self.test_file) as f:
self.content = f.readlines()
self.assertIsNotNone(self.content)
sleep(10)
def test_credentials_with_regex(self):
"""
Function to check credentials are properly returned
"""
self.credentials = self.auth.local_webserver_auth()
self.assertIsInstance(self.credentials, OAuth2Credentials)
self.assertRegexpMatches(self.credentials.token_uri, "^http(?:s)*://\S+")
self.assertRegexpMatches(self.credentials.access_token, ".+")
self.assertRegexpMatches(self.credentials.client_id, ".+")
sleep(10)
def setUp(self):
"""
Instantiate GoogleAuth object
"""
self.auth = GoogleAuth()
open(self.test_file, "w+").close()