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.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_reload_logs(self): _, filename = tempfile.mkstemp() r = PsDashRunner(args=['-l', filename]) pre_count = len(r.get_local_node().logs.available) r.get_local_node().logs.add_patterns(r.app.config['PSDASH_LOGS']) post_count = len(r.get_local_node().logs.available) self.assertEqual(pre_count, post_count)
def test_register_node_creates_proper_node_dict(self): r = PsDashRunner() now = int(time.time()) node = r.register_node('examplehost', 'example.org', 5000) self.assertEqual(node.host, 'example.org') self.assertEqual(node.port, 5000) self.assertEqual(node.last_registered, now)
def test_reregister_node(self): r = PsDashRunner() now = int(time.time()) r.register_node("examplehost", "example.org", 5000) node = r.register_node("examplehost", "example.org", 5000) self.assertEqual(node.host, "example.org") self.assertEqual(node.port, 5000) self.assertEqual(node.last_registered, now)
def test_default_args_dont_override_config(self): _, filename = tempfile.mkstemp() with open(filename, "w") as f: f.write("PSDASH_LOGS = ['/var/log/boot.log', '/var/log/dmesg']\n") f.flush() os.environ['PSDASH_CONFIG'] = filename r = PsDashRunner() self.assertEquals(r.app.config['PSDASH_LOGS'], ['/var/log/boot.log', '/var/log/dmesg']) del os.environ['PSDASH_CONFIG']
def test_update_net_io_counters(self): r = PsDashRunner() socket.getaddrinfo('example.org', 80) counters = r.get_local_node().net_io_counters.update() for c in counters.itervalues(): if c['rx_per_sec'] > 0 and c['tx_per_sec'] > 0: break else: self.fail("Didn't find any changed network interface")
def _run(self, https=False): options = {'PSDASH_PORT': 5051} if https: options.update({ 'PSDASH_HTTPS_KEYFILE': os.path.join(os.path.dirname(__file__), 'keyfile'), 'PSDASH_HTTPS_CERTFILE': os.path.join(os.path.dirname(__file__), 'cacert.pem') }) self.r = PsDashRunner(options) self.runner = gevent.spawn(self.r.run) gevent.sleep(0.3)
def _run(self, https=False): options = {"PSDASH_PORT": 5051} if https: options.update({ "PSDASH_HTTPS_KEYFILE": os.path.join(os.path.dirname(__file__), "keyfile"), "PSDASH_HTTPS_CERTFILE": os.path.join(os.path.dirname(__file__), "cacert.pem"), }) self.r = PsDashRunner(options) self.runner = gevent.spawn(self.r.run) gevent.sleep(0.3)
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): 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_multiple_remote_addresses_using_list(self): r = PsDashRunner( {'PSDASH_ALLOWED_REMOTE_ADDRESSES': ['127.0.0.1', '10.0.0.1']}) resp = r.app.test_client().get( '/', environ_overrides={'REMOTE_ADDR': '10.0.0.1'}) self.assertEqual(resp.status_code, httplib.OK) resp = r.app.test_client().get( '/', environ_overrides={'REMOTE_ADDR': '127.0.0.1'}) self.assertEqual(resp.status_code, httplib.OK) resp = r.app.test_client().get( '/', environ_overrides={'REMOTE_ADDR': '10.124.0.1'}) self.assertEqual(resp.status_code, httplib.UNAUTHORIZED)
def test_multiple_remote_addresses_using_list(self): r = PsDashRunner( {"PSDASH_ALLOWED_REMOTE_ADDRESSES": ["127.0.0.1", "10.0.0.1"]}) resp = r.app.test_client().get( "/", environ_overrides={"REMOTE_ADDR": "10.0.0.1"}) self.assertEqual(resp.status_code, http.client.OK) resp = r.app.test_client().get( "/", environ_overrides={"REMOTE_ADDR": "127.0.0.1"}) self.assertEqual(resp.status_code, http.client.OK) resp = r.app.test_client().get( "/", environ_overrides={"REMOTE_ADDR": "10.124.0.1"}) self.assertEqual(resp.status_code, http.client.UNAUTHORIZED)
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_works_on_prefix(self): r = PsDashRunner({'PSDASH_URL_PREFIX': self.default_prefix}) resp = r.app.test_client().get(self.default_prefix) self.assertEqual(resp.status_code, httplib.OK)
def test_incorrect_remote_address(self): r = PsDashRunner({'PSDASH_ALLOWED_REMOTE_ADDRESSES': '127.0.0.1'}) resp = r.app.test_client().get( '/', environ_overrides={'REMOTE_ADDR': '10.0.0.1'}) self.assertEqual(resp.status_code, httplib.UNAUTHORIZED)
def setUp(self): self.r = PsDashRunner() self.app = self.r.app self.client = self.app.test_client() self.filename = self._create_log_file() self.r.get_local_node().logs.add_available(self.filename)
def setUp(self): self.app = PsDashRunner().app self.client = self.app.test_client()
def setUp(self): self.r = PsDashRunner() self.app = self.r.app self.client = self.app.test_client() self.pid = os.getpid() self.r.get_local_node().net_io_counters.update()
def test_args_bind(self): r = PsDashRunner(args=['-b', '10.0.0.1']) self.assertEqual(r.app.config['PSDASH_BIND_HOST'], '10.0.0.1')
def test_missing_trailing_slash_works(self): r = PsDashRunner({'PSDASH_URL_PREFIX': '/subfolder'}) resp = r.app.test_client().get('/subfolder/') self.assertEqual(resp.status_code, httplib.OK)
def test_multiple_level_prefix(self): r = PsDashRunner({'PSDASH_URL_PREFIX': '/use/this/folder/'}) resp = r.app.test_client().get('/use/this/folder/') self.assertEqual(resp.status_code, httplib.OK)
def test_args_debug(self): r = PsDashRunner(args=['-d']) self.assertTrue(r.app.debug)
def test_page_not_found_on_root(self): r = PsDashRunner({'PSDASH_URL_PREFIX': self.default_prefix}) resp = r.app.test_client().get('/') self.assertEqual(resp.status_code, httplib.NOT_FOUND)
def test_show_only_whitelisted(self): r = PsDashRunner({'PSDASH_ENVIRON_WHITELIST': ['USER']}) resp = r.app.test_client().get('/process/%d/environment' % os.getpid()) self.assertTrue(os.environ['USER'] in resp.data) self.assertTrue('*hidden by whitelist*' in resp.data)
def test_args_log(self): _, filename = tempfile.mkstemp() r = PsDashRunner(args=['-l', filename]) self.assertEqual(r.app.config['PSDASH_LOGS'][0], filename)
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_args_port(self): r = PsDashRunner(args=['-p', '5555']) self.assertEqual(r.app.config['PSDASH_PORT'], 5555)
def test_local_node_is_added(self): r = PsDashRunner() self.assertIsInstance(r.get_local_node(), LocalNode)