def get_auth(token_details):
if token_details['auth_type'] == 'api_key':
return SoftLayer.BasicAuthentication(token_details['username'],
token_details['api_key'])
elif token_details['auth_type'] == 'token':
return SoftLayer.TokenAuthentication(token_details['user_id'],
token_details['api_key'])
return None
def setUp(self): # NOQA
self.env = environment.Environment()
# Create a crazy mockable, fixture client
auth = SoftLayer.BasicAuthentication('default-user', 'default-key')
self.mocks = MockableTransport(SoftLayer.FixtureTransport())
self.transport = SoftLayer.TimingTransport(self.mocks)
self.client = SoftLayer.Client(transport=self.transport,
auth=auth,
timeout=10,
endpoint_url='default-endpoint-url')
self.env.client = self.client
return self.set_up()
