def test_purge_api_key():
with fake_creds():
assert read_cred(API_TOKEN_KEY) == 'fake_api_key'
purge_api_key()
with pytest.raises(KeyError):
read_cred(API_TOKEN_KEY)
def test_purge_api_key(self):
with fake_creds('.jovian', 'credentials.json'):
assert read_cred(API_TOKEN_KEY) == 'fake_api_key'
purge_api_key()
with self.assertRaises(KeyError):
read_cred(API_TOKEN_KEY)
def test_write_guest_key():
with fake_creds():
write_guest_key("fake_guest_key")
from jovian.utils.credentials import GUEST_TOKEN_KEY
assert read_cred(GUEST_TOKEN_KEY) == "fake_guest_key"
def test_write_api_key():
with fake_creds():
write_api_key("fake_api_key")
from jovian.utils.credentials import API_TOKEN_KEY
assert read_cred(API_TOKEN_KEY) == "fake_api_key"
def test_write_webapp_url():
with fake_creds():
write_webapp_url("http://fake-webapp-url.ai/")
from jovian.utils.credentials import WEBAPP_URL_KEY
assert read_cred(WEBAPP_URL_KEY) == "http://fake-webapp-url.ai/"
def test_write_org_id():
with fake_creds():
write_org_id("fake_org_id")
from jovian.utils.credentials import ORG_ID_KEY
assert read_cred(ORG_ID_KEY) == "fake_org_id"
def test_write_api_url():
with fake_creds():
write_api_url("https://fake_api.jovian.ai/")
from jovian.utils.credentials import API_URL_KEY
assert read_cred(API_URL_KEY) == "https://fake_api.jovian.ai/"
def test_read_cred_with_default():
with fake_creds():
assert read_cred('FAKE_ID', 'default_value') == 'default_value'
def test_read_cred_no_default():
with fake_creds():
assert read_cred('ORG_ID') == 'staging'
def test_write_org_id():
with fake_creds('.jovian-write-api-url', 'credentials.json'):
write_org_id("fake_org_id")
from jovian.utils.credentials import ORG_ID_KEY
assert read_cred(ORG_ID_KEY) == "fake_org_id"
def test_read_cred_with_default():
with fake_creds('.jovian', 'credentials.json'):
assert read_cred('FAKE_ID', 'default_value') == 'default_value'
def test_read_cred_no_default():
with fake_creds('.jovian', 'credentials.json'):
assert read_cred('ORG_ID') == 'staging'
