def setUp(self): self.rpc_addresses = '' if sentry_site.get_sentry_server_rpc_addresses() is not None: self.rpc_addresses = ','.join(sentry_site.get_sentry_server_rpc_addresses()) self.rpc_port = sentry_site.get_sentry_server_rpc_port() or '8038' self.tmpdir = tempfile.mkdtemp() self.resets = [ SENTRY_CONF_DIR.set_for_testing(self.tmpdir), ] clear_api_cache() clear_api2_cache()
def setUp(self): self.rpc_addresses = '' if sentry_site.get_sentry_server_rpc_addresses() is not None: self.rpc_addresses = ','.join( sentry_site.get_sentry_server_rpc_addresses()) self.tmpdir = tempfile.mkdtemp() self.resets = [ SENTRY_CONF_DIR.set_for_testing(self.tmpdir), ] if API_CACHE is not None: self.resets.append(API_CACHE.set_for_testing(None)) if API2_CACHE is not None: self.resets.append(API2_CACHE.set_for_testing(None))
def test_ha_failover_all_bad(self): # Test with all bad hosts xml = self._sentry_site_xml(rpc_addresses='bad-host-1:8039,bad-host-2', rpc_port=self.rpc_port) file(os.path.join(self.tmpdir, 'sentry-site.xml'), 'w').write(xml) sentry_site.reset() api = get_api(self.user) assert_equal('bad-host-1:8039,bad-host-2', ','.join(sentry_site.get_sentry_server_rpc_addresses())) assert_raises(PopupException, api.list_sentry_roles_by_group, '*') api2 = get_api2(self.user, 'solr') assert_raises(PopupException, api2.list_sentry_roles_by_group, '*')
def test_no_rpc_hosts(self): # Test with no rpc hosts and fallback to hostname and port xml = self._sentry_site_xml(rpc_addresses='') file(os.path.join(self.tmpdir, 'sentry-site.xml'), 'w').write(xml) sentry_site.reset() api = get_api(self.user) assert_false(sentry_site.is_ha_enabled(), sentry_site.get_sentry_server_rpc_addresses()) assert_true(is_enabled() and HOSTNAME.get() and HOSTNAME.get() != 'localhost') resp = api.list_sentry_roles_by_group(groupName='*') assert_true(isinstance(resp, list)) api2 = get_api2(self.user, 'solr') resp = api2.list_sentry_roles_by_group(groupName='*') assert_true(isinstance(resp, list))
def test_ha_failover_good_bad_bad(self): # Test with good-host,bad-host-1,bad-host-2 xml = self._sentry_site_xml(rpc_addresses='%s,bad-host-1,bad-host-2' % self.rpc_addresses) file(os.path.join(self.tmpdir, 'sentry-site.xml'), 'w').write(xml) sentry_site.reset() api = get_api(self.user) assert_equal('%s,bad-host-1,bad-host-2' % self.rpc_addresses, ','.join(sentry_site.get_sentry_server_rpc_addresses())) resp = api.list_sentry_roles_by_group(groupName='*') assert_true(isinstance(resp, list)) api2 = get_api2(self.user, 'solr') resp = api2.list_sentry_roles_by_group(groupName='*') assert_true(isinstance(resp, list))