예제 #1
0
    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)
예제 #2
0
    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)
예제 #3
0
 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)
예제 #4
0
 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)
예제 #5
0
 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)
예제 #6
0
파일: test_run.py 프로젝트: fingul/psdash
 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']
예제 #7
0
    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")
예제 #8
0
 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)
예제 #9
0
 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)
예제 #10
0
    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)
예제 #11
0
    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)
예제 #12
0
    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)
예제 #13
0
    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)
예제 #14
0
 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)
예제 #15
0
 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)
예제 #16
0
 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)
예제 #17
0
 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)
예제 #18
0
 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)
예제 #19
0
 def setUp(self):
     self.app = PsDashRunner().app
     self.client = self.app.test_client()
예제 #20
0
 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()
예제 #21
0
 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')
예제 #22
0
 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)
예제 #23
0
 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)
예제 #24
0
 def test_args_debug(self):
     r = PsDashRunner(args=['-d'])
     self.assertTrue(r.app.debug)
예제 #25
0
 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)
예제 #26
0
 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)
예제 #27
0
 def test_args_log(self):
     _, filename = tempfile.mkstemp()
     r = PsDashRunner(args=['-l', filename])
     self.assertEqual(r.app.config['PSDASH_LOGS'][0], filename)
예제 #28
0
 def test_get_all_nodes(self):
     r = PsDashRunner()
     r.register_node('examplehost', 'example.org', 5000)
     self.assertEqual(len(r.get_nodes()), 2)  # local + registered
예제 #29
0
 def test_args_port(self):
     r = PsDashRunner(args=['-p', '5555'])
     self.assertEqual(r.app.config['PSDASH_PORT'], 5555)
예제 #30
0
 def test_local_node_is_added(self):
     r = PsDashRunner()
     self.assertIsInstance(r.get_local_node(), LocalNode)