def test_bad_version_raises(self):
obj = Builder(config,
autoconnect=False,
version_min=(98, 98, 98),
version_max=(99, 99, 99))
obj._get_client()
self.assertRaises(ESClientException, obj._check_version)
def test_multiple_hosts_raises(self):
local_conf = {
'elasticsearch': {
'client': {
'hosts': ['127.0.0.1', 'localhost']
}
}
}
local_conf['elasticsearch']['master_only'] = True
obj = Builder(local_conf, autoconnect=False)
obj._get_client()
self.assertRaises(ConfigurationError, obj._check_master)
def test_skip_version_check(self):
obj = Builder(config,
autoconnect=False,
version_min=(98, 98, 98),
version_max=(99, 99, 99))
obj.skip_version_test = True
obj._get_client()
self.assertIsNone(obj._check_version())
def test_http_auth_tuple(self):
u = 'username'
p = 'password'
obj = Builder({}, autoconnect=False)
obj.client_args[u] = u
obj.client_args[p] = p
obj._check_http_auth()
self.assertFalse(u in obj.client_args)
self.assertFalse(p in obj.client_args)
self.assertEqual((u, p), obj.client_args['http_auth'])
def test_mock_session_creds(self, mock_session):
session_instance = mock_session.return_value
credentials_instance = session_instance.get_credentials.return_value
credentials_instance.access_key = 'access_key'
credentials_instance.secret_key = 'secret_key'
credentials_instance.token = 'token'
obj = Builder({}, autoconnect=False)
obj.use_aws = True
obj.aws['aws_region'] = 'us_east'
obj._parse_aws()
self.assertTrue(obj.client_args['use_ssl'])
def test_client_info(self):
obj = Builder(config)
client = obj.client
expected = client.info()
self.assertEqual(expected, obj.test_connection())
def test_exit_if_not_master(self):
obj = Builder(config, autoconnect=False)
obj.master_only = True
obj.is_master = False
obj._get_client()
self.assertRaises(NotMaster, obj._check_master)
def test_no_credentials_raises(self):
obj = Builder({}, autoconnect=False)
obj.use_aws = True
obj.aws['aws_region'] = 'us_east'
self.assertRaises(NoCredentialsError, obj._parse_aws)
def test_no_region_raises(self):
obj = Builder({}, autoconnect=False)
obj.use_aws = True
self.assertRaises(MissingArgument, obj._parse_aws)
def test_certifi(self):
obj = Builder({}, autoconnect=False)
obj.client_args['use_ssl'] = True
obj._check_ssl()
self.assertTrue(obj.client_args['verify_certs'])
self.assertEqual(certifi.where(), obj.client_args['ca_certs'])
def test_pass_but_no_user(self):
obj = Builder({}, autoconnect=False)
obj.client_args['password'] = '******'
self.assertRaises(ConfigurationError, obj._check_http_auth)
def test_url_prefix_is_none(self):
obj = Builder({}, autoconnect=False)
obj.client_args['url_prefix'] = None
obj._fix_url_prefix()
self.assertEqual('', obj.client_args['url_prefix'])