def take_action(self, parsed_args):
api = AtmosphereAPI(self.app_args.auth_token,
base_url=self.app_args.base_url,
timeout=self.app_args.api_server_timeout,
verify=self.app_args.verify_cert)
data = api.get_version()
if data.ok:
message = data.message
self.app.stdout.write('Atmosphere {} {} [Built: {}]\n'.format(
message['git_branch'], message['git_sha_abbrev'],
ts_to_isodate(message['commit_date'], include_time=True)))
def test_getting_version_when_response_is_ok(self):
api = AtmosphereAPI('token', base_url=self.mock_users_base_url)
response = api.get_version()
assert response.ok and response.message['git_branch'] == 'zesty-zapdos'
def test_getting_version_when_response_is_not_ok(self):
api = AtmosphereAPI('token', base_url=self.mock_users_bad_base_url)
response = api.get_version()
assert not response.ok