def test_ignoring_comment(
self, m_load_profiles, m_migrate, m_mongo_collection, m_taskManager, m_open
):
periodic_obj_mock = Mock()
m_taskManager.return_value = periodic_obj_mock
m_load_profiles.return_value = default_profiles
self.assertEqual(False, load())
m_mongo_collection.assert_not_called()
periodic_obj_mock.manage_task.assert_not_called()
def test_load_unchanged_record(
self, m_load_profiles, m_migrate, m_mongo_collection, m_taskManager, m_open
):
m_mongo_collection.return_value = UpdateResult(
{"n": 1, "nModified": 0, "upserted": None}, True
)
periodic_obj_mock = Mock()
m_taskManager.return_value = periodic_obj_mock
m_load_profiles.return_value = default_profiles
self.assertEqual(False, load())
periodic_obj_mock.manage_task.assert_not_called()
def test_deleting_record_non_default_port(
self, m_load_profiles, m_migrate, m_remove, m_delete, m_taskManager, m_open
):
periodic_obj_mock = Mock()
m_taskManager.return_value = periodic_obj_mock
m_load_profiles.return_value = default_profiles
self.assertEqual(False, load())
periodic_obj_mock.disable_tasks.assert_called_with("192.168.0.1:345")
m_delete.assert_called_with({"address": "192.168.0.1", "port": 345})
calls = m_remove.call_args_list
self.assertEqual(2, len(calls))
self.assertEqual(({"address": "192.168.0.1:345"},), calls[0].args)
self.assertEqual(({"address": "192.168.0.1:345"},), calls[1].args)
def test_inventory_errors(
self,
m_load_profiles,
m_migrate,
m_manage_task,
m_mongo_collection,
m_open,
walk_task,
):
walk_task.return_value = expected_managed_task
m_mongo_collection.return_value = UpdateResult(
{"n": 0, "nModified": 1, "upserted": 1}, True
)
m_manage_task.side_effect = Exception("Boom!")
m_load_profiles.return_value = default_profiles
self.assertEqual(True, load())
def test_load_new_record(
self,
m_load_profiles,
m_migrate,
m_mongo_collection,
m_taskManager,
m_open,
walk_task,
):
walk_task.return_value = expected_managed_task
m_mongo_collection.return_value = UpdateResult(
{"n": 0, "nModified": 1, "upserted": 1}, True
)
periodic_obj_mock = Mock()
m_taskManager.return_value = periodic_obj_mock
m_load_profiles.return_value = default_profiles
self.assertEqual(False, load())
periodic_obj_mock.manage_task.assert_called_with(**expected_managed_task)
def test_load_new_record_small_walk(
self,
m_load_profiles,
m_migrate,
m_mongo_collection,
m_taskManager,
m_open,
):
profiles = {
"walk1": {
"condition": {"type": "walk"},
"varBinds": [
["IF-MIB", "ifInDiscards", 1],
["IF-MIB", "ifOutErrors"],
["SNMPv2-MIB", "sysDescr", 0],
],
},
"walk2": {
"condition": {"type": "walk"},
"varBinds": [
["IF-MIB", "ifInDiscards", 1],
["IF-MIB", "ifOutErrors"],
["SNMPv2-MIB", "sysDescr", 0],
],
},
}
m_mongo_collection.return_value = UpdateResult(
{"n": 0, "nModified": 1, "upserted": 1}, True
)
periodic_obj_mock = Mock()
m_taskManager.return_value = periodic_obj_mock
m_load_profiles.return_value = profiles
self.assertEqual(False, load())
self.assertEqual(
{"address": "192.168.0.1", "profile": "walk2"},
periodic_obj_mock.manage_task.call_args.kwargs["kwargs"],
)