def setUp(self):
with patch('auklet.broker.MQTTClient._get_conf') as _get_conf:
with patch("auklet.monitoring.processing.Client._register_device",
new=self.__register_device):
_get_conf.side_effect = self.get_conf
self.monitoring = Monitoring(apikey="",
app_id="",
base_url="https://api-staging.io",
monitoring=True)
self.monitoring.monitor = True
def setUp(self):
with patch('auklet.broker.MQTTClient._get_conf') as _get_conf:
with patch("auklet.monitoring.processing.Client._register_device",
new=self.__register_device):
with patch("os.path.isfile") as is_file_mock:
is_file_mock.return_value = False
_get_conf.side_effect = self.get_conf
self.monitoring = Monitoring(
api_key="",
app_id="",
release="",
base_url="https://api-staging.io",
monitoring=True)
self.monitoring.monitor = True
def with_auklet_and_mqtt(get_certs_mock, update_limits_mock):
"""
With a little help from mock, this test will push data that
would normally go to the front end, to a MQTT container created locally.
"""
print("\n\nStarting benchmark tests with the Auklet Agent and MQTT...")
def _load_conf(self):
self.brokers = "mqtt"
self.port = 1883
def create_producer(self):
# Make non SSL producer for testing
self.producer = mqtt.Client()
self.producer.on_disconnect = self.on_disconnect
self.producer.connect_async("mqtt", 1883)
self.producer.loop_start()
def _get_conf(self):
return True
def register_device(self):
return True
update_limits_mock.return_value = 10000
get_certs_mock.return_value = True
conf_patcher = patch('auklet.broker.MQTTClient._read_from_conf',
new=_load_conf)
producer_patcher = patch('auklet.broker.MQTTClient.create_producer',
new=create_producer)
get_conf_patcher = patch('auklet.broker.MQTTClient._get_conf',
new=_get_conf)
register_device_patcher = patch(
'auklet.monitoring.processing.Client._register_device',
new=register_device)
conf_patcher.start()
producer_patcher.start()
get_conf_patcher.start()
register_device_patcher.start()
auklet_monitoring = Monitoring("", "", "")
auklet_monitoring.start()
base.start(state="WithAukletMQTT")
conf_patcher.stop()
producer_patcher.stop()
class TestMonitoring(unittest.TestCase):
def setUp(self):
with patch('auklet.broker.MQTTClient._get_conf') as _get_conf:
with patch("auklet.monitoring.processing.Client._register_device",
new=self.__register_device):
with patch("os.path.isfile") as is_file_mock:
is_file_mock.return_value = False
_get_conf.side_effect = self.get_conf
self.monitoring = Monitoring(
api_key="",
app_id="",
release="",
base_url="https://api-staging.io",
monitoring=True)
self.monitoring.monitor = True
def test_start(self):
self.assertIsNone(self.monitoring.start())
self.monitoring.stop()
def test_initialize_without_release(self):
self.assertRaises(AukletConfigurationError,
Monitoring,
api_key="",
app_id="",
base_url="https://api.auklet.io/")
def test_stop(self):
self.monitoring.start()
self.monitoring.stop()
self.assertTrue(self.monitoring.stopping)
def test_sample(self):
class CoCode:
co_code = None
co_firstlineno = None
co_name = None
class FBack:
f_back = None
f_code = CoCode()
class Frame:
f_back = FBack()
f_code = CoCode()
def update_hash(self, stack):
global test_sample_stack # used to tell if stack was created
test_sample_stack = stack
with patch('auklet.stats.MonitoringTree.update_hash', new=update_hash):
self.monitoring.sample(None, current_frame=Frame())
self.assertIsNotNone(test_sample_stack)
self.monitoring.sample(None, current_frame=Frame())
self.assertTrue(test_sample_stack)
def test_process_periodic(self):
def produce(self, data):
global test_process_periodic_produce_data
test_process_periodic_produce_data = True
def check_date(self):
global test_process_periodic_check_date
test_process_periodic_check_date = True
with patch('auklet.broker.MQTTClient.produce', new=produce):
with patch('auklet.monitoring.processing.Client.check_date',
new=check_date):
self.monitoring.process_periodic()
self.assertTrue(test_process_periodic_produce_data)
self.assertTrue(test_process_periodic_check_date)
self.assertEqual(60000, self.monitoring.emission_rate)
def test_handle_exc(self):
with patch('auklet.broker.MQTTClient.produce') as _produce:
with patch(
'auklet.monitoring.processing.'
'Client.build_msgpack_event_data') \
as _build_msgpack_event_data:
with patch('sys.__excepthook__') as ___excepthook__:
___excepthook__.side_effect = self.__excepthook__
_build_msgpack_event_data.return_value = True
_produce.side_effect = self.produce
self.monitoring.handle_exc(None, None, None)
self.assertTrue(test_handle_exc___excepthook___)
def test_log(self):
with patch('auklet.broker.MQTTClient.produce') as _produce:
_produce.side_effect = self.produce
self.monitoring.log("", "")
self.assertIsNotNone(test_log_data)
def build_msgpack_tree(self, app_id):
print(app_id)
def __register_device(self):
return True
@staticmethod
def produce(data, data_type):
global test_log_data
test_log_data = data
@staticmethod
def get_conf():
return True
@staticmethod
def __excepthook__(type, value, traceback):
global test_handle_exc___excepthook___
test_handle_exc___excepthook___ = True
def main():
auklet_monitoring = Monitoring("api_key", "app_id", base_url="api-staging.auklet.io")
auklet_monitoring.start()
generator()
auklet_monitoring.stop()
class TestMonitoring(unittest.TestCase):
def setUp(self):
with patch('auklet.broker.MQTTClient._get_conf') as _get_conf:
with patch("auklet.monitoring.processing.Client._register_device",
new=self.__register_device):
_get_conf.side_effect = self.get_conf
self.monitoring = Monitoring(apikey="",
app_id="",
base_url="https://api-staging.io",
monitoring=True)
self.monitoring.monitor = True
def test_start(self):
self.assertIsNone(self.monitoring.start())
self.monitoring.stop()
def test_stop(self):
def _wait_for_stop(self):
global test_stop_wait_for_stop
test_stop_wait_for_stop = True
self.monitoring.start()
with patch('auklet.monitoring.Monitoring.wait_for_stop',
new=_wait_for_stop):
self.monitoring.stop()
self.assertTrue(self.monitoring.stopping)
self.assertTrue(test_stop_wait_for_stop)
def test_wait_for_stop(self):
def _sleep(interval):
raise (KeyboardInterrupt)
with patch('time.sleep', new=_sleep):
self.assertRaises(KeyboardInterrupt,
lambda: self.monitoring.wait_for_stop())
def test_sample(self):
class CoCode:
co_code = None
co_firstlineno = None
co_name = None
class FBack:
f_back = None
f_code = CoCode()
class Frame:
f_back = FBack()
f_code = CoCode()
def update_hash(self, stack):
global test_sample_stack # used to tell if stack was created
test_sample_stack = stack
with patch('auklet.stats.MonitoringTree.update_hash', new=update_hash):
self.monitoring.sample(None, current_frame=Frame())
self.assertIsNotNone(test_sample_stack)
self.monitoring.sample(None, current_frame=Frame())
self.assertTrue(test_sample_stack)
def test_process_periodic(self):
def produce(self, data):
global test_process_periodic_produce_data
test_process_periodic_produce_data = True
def update_network_metrics(self, interval):
global test_process_periodic_update_network_metrics_interval
test_process_periodic_update_network_metrics_interval = True
def check_date(self):
global test_process_periodic_check_date
test_process_periodic_check_date = True
with patch('auklet.broker.MQTTClient.produce', new=produce):
with patch(
'auklet.monitoring.processing.'
'Client.update_network_metrics',
new=update_network_metrics):
with patch('auklet.monitoring.processing.Client.check_date',
new=check_date):
self.monitoring.process_periodic()
self.assertTrue(test_process_periodic_produce_data)
self.assertTrue(
test_process_periodic_update_network_metrics_interval)
self.assertTrue(test_process_periodic_check_date)
self.assertEqual(60000, self.monitoring.emission_rate)
def test_handle_exc(self):
with patch('auklet.broker.MQTTClient.produce') as _produce:
with patch(
'auklet.monitoring.processing.'
'Client.build_msgpack_event_data') \
as _build_msgpack_event_data:
with patch('sys.__excepthook__') as ___excepthook__:
___excepthook__.side_effect = self.__excepthook__
_build_msgpack_event_data.return_value = True
_produce.side_effect = self.produce
self.monitoring.handle_exc(None, None, None)
self.assertTrue(test_handle_exc___excepthook___)
def test_log(self):
with patch('auklet.broker.MQTTClient.produce') as _produce:
_produce.side_effect = self.produce
self.monitoring.log("", "")
self.assertIsNotNone(test_log_data)
def build_msgpack_tree(self, app_id):
print(app_id)
def __register_device(self):
return True
@staticmethod
def produce(data, data_type):
global test_log_data
test_log_data = data
@staticmethod
def get_conf():
return True
@staticmethod
def __excepthook__(type, value, traceback):
global test_handle_exc___excepthook___
test_handle_exc___excepthook___ = True