Esempio n. 1
0
 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)
Esempio n. 2
0
 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])
Esempio n. 3
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)
Esempio n. 4
0
 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)
Esempio n. 5
0
 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)
Esempio n. 6
0
 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)
Esempio n. 7
0
 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])
Esempio n. 8
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)
Esempio n. 9
0
 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)
Esempio n. 10
0
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)
Esempio n. 11
0
 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')
Esempio n. 12
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)
Esempio n. 13
0
 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)
Esempio n. 14
0
 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)
Esempio n. 15
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)
Esempio n. 16
0
 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')
Esempio n. 17
0
 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)