def initialise_righteous(self):
config = SafeConfigParser()
config.read('righteous.config')
if not config.has_section('auth'):
raise Exception('Please create a righteous.config file with appropriate credentials')
self.auth = dict((key,config.get('auth', key)) for key in config.options('auth'))
self.server = dict((key,config.get('server-defaults', key)) for key in config.options('server-defaults'))
righteous.init(self.auth['username'], self.auth['password'], self.auth['account_id'], **self.server)
if not righteous.config.settings.cookies:
righteous.login()
def initialise(arguments):
verbose = arguments['--verbose']
config_file = arguments['--config']
if verbose:
logging.basicConfig(level=logging.DEBUG)
config = read_authentication(config_file or AUTH_FILE)
if not config:
# TODO: prompt for config and store
pass
username, password, account_id = tuple(config.get('auth', key)
for key in config.options('auth'))
server_parameters = dict(config.items('server-defaults'))
righteous.init(username, password, account_id, **server_parameters)
if righteous.login():
cache_authentication(username, password, account_id,
config_file or AUTH_FILE)
else:
puts_err(colored.red('Authentication failed'))
exit(2)
return verbose
def initialise(arguments):
verbose = arguments['--verbose']
config_file = arguments['--config']
if verbose:
logging.basicConfig(level=logging.DEBUG)
config = read_authentication(config_file or AUTH_FILE)
if not config:
error('No configuration found. Either create "%s" or specify a '
'configuration file with -c FILE' % (AUTH_FILE))
username, password, account_id = tuple(
config.get('auth', key) for key in config.options('auth')
)
server_parameters = dict(config.items('server-defaults'))
righteous.initialise(username, password, account_id, **server_parameters)
if righteous.login():
cache_authentication(
username, password, account_id, config_file or AUTH_FILE)
else:
error('Authentication failed')
return verbose
def test_login_with_credentials(self):
username, password, account_id = 'foo', 'bar', '123'
auth = base64.encodestring('%s:%s' % (username, password))[:-1]
self.response.status_code = 204
self.response.headers['set-cookie'] = 'cookie_value'
login_result = righteous.login(username, password, account_id)
assert login_result
self.request.assert_called_once_with('/login', headers={'Authorization': 'Basic %s' % auth})
def test_login_with_init_credentials(self):
username, password, account_id = 'user', 'pass', 'account_id'
auth = base64.b64encode(six.b('%s:%s' % (username, password)))[:-1]
righteous.init(username, password, account_id)
self.response.status_code = 204
self.response.headers['set-cookie'] = 'cookie_value'
login_result = righteous.login()
assert login_result
self.request.assert_called_once_with(
'/login', headers={'Authorization': 'Basic %s' % auth})
def test_login_failure(self):
assert not righteous.login('foo', 'bar', 'buzz')
def test_login(self):
self.assertTrue(righteous.login())
