def test0001(self, **mocks): """ test de ok call add_system_from_ip """ self._init_mocks_remote_system(mocks) res = add_system_from_ip('192.168.0.1', 'ascodevida', add_to_database=True) self.assertTrue(res[0]) # Check params (ordered, keydict) = mocks['ansible_add_system'].call_args self.assertTrue(keydict['local_system_id'] == self.uuidlocal) self.assertTrue(keydict['remote_system_ip'] == '192.168.0.1') self.assertTrue(keydict['password'] == 'ascodevida') # (ordered, keydict) = mocks['add_child_server'].call_args self.assertTrue(ordered[0] == '192.168.0.1') self.assertTrue(ordered[1] == self.system_info['server_id']) # add_db_system (ordered, keydict) = mocks['db_add_system'].call_args self.assertTrue(keydict['system_id'] == self.system_info['system_id']) self.assertTrue(keydict['name'] == self.system_info['hostname']) self.assertTrue(keydict['admin_ip'] == '192.168.0.1') self.assertTrue( keydict['profile'] == ",".join(self.system_info['profile'])) self.assertTrue(keydict['server_id'] == self.system_info['server_id']) self.assertTrue(keydict['sensor_id'] is None)
def test0002(self, **mocks): """ test de fail call get_system_id_from_local """ self._init_mocks_remote_system(mocks) mocks['get_system_id_from_local'].return_value = (False, "Mock ERROR get_system_id_from_local") res = add_system_from_ip('192.168.0.1', 'ascodevida', add_to_database=True) self.assertFalse(res[0])
def test0008(self, **mocks): """ add remote system. Fails create_directory_for_ossec_remote """ self._init_mocks_remote_system(mocks) mocks['create_directory_for_ossec_remote'].return_value = (False, "Mock ERROR create_directory_for_ossec_remote") res = add_system_from_ip('192.168.0.1', 'ascodevida', add_to_database=True) self.assertFalse(res[0]) self.assertTrue(mocks['create_directory_for_ossec_remote'].called)
def test004(self, **mocks): """ add a remote sensor. Don't store database """ self._init_mocks_remote_sensor(mocks) res = add_system_from_ip('192.168.0.1', 'ascodevida', add_to_database=False) self.assertTrue(res[0]) # Check self.assertFalse(mocks['db_add_system'].called)
def test0007(self, **mocks): """ add remote system. Fails add_child_server """ self._init_mocks_remote_system(mocks) mocks['add_child_server'].return_value = (False, "Mock ERROR add_child_server") res = add_system_from_ip('192.168.0.1', 'ascodevida', add_to_database=True) self.assertFalse(res[0]) self.assertTrue(mocks['add_child_server'].called)
def test005(self, **mocks): """ add remote sensor. Fails ansible_get_system_info """ self._init_mocks_remote_sensor(mocks) mocks['ansible_get_system_info'].return_value = \ (False, "Mock ERROR ansible_get_system_info") res = add_system_from_ip('192.168.0.1', 'ascodevida', add_to_database=False) self.assertFalse(res[0]) self.assertTrue(mocks['ansible_get_system_info'].called)
def test0002(self, **mocks): """ test de fail call get_system_id_from_local """ self._init_mocks_remote_system(mocks) mocks['get_system_id_from_local'].return_value = ( False, "Mock ERROR get_system_id_from_local") res = add_system_from_ip('192.168.0.1', 'ascodevida', add_to_database=True) self.assertFalse(res[0])
def test0006(self, **mocks): """ add remote sensor. Fails get_sensor_id_from_sensor_ip """ self._init_mocks_remote_sensor(mocks) mocks['get_sensor_id_from_sensor_ip'].return_value = (False, None) res = add_system_from_ip('192.168.0.1', 'ascodevida', add_to_database=True) self.assertTrue(res[0]) self.assertTrue(mocks['get_sensor_id_from_sensor_ip'].called) (_, keydict) = mocks['db_add_system'].call_args self.assertTrue(keydict['sensor_id'] is None)
def add_system(): if not is_valid_ipv4(request.form['system_ip']): return make_bad_request("Bad system_ip: %s" % request.form['system_ip']) (success, system_data) = system.add_system_from_ip(request.form['system_ip'], request.form['password']) if not success: current_app.logger.error("system: add_system error: " + str(system_data)) return make_error(system_data, 500) return make_ok(**system_data)
def test0003(self, **mocks): """ Verify a remote sensor """ self._init_mocks_remote_sensor(mocks) res = add_system_from_ip('192.168.0.1', 'ascodevida', add_to_database=True) self.assertTrue(res[0]) # Check self.assertFalse(mocks['add_child_server'].called) self.assertTrue(mocks['get_sensor_id_from_sensor_ip'].called) (ordered, _) = mocks['get_sensor_id_from_sensor_ip'].call_args self.assertTrue(ordered[0] == '192.168.0.1')
def test0008(self, **mocks): """ add remote system. Fails create_directory_for_ossec_remote """ self._init_mocks_remote_system(mocks) mocks['create_directory_for_ossec_remote'].return_value = ( False, "Mock ERROR create_directory_for_ossec_remote") res = add_system_from_ip('192.168.0.1', 'ascodevida', add_to_database=True) self.assertFalse(res[0]) self.assertTrue(mocks['create_directory_for_ossec_remote'].called)
def test0007(self, **mocks): """ add remote system. Fails add_child_server """ self._init_mocks_remote_system(mocks) mocks['add_child_server'].return_value = ( False, "Mock ERROR add_child_server") res = add_system_from_ip('192.168.0.1', 'ascodevida', add_to_database=True) self.assertFalse(res[0]) self.assertTrue(mocks['add_child_server'].called)
def test0001(self, **mocks): """ test de ok call add_system_from_ip """ self._init_mocks_remote_system(mocks) res = add_system_from_ip('192.168.0.1', 'ascodevida', add_to_database=True) self.assertTrue(res[0]) # Check params (ordered, keydict) = mocks['ansible_add_system'].call_args self.assertTrue(keydict['local_system_id'] == self.uuidlocal) self.assertTrue(keydict['remote_system_ip'] == '192.168.0.1') self.assertTrue(keydict['password'] == 'ascodevida') # (ordered, keydict) = mocks['add_child_server'].call_args self.assertTrue(ordered[0] == '192.168.0.1') self.assertTrue(ordered[1] == self.system_info['server_id']) # add_db_system (ordered, keydict) = mocks['db_add_system'].call_args self.assertTrue(keydict['system_id'] == self.system_info['system_id']) self.assertTrue(keydict['name'] == self.system_info['hostname']) self.assertTrue(keydict['admin_ip'] == '192.168.0.1') self.assertTrue(keydict['profile'] == ",".join(self.system_info['profile'])) self.assertTrue(keydict['server_id'] == self.system_info['server_id']) self.assertTrue(keydict['sensor_id'] is None)