def test_nodes_from_config(self):
config = {
'PSDASH_NODES': [
{
'name': 'test-node',
'host': 'remotehost.org',
'port': 5000
}
]
}
r = PsDashRunner(config)
self.assertEqual(len(r.get_nodes()), 2)
self.assertIn('remotehost.org:5000', r.get_nodes())
self.assertEqual(r.get_nodes()['remotehost.org:5000'].name, 'test-node')
self.assertEqual(r.get_nodes()['remotehost.org:5000'].host, 'remotehost.org')
self.assertEqual(r.get_nodes()['remotehost.org:5000'].port, 5000)
def test_nodes_from_config(self):
config = {
'PSDASH_NODES': [{
'name': 'test-node',
'host': 'remotehost.org',
'port': 5000
}]
}
r = PsDashRunner(config)
self.assertEqual(len(r.get_nodes()), 2)
self.assertIn('remotehost.org:5000', r.get_nodes())
self.assertEqual(r.get_nodes()['remotehost.org:5000'].name,
'test-node')
self.assertEqual(r.get_nodes()['remotehost.org:5000'].host,
'remotehost.org')
self.assertEqual(r.get_nodes()['remotehost.org:5000'].port, 5000)
def test_register_agent_to_auth_protected_host(self):
r = PsDashRunner({
'PSDASH_AUTH_USERNAME': '******',
'PSDASH_AUTH_PASSWORD': '******'
})
agent = PsDashRunner({
'PSDASH_AGENT': True,
'PSDASH_PORT': 5001,
'PSDASH_REGISTER_TO': 'http://localhost:5000',
'PSDASH_AUTH_USERNAME': '******',
'PSDASH_AUTH_PASSWORD': '******'
})
jobs = []
jobs.append(gevent.spawn(r.run))
gevent.sleep(0.3)
jobs.append(gevent.spawn(agent.run))
gevent.sleep(0.3)
self.assertIn('127.0.0.1:5001', r.get_nodes())
self.assertEquals(
r.get_node('127.0.0.1:5001').name, socket.gethostname())
self.assertEquals(r.get_node('127.0.0.1:5001').port, 5001)
r.server.close()
agent.server.close()
gevent.killall(jobs)
def test_register_agent_to_auth_protected_host(self):
r = PsDashRunner({
'PSDASH_AUTH_USERNAME': '******',
'PSDASH_AUTH_PASSWORD': '******'
})
agent = PsDashRunner({
'PSDASH_AGENT': True,
'PSDASH_PORT': 5001,
'PSDASH_REGISTER_TO': 'http://localhost:5000',
'PSDASH_AUTH_USERNAME': '******',
'PSDASH_AUTH_PASSWORD': '******'
})
jobs = []
jobs.append(gevent.spawn(r.run))
gevent.sleep(0.3)
jobs.append(gevent.spawn(agent.run))
gevent.sleep(0.3)
self.assertIn('127.0.0.1:5001', r.get_nodes())
self.assertEquals(r.get_node('127.0.0.1:5001').name, socket.gethostname())
self.assertEquals(r.get_node('127.0.0.1:5001').port, 5001)
r.server.close()
agent.server.close()
gevent.killall(jobs)
def test_nodes_from_config(self):
config = {
"PSDASH_NODES": [{
"name": "test-node",
"host": "remotehost.org",
"port": 5000
}]
}
r = PsDashRunner(config)
self.assertEqual(len(r.get_nodes()), 2)
self.assertIn("remotehost.org:5000", r.get_nodes())
self.assertEqual(r.get_nodes()["remotehost.org:5000"].name,
"test-node")
self.assertEqual(r.get_nodes()["remotehost.org:5000"].host,
"remotehost.org")
self.assertEqual(r.get_nodes()["remotehost.org:5000"].port, 5000)
def test_register_agent_to_auth_protected_host(self):
r = PsDashRunner({
"PSDASH_AUTH_USERNAME": "******",
"PSDASH_AUTH_PASSWORD": "******"
})
agent = PsDashRunner({
"PSDASH_AGENT": True,
"PSDASH_PORT": 5001,
"PSDASH_REGISTER_TO": "http://localhost:5000",
"PSDASH_AUTH_USERNAME": "******",
"PSDASH_AUTH_PASSWORD": "******",
})
jobs = []
jobs.append(gevent.spawn(r.run))
gevent.sleep(0.3)
jobs.append(gevent.spawn(agent.run))
gevent.sleep(0.3)
self.assertIn("127.0.0.1:5001", r.get_nodes())
self.assertEqual(
r.get_node("127.0.0.1:5001").name, socket.gethostname())
self.assertEqual(r.get_node("127.0.0.1:5001").port, 5001)
r.server.close()
agent.server.close()
gevent.killall(jobs)
def test_register_agent_without_name_defaults_to_hostname(self):
agent_options = {
'PSDASH_AGENT': True,
'PSDASH_PORT': 5001,
'PSDASH_REGISTER_TO': 'http://localhost:5000'
}
r = PsDashRunner()
agent = PsDashRunner(agent_options)
jobs = []
jobs.append(gevent.spawn(r.run))
gevent.sleep(0.3)
jobs.append(gevent.spawn(agent.run))
gevent.sleep(0.3)
self.assertIn('127.0.0.1:5001', r.get_nodes())
self.assertEquals(r.get_node('127.0.0.1:5001').name, socket.gethostname())
self.assertEquals(r.get_node('127.0.0.1:5001').port, 5001)
r.server.close()
agent.server.close()
gevent.killall(jobs)
def test_register_agent(self):
jobs = []
agent_options = {
'PSDASH_AGENT': True,
'PSDASH_PORT': 5001,
'PSDASH_REGISTER_TO': 'http://localhost:5000',
'PSDASH_REGISTER_AS': 'the_agent'
}
r = PsDashRunner()
agent = PsDashRunner(agent_options)
jobs.append(gevent.spawn(r.run))
gevent.sleep(0.3)
jobs.append(gevent.spawn(agent.run))
gevent.sleep(0.3)
self.assertIn('127.0.0.1:5001', r.get_nodes())
self.assertEquals(r.get_node('127.0.0.1:5001').name, 'the_agent')
self.assertEquals(r.get_node('127.0.0.1:5001').port, 5001)
r.server.close()
agent.server.close()
gevent.killall(jobs)
def test_register_agent_without_name_defaults_to_hostname(self):
agent_options = {
"PSDASH_AGENT": True,
"PSDASH_PORT": 5001,
"PSDASH_REGISTER_TO": "http://localhost:5000",
}
r = PsDashRunner()
agent = PsDashRunner(agent_options)
jobs = []
jobs.append(gevent.spawn(r.run))
gevent.sleep(0.3)
jobs.append(gevent.spawn(agent.run))
gevent.sleep(0.3)
self.assertIn("127.0.0.1:5001", r.get_nodes())
self.assertEqual(
r.get_node("127.0.0.1:5001").name, socket.gethostname())
self.assertEqual(r.get_node("127.0.0.1:5001").port, 5001)
r.server.close()
agent.server.close()
gevent.killall(jobs)
def test_register_agent(self):
agent_options = {
'PSDASH_AGENT': True,
'PSDASH_PORT': 5001,
'PSDASH_REGISTER_TO': 'localhost:5000',
'PSDASH_REGISTER_AS': 'the_agent'
}
r = PsDashRunner()
agent = PsDashRunner(agent_options)
jobs = []
jobs.append(gevent.spawn(r.run))
gevent.sleep(0.2)
jobs.append(gevent.spawn(agent.run))
gevent.sleep(0.2)
self.assertIn('127.0.0.1:5001', r.get_nodes())
self.assertEquals(r.get_node('127.0.0.1:5001').name, 'the_agent')
self.assertEquals(r.get_node('127.0.0.1:5001').port, 5001)
r.server.close()
agent.server.close()
gevent.killall(jobs)
def test_get_all_nodes(self):
r = PsDashRunner()
r.register_node('examplehost', 'example.org', 5000)
self.assertEqual(len(r.get_nodes()), 2) # local + registered
def test_get_all_nodes(self):
r = PsDashRunner()
r.register_node('examplehost', 'example.org', 5000)
self.assertEqual(len(r.get_nodes()), 2) # local + registered
