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)
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)
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)
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)
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, '')
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, '')
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
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(),
}
)
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(),
}
)
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)
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)
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)