def test_http(self):
service = Service(name='test service')
service.domains[u'http'] = ('example.com',)
self.assertNotEqual(service.get_nginx_config().find('server_name example.com *.example.com;'), -1,
'server_name and wildcard present')
self.assertTrue(service.validate(), 'nginx config is valid')
def test_upstream_nodes(self):
service = Service(name='test service')
service.domains[u'http'] = ('example.com',)
service.nodes['test_node'] = {'address': '127.0.0.1', 'port': '10000'}
self.assertTrue(service.validate(), 'nginx config is valid')
config = service.get_nginx_config()
self.assertNotEqual(config.find('server 127.0.0.1:10000;'), -1, 'upstream node present')
self.assertEqual(config.find('server 127.0.0.1:6666'), -1, 'backup node deleted')
def test_http(self):
service = Service(name='test service')
service.binds['http'] = {'example.com'}
self.assertNotEqual(
service.get_nginx_config('http').find(
'server_name example.com *.example.com;'), -1,
'server_name and wildcard present')
self.assertTrue(service.validate(), 'nginx config is valid')
def test_base(self):
service = Service(name='test service')
service.flush_nginx_config()
config_file = service.get_nginx_config_path('upstream')
self.assertNotEqual(
service.read_nginx_config_file('upstream').find(
'server 127.0.0.1:6666'), -1,
'config written and has backup 503')
self.assertTrue(service.validate(), 'nginx config is valid')
service.delete()
with self.assertRaises(IOError):
open(config_file, 'r')
def test_base(self):
service = Service(name='test service')
service.flush_nginx_config()
config_file = service.get_nginx_config_path()
self.assertNotEqual(service.read_nginx_config_file().find('server 127.0.0.1:6666'), -1,
'config written and has backup 503')
self.assertTrue(service.validate(), 'nginx config is valid')
service.delete()
with self.assertRaises(IOError):
open(config_file, 'r')
def check_services(self, data):
# check if service has any of our tags
services_to_publish = dict(
(k, v) for k, v in data.items() if any(x in v for x in [u'http', u'http2', u'tcp', u'udp']))
for service_name in services_to_publish:
if service_name not in self.services:
vergilius.logger.info('[service watcher]: new service: %s' % service_name)
self.services[service_name] = Service(service_name)
# cleanup stale services
for service_name in self.services.keys():
if service_name not in services_to_publish.iterkeys():
vergilius.logger.info('[service watcher]: removing stale service: %s' % service_name)
del self.services[service_name]
def test_udp(self, _):
service = Service(name='test service')
service.binds['udp'] = {'10000'}
service.nodes['test_node'] = {'address': '127.0.0.1', 'port': '10000'}
self.assertTrue(service.validate(), 'nginx config is valid')
config = service.get_nginx_config('udp')
self.assertNotEqual(config.find('listen %s udp;' % service.port), -1,
'udp listen valid')
def test_port_allocate(self):
service = Service(name='test service')
service.check_port()
self.assertEqual(service.port, 7000)
consul_port_data = consul.kv.get('vergilius/ports/test service')
self.assertIsNotNone(consul_port_data)
self.assertEqual('7000', consul_port_data[1][u'Value'])
service.delete()
self.assertFalse(7000 in port_allocator.allocated)
consul_port_data = consul.kv.get('vergilius/ports/test service')
self.assertIsNotNone(consul_port_data)
def test_upstream_nodes(self, _):
service = Service(name='test service')
service.binds['http'] = {'example.com'}
service.nodes['test_node'] = {'address': '127.0.0.1', 'port': '10000'}
self.assertTrue(service.validate(), 'nginx config is valid')
config = service.get_nginx_config('upstream')
self.assertNotEqual(config.find('server 127.0.0.1:10000;'), -1,
'upstream node present')
self.assertEqual(config.find('server 127.0.0.1:6666'), -1,
'backup node deleted')
def test_http2(self):
service = Service(name='test service')
service.domains[u'http2'] = ('example.com',)
self.assertTrue(service.validate(), 'nginx config is valid')
def test_http2(self):
service = Service(name='test service')
service.binds['http2'] = {'example.com'}
self.assertTrue(service.validate(), 'nginx config is valid')
def __init__(self, methodName='runTest'):
super(Test, self).__init__(methodName)
self.service = Service('test')
self.service.binds['http2'] = {'example.com'}