コード例 #1
0
ファイル: test_ona.py プロジェクト: sararadh-cool/ona
    def test_valid_config(self, mock_requests, mock_report_to_site):
        mo = mock_open(read_data='my_version')
        mock_requests.get.return_value = self.cool_response

        with patch('ona_service.ona.open', mo, create=True):
            ona = ONA(data_type='ona', poll_seconds=1, update_only=True)
            ona.config_mode = 'auto'
            # The configuration update should cause the service to exit
            # with a special return code
            with self.assertRaises(SystemExit):
                ona.execute()

        mock_report_to_site.assert_called_once_with(ona)

        expected_config = '\n'.join([
            'OBSRVBL_IPFIX_PROBE_4_PORT="9996"',
            'OBSRVBL_IPFIX_PROBE_4_TYPE="netflow-v9"',
            'OBSRVBL_NETWORKS="10.0.0.0/8 172.16.0.0/12 192.168.0.0/16"',
            'OBSRVBL_PDNS_PPS_LIMIT="102"',
            'OBSRVBL_SNMP_ENABLED="true"',
            'OBSRVBL_SNMP_OBJECTID="1.3.6.1.4.1.3375.2.100"',
            'OBSRVBL_SNMP_SERVER="127.0.0.1"',
            'OBSRVBL_SNMP_SERVER_PORT="162"',
            'OBSRVBL_SNMP_USER="******"',
            'OBSRVBL_SNMP_VERSION="2"',
            'OBSRVBL_SYSLOG_ENABLED="true"',
            'OBSRVBL_SYSLOG_FACILITY="user"',
            'OBSRVBL_SYSLOG_SERVER_PORT="51"',
        ])
        mo().write.assert_called_once_with(expected_config)
コード例 #2
0
ファイル: test_ona.py プロジェクト: obsrvbl/ona
    def test_valid_config(self, mock_requests, mock_report_to_site):
        mo = mock_open(read_data='my_version')
        mock_requests.get.return_value = self.cool_response

        with patch('ona_service.ona.open', mo, create=True):
            ona = ONA(data_type='ona', poll_seconds=1, update_only=True)
            ona.config_mode = 'auto'
            # The configuration update should cause the service to exit
            # with a special return code
            with self.assertRaises(SystemExit):
                ona.execute()

        mock_report_to_site.assert_called_once_with(ona)

        expected_config = '\n'.join([
            'OBSRVBL_NETWORKS="10.0.0.0/8 172.16.0.0/12 192.168.0.0/16"',
            'OBSRVBL_PDNS_PPS_LIMIT="102"',
            'OBSRVBL_SNMP_ENABLED="true"',
            'OBSRVBL_SNMP_OBJECTID="1.3.6.1.4.1.3375.2.100"',
            'OBSRVBL_SNMP_SERVER="127.0.0.1"',
            'OBSRVBL_SNMP_SERVER_PORT="162"',
            'OBSRVBL_SNMP_USER="******"',
            'OBSRVBL_SNMP_VERSION="2"',
            'OBSRVBL_SYSLOG_ENABLED="true"',
            'OBSRVBL_SYSLOG_FACILITY="user"',
            'OBSRVBL_SYSLOG_SERVER_PORT="51"',
        ])
        mo().write.assert_called_once_with(expected_config)
コード例 #3
0
ファイル: test_ona.py プロジェクト: sararadh-cool/ona
    def test_valid_no_config(self, mock_requests, mock_write_config):
        mock_requests.get.return_value = self.my_response
        ona = ONA(data_type='ona', poll_seconds=1, update_only=True)
        ona.config_mode = 'auto'

        with patch('ona_service.ona.open', mock_open(), create=True):
            ona.execute()

        self.assertEqual(mock_write_config.call_count, 0)
コード例 #4
0
ファイル: test_ona.py プロジェクト: obsrvbl/ona
    def test_valid_no_config(self, mock_requests, mock_write_config):
        mock_requests.get.return_value = self.my_response
        ona = ONA(data_type='ona', poll_seconds=1, update_only=True)
        ona.config_mode = 'auto'

        with patch('ona_service.ona.open', mock_open(), create=True):
            ona.execute()

        self.assertEqual(mock_write_config.call_count, 0)
コード例 #5
0
ファイル: test_ona.py プロジェクト: sararadh-cool/ona
    def test__load_config(self):
        file_contents = 'MYVAR="awesome"\n\n'
        mo = mock_open(read_data=file_contents)
        ona = ONA(data_type='ona', poll_seconds=1)

        with patch('ona_service.ona.open', mo, create=True):
            config = ona._load_config()
        self.assertEqual(config, file_contents.strip())

        mo.side_effect = IOError()
        with patch('ona_service.ona.open', mo, create=True):
            config = ona._load_config()
        self.assertEqual(config, '')
コード例 #6
0
ファイル: test_ona.py プロジェクト: obsrvbl/ona
    def test__load_config(self):
        file_contents = 'MYVAR="awesome"\n\n'
        mo = mock_open(read_data=file_contents)
        ona = ONA(data_type='ona', poll_seconds=1)

        with patch('ona_service.ona.open', mo, create=True):
            config = ona._load_config()
        self.assertEqual(config, file_contents.strip())

        mo.side_effect = IOError()
        with patch('ona_service.ona.open', mo, create=True):
            config = ona._load_config()
        self.assertEqual(config, '')
コード例 #7
0
    def _get_instance(self, update_only, env=None):
        env = env or {}
        with patch.dict('ona_service.ona.environ', env):
            ona = ONA(
                config_file=self.auto_config_path,
                poll_seconds=0,
                update_only=update_only,
            )
            ona.api = MagicMock()
            ona.api.proxy_uri = 'https://127.0.0.1:8080'
            ona.api.ona_name = 'ona-test-01'

            return ona
コード例 #8
0
ファイル: test_ona.py プロジェクト: sararadh-cool/ona
    def test__report_to_site(self, mock_requests, mock_datetime):
        mock_datetime.now.return_value = self.now

        mo = mock_open(read_data='my_version')
        with patch('ona_service.ona.open', mo, create=True):
            ona = ONA(data_type='ona', poll_seconds=1, update_only=True)
            ona._report_to_site()

        node = platform.node()
        mock_requests.post.assert_called_once_with(
            'https://sensor.ext.obsrvbl.com/signal/sensors/{}'.format(node),
            verify=True,
            timeout=HTTP_TIMEOUT,
            data={
                'platform': platform.platform(),
                'python_version': platform.python_version(),
                'config_mode': 'manual',
                'ona_version': 'my_version',
                'last_start': self.now.isoformat(),
            }
        )
コード例 #9
0
ファイル: test_ona.py プロジェクト: obsrvbl/ona
    def test__report_to_site(self, mock_requests, mock_datetime):
        mock_datetime.now.return_value = self.now

        mo = mock_open(read_data='my_version')
        with patch('ona_service.ona.open', mo, create=True):
            ona = ONA(data_type='ona', poll_seconds=1, update_only=True)
            ona._report_to_site()

        node = platform.node()
        mock_requests.post.assert_called_once_with(
            'https://sensor.ext.obsrvbl.com/signal/sensors/{}'.format(node),
            verify=True,
            timeout=HTTP_TIMEOUT,
            data={
                'platform': platform.platform(),
                'python_version': platform.python_version(),
                'config_mode': 'manual',
                'ona_version': 'my_version',
                'last_start': self.now.isoformat(),
            }
        )
コード例 #10
0
ファイル: test_ona.py プロジェクト: sararadh-cool/ona
    def test_check_config(self, mock_requests, mock_exit):
        ona = ONA(data_type='ona', poll_seconds=1)
        ona.config_mode = 'auto'

        # first execution, nothing has changed - don't exit
        mock_requests.get.return_value = self.my_response
        with patch('ona_service.ona.open', mock_open(), create=True):
            ona.execute()

        self.assertEqual(mock_exit.call_count, 0)

        # second execution, config has changed - exit
        mock_requests.get.return_value = self.cool_response
        with patch('ona_service.ona.open', mock_open(), create=True):
            ona.execute()

        mock_exit.assert_called_once_with(EAGAIN)
コード例 #11
0
ファイル: test_ona.py プロジェクト: obsrvbl/ona
    def test_check_config(self, mock_requests, mock_exit):
        ona = ONA(data_type='ona', poll_seconds=1)
        ona.config_mode = 'auto'

        # first execution, nothing has changed - don't exit
        mock_requests.get.return_value = self.my_response
        with patch('ona_service.ona.open', mock_open(), create=True):
            ona.execute()

        self.assertEqual(mock_exit.call_count, 0)

        # second execution, config has changed - exit
        mock_requests.get.return_value = self.cool_response
        with patch('ona_service.ona.open', mock_open(), create=True):
            ona.execute()

        mock_exit.assert_called_once_with(EAGAIN)
コード例 #12
0
ファイル: test_ona.py プロジェクト: sararadh-cool/ona
    def test_watch_ifaces(
        self, mock_requests, mock_exit, mock_getenv, mock_listdir
    ):
        # instantiation - ona.network_ifaces should be tracked
        env = {
            'OBSRVBL_MANAGE_MODE': 'manual',
            'OBSRVBL_WATCH_IFACES': 'true',
        }
        mock_getenv.side_effect = env.get
        mock_listdir.return_value = ['eth1', 'eth0']

        ona = ONA(data_type='ona', poll_seconds=1)
        self.assertEqual(ona.network_ifaces, {'eth0', 'eth1'})

        # first execution, nothing has changed - don't exit
        ona.execute()
        self.assertEqual(mock_exit.call_count, 0)

        # second execution, network interfaces have changed - exit
        ona.network_ifaces.pop()
        ona.execute()
        mock_exit.assert_called_once_with(EAGAIN)