コード例 #1
0
    def test_commands(self, mock_logger_warn, mock_logger_info, sock_file):
        wm = WorkerMaster('sock://%s' % sock_file)
        wm.run()

        try:
            s = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
            s.connect(sock_file)

            s.send(bytes('PING\n', 'UTF-8'))
            self.assertEqual(s.recv(1024).decode('UTF-8').strip(), 'PONG')

            s.send(bytes('FOOBAR\n', 'UTF-8'))
            self.assertEqual(
                s.recv(1024).decode('UTF-8').strip(), 'UNKNOWN COMMAND')

            mock_logger_info.reset_mock()
            s.send(bytes('HIT grp 1.1.1.1\n', 'UTF-8'))
            self.assertEqual(s.recv(1024).decode('UTF-8').strip(), 'BULLSEYE')
            mock_logger_info.assert_called_once_with('HIT grp 1.1.1.1')

            s.send(b"\xFF\xFF\n")
            self.assertEqual(s.recv(1024).decode('UTF-8').strip(), 'WHAT?')
            mock_logger_warn.assert_called_once_with(
                'WorkMasterThread recieved invalid data')

            s.send(bytes('QUIT\n', 'UTF-8'))
            self.assertEqual(s.recv(1024).decode('UTF-8').strip(), 'BYE')

            with self.assertRaises(BrokenPipeError):
                s.send(bytes('ANYTHING\n', 'UTF-8'))
            s.close()
        finally:
            wm.stop()
コード例 #2
0
    def test_match_line(self, mock_logger_error, sock_file):
        wm = WorkerMaster('sock://%s' % sock_file, filters=['grp1:aaa', 'grp1:bbb', 'grp2:ccc'])
        wm.run()

        try:
            wm.match_line('test line')
            mock_logger_error.assert_called_once_with('No workers connected - unable to process line "test line"')

            s1 = utils.socket_connect_from_text("sock://%s" % sock_file)
            f1 = s1.makefile('r')
            # ignore filter definitions
            for _ in range(3):
                f1.readline().strip()

            s2 = utils.socket_connect_from_text("sock://%s" % sock_file)
            f2 = s2.makefile('r')
            # ignore filter definitions
            for _ in range(3):
                f2.readline().strip()

            wm.match_line('test line2')
            wm.match_line('test line3')

            l1 = f1.readline().strip()
            l2 = f2.readline().strip()
            self.assertIn(l1, ('MATCH test line2', 'MATCH test line3', ))
            self.assertIn(l2, ('MATCH test line2', 'MATCH test line3', ))
            self.assertNotEqual(l1, l2)

            f1.close()
            s1.close()
            f2.close()
            s2.close()
        finally:
            wm.stop()
コード例 #3
0
    def test_commands(self, mock_logger_warn, mock_logger_info, sock_file):
        wm = WorkerMaster('sock://%s' % sock_file)
        wm.run()

        try:
            s = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
            s.connect(sock_file)

            s.send(bytes('PING\n', 'UTF-8'))
            self.assertEqual(s.recv(1024).decode('UTF-8').strip(), 'PONG')

            s.send(bytes('FOOBAR\n', 'UTF-8'))
            self.assertEqual(s.recv(1024).decode('UTF-8').strip(), 'UNKNOWN COMMAND')

            mock_logger_info.reset_mock()
            s.send(bytes('HIT grp 1.1.1.1\n', 'UTF-8'))
            self.assertEqual(s.recv(1024).decode('UTF-8').strip(), 'BULLSEYE')
            mock_logger_info.assert_called_once_with('HIT grp 1.1.1.1')

            s.send(b"\xFF\xFF\n")
            self.assertEqual(s.recv(1024).decode('UTF-8').strip(), 'WHAT?')
            mock_logger_warn.assert_called_once_with('WorkMasterThread recieved invalid data')

            s.send(bytes('QUIT\n', 'UTF-8'))
            self.assertEqual(s.recv(1024).decode('UTF-8').strip(), 'BYE')


            with self.assertRaises(BrokenPipeError):
                s.send(bytes('ANYTHING\n', 'UTF-8'))
            s.close()
        finally:
            wm.stop()
コード例 #4
0
    def test_listen_sock(self, sock_file):
        s = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
        wm = WorkerMaster('sock://%s' % sock_file)
        wm.run()

        s.connect(sock_file)
        s.close()
        wm.stop()
コード例 #5
0
    def test_listen_tcp6(self):
        s = socket.socket(socket.AF_INET6, socket.SOCK_STREAM)
        wm = WorkerMaster('tcp6://[::1]:4432')
        wm.run()

        s.connect(('::1', 4432))
        s.close()
        wm.stop()
コード例 #6
0
    def test_listen_tcp(self):
        s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        wm = WorkerMaster('tcp://0.0.0.0:4432')
        wm.run()

        s.connect(('127.0.0.1', 4432))
        s.close()
        wm.stop()
コード例 #7
0
    def test_listen_sock(self, sock_file):
        s = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
        wm = WorkerMaster('sock://%s' % sock_file)
        wm.run()

        s.connect(sock_file)
        s.close()
        wm.stop()
コード例 #8
0
    def test_listen_tcp6(self):
        s = socket.socket(socket.AF_INET6, socket.SOCK_STREAM)
        wm = WorkerMaster('tcp6://[::1]:4432')
        wm.run()

        s.connect(('::1', 4432))
        s.close()
        wm.stop()
コード例 #9
0
    def test_listen_tcp(self):
        s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        wm = WorkerMaster('tcp://0.0.0.0:4432')
        wm.run()

        s.connect(('127.0.0.1', 4432))
        s.close()
        wm.stop()
コード例 #10
0
ファイル: run_server.py プロジェクト: nextsux/logemperor
def main(config):
    log.init(config.get('server', 'log_level'))

    wm = WorkerMaster(config.get('server', 'worker_listen'), filters=config.get('filters', 'regex').split('\n'))
    pm = ProviderMaster(config.get('server', 'provider_listen'), worker_master=wm)
    try:
        wm.run()
        pm.run()
    except KeyboardInterrupt:
        wm.stop()
        pm.stop()
コード例 #11
0
    def test_client_close_connection(self, sock_file):
        wm = WorkerMaster('sock://%s' % sock_file)
        wm.run()

        try:
            s = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
            s.connect(sock_file)

            s.send(bytes('PING\n', 'UTF-8'))
            self.assertEqual(s.recv(1024).decode('UTF-8').strip(), 'PONG')
            s.close()
        finally:
            wm.stop()
コード例 #12
0
    def test_client_close_connection(self, sock_file):
        wm = WorkerMaster('sock://%s' % sock_file)
        wm.run()

        try:
            s = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
            s.connect(sock_file)

            s.send(bytes('PING\n', 'UTF-8'))
            self.assertEqual(s.recv(1024).decode('UTF-8').strip(), 'PONG')
            s.close()
        finally:
            wm.stop()
コード例 #13
0
ファイル: run_server.py プロジェクト: nextsux/logemperor
def main(config):
    log.init(config.get('server', 'log_level'))

    wm = WorkerMaster(config.get('server', 'worker_listen'),
                      filters=config.get('filters', 'regex').split('\n'))
    pm = ProviderMaster(config.get('server', 'provider_listen'),
                        worker_master=wm)
    try:
        wm.run()
        pm.run()
    except KeyboardInterrupt:
        wm.stop()
        pm.stop()
コード例 #14
0
    def test_filters(self, sock_file):
        wm = WorkerMaster('sock://%s' % sock_file, filters=['grp1:aaa', 'grp1:bbb', 'grp2:ccc'])
        wm.run()

        try:
            s = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
            s.connect(sock_file)
            f = s.makefile('r')

            self.assertEqual(sorted([f.readline().strip() for _ in range(3)]), ['FILTER ADD grp1 aaa', 'FILTER ADD grp1 bbb', 'FILTER ADD grp2 ccc'])

            f.close()
            s.close()
        finally:
            wm.stop()
コード例 #15
0
    def test_match_line(self, mock_logger_error, sock_file):
        wm = WorkerMaster('sock://%s' % sock_file,
                          filters=['grp1:aaa', 'grp1:bbb', 'grp2:ccc'])
        wm.run()

        try:
            wm.match_line('test line')
            mock_logger_error.assert_called_once_with(
                'No workers connected - unable to process line "test line"')

            s1 = utils.socket_connect_from_text("sock://%s" % sock_file)
            f1 = s1.makefile('r')
            # ignore filter definitions
            for _ in range(3):
                f1.readline().strip()

            s2 = utils.socket_connect_from_text("sock://%s" % sock_file)
            f2 = s2.makefile('r')
            # ignore filter definitions
            for _ in range(3):
                f2.readline().strip()

            wm.match_line('test line2')
            wm.match_line('test line3')

            l1 = f1.readline().strip()
            l2 = f2.readline().strip()
            self.assertIn(l1, (
                'MATCH test line2',
                'MATCH test line3',
            ))
            self.assertIn(l2, (
                'MATCH test line2',
                'MATCH test line3',
            ))
            self.assertNotEqual(l1, l2)

            f1.close()
            s1.close()
            f2.close()
            s2.close()
        finally:
            wm.stop()
コード例 #16
0
    def test_filters(self, sock_file):
        wm = WorkerMaster('sock://%s' % sock_file,
                          filters=['grp1:aaa', 'grp1:bbb', 'grp2:ccc'])
        wm.run()

        try:
            s = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
            s.connect(sock_file)
            f = s.makefile('r')

            self.assertEqual(sorted([f.readline().strip() for _ in range(3)]),
                             [
                                 'FILTER ADD grp1 aaa', 'FILTER ADD grp1 bbb',
                                 'FILTER ADD grp2 ccc'
                             ])

            f.close()
            s.close()
        finally:
            wm.stop()