21
22
from provisioningserver.testing.testcase import PservTestCase
24
def make_credentials():
25
"""Produce a tuple of API credentials."""
27
factory.make_name('consumer-key'),
28
factory.make_name('resource-token'),
29
factory.make_name('resource-secret'),
33
25
class TestAuth(PservTestCase):
35
27
def test_record_api_credentials_records_credentials_string(self):
36
creds_string = convert_tuple_to_string(make_credentials())
28
creds_string = convert_tuple_to_string(make_api_credentials())
37
29
auth.record_api_credentials(creds_string)
39
31
creds_string, cache.cache.get(auth.API_CREDENTIALS_CACHE_KEY))
41
33
def test_get_recorded_api_credentials_returns_credentials_as_tuple(self):
42
creds = make_credentials()
34
creds = make_api_credentials()
43
35
auth.record_api_credentials(convert_tuple_to_string(creds))
44
36
self.assertEqual(creds, auth.get_recorded_api_credentials())
46
38
def test_get_recorded_api_credentials_returns_None_without_creds(self):
47
39
self.assertIsNone(auth.get_recorded_api_credentials())
49
def test_get_recorded_nodegroup_name_vs_record_nodegroup_name(self):
50
nodegroup_name = factory.make_name('nodegroup')
51
auth.record_nodegroup_name(nodegroup_name)
52
self.assertEqual(nodegroup_name, auth.get_recorded_nodegroup_name())
41
def test_get_recorded_nodegroup_uuid_vs_record_nodegroup_uuid(self):
42
nodegroup_uuid = factory.make_name('nodegroupuuid')
43
auth.record_nodegroup_uuid(nodegroup_uuid)
44
self.assertEqual(nodegroup_uuid, auth.get_recorded_nodegroup_uuid())