def test_get_task_status_no_filter(mock_task_db):
"""Check system's tasks (No filters)
"""
result = task.get_task_status()
cur = get_fake_database_data('schema_tasks_test.sql').cursor()
cur.execute("SELECT COUNT(DISTINCT task_id) FROM tasks")
rows = cur.fetchone()
assert result.total_affected_items == rows[0]
def test_get_task_status_sort(mock_wdb):
"""Test sort filter."""
result_asc = task.get_task_status(sort={
"fields": ["task_id"],
"order": "asc"
},
limit=10)
assert result_asc.affected_items[0]['task_id'] < result_asc.affected_items[
1]['task_id']
result_desc = task.get_task_status(sort={
"fields": ["task_id"],
"order": "desc"
},
limit=10)
assert result_desc.affected_items[0][
'task_id'] > result_desc.affected_items[1]['task_id']
assert result_asc.affected_items[0][
'task_id'] < result_desc.affected_items[0]['task_id']
def test_get_task_status_search(mock_task_db, search, total):
"""Check system's tasks (search)
Parameters
----------
search : str
Term to be search
total : int
Total records for the specific search
"""
result = task.get_task_status(search={'value': search, 'negation': 0})
assert result.total_affected_items == total
def test_get_task_status_task_list(mock_task_db, task_list, total):
"""Check system's tasks (task_list)
Parameters
----------
task_list : list
Specific task ids
total : int
Total records for agent id
"""
filters = {'task_list': task_list}
result = task.get_task_status(filters=filters)
assert result.total_affected_items == total
def test_get_task_status_offset_limit(mock_wdb, offset, limit):
"""Test if data are retrieved properly from Tasks database."""
# Check error when limit = 0
try:
result = task.get_task_status(offset=offset, limit=limit)
except WazuhError as e:
if e.code == 1406:
return
else:
raise e
# check result length
try:
assert len(result.affected_items) == limit
except AssertionError:
assert len(result.affected_items) <= 6
def test_get_task_status_select(mock_task_db, select, total, agents):
"""Check system's tasks (select)
Parameters
----------
select : list
Select which fields to return (separated by comma)
total : int
Total records for the specific search
"""
result = task.get_task_status(select=select)
assert result.total_affected_items == total
for specified_select in select:
for element in result.affected_items:
assert specified_select in element.keys()
def test_get_task_status_agent_id(mock_task_db, agent_id):
"""Check system's tasks (agent_id)
Parameters
----------
agent_id : str
Specific agent id
"""
cur = get_fake_database_data('schema_tasks_test.sql').cursor()
cur.execute("SELECT COUNT(DISTINCT task_id) FROM tasks WHERE "
f"(agent_id='{int(agent_id)}' COLLATE NOCASE)")
rows = cur.fetchone()
expected_total_items = rows[0]
filters = {'agent_id': agent_id}
result = task.get_task_status(filters=filters)
assert result.total_affected_items == expected_total_items
def test_get_task_status_module(mock_task_db, module):
"""Check system's tasks (module)
Parameters
----------
module : str
Search tasks with a specific module
"""
cur = get_fake_database_data('schema_tasks_test.sql').cursor()
cur.execute("SELECT COUNT(DISTINCT task_id) FROM tasks WHERE "
f"(module='{module}' COLLATE NOCASE)")
rows = cur.fetchone()
expected_total_items = rows[0]
filters = {'module': module}
result = task.get_task_status(filters=filters)
assert result.total_affected_items == expected_total_items
def test_get_task_status_command(mock_task_db, command):
"""Check system's tasks (command)
Parameters
----------
command : list
Search for tasks with a specific command
"""
cur = get_fake_database_data('schema_tasks_test.sql').cursor()
cur.execute("SELECT COUNT(DISTINCT task_id) FROM tasks WHERE "
f"(command='{command}' COLLATE NOCASE)")
rows = cur.fetchone()
expected_total_items = rows[0]
filters = {'command': command}
result = task.get_task_status(filters=filters)
assert result.total_affected_items == expected_total_items
def test_get_task_status_node(mock_task_db, node):
"""Check system's tasks (node)
Parameters
----------
node : str
Search for the tasks of a specific node
"""
cur = get_fake_database_data('schema_tasks_test.sql').cursor()
cur.execute("SELECT COUNT(DISTINCT task_id) FROM tasks WHERE "
f"(node='{node}' COLLATE NOCASE)")
rows = cur.fetchone()
expected_total_items = rows[0]
filters = {'node': node}
result = task.get_task_status(filters=filters)
assert result.total_affected_items == expected_total_items
def test_get_task_status_status(mock_task_db, status):
"""Check system's tasks (status)
Parameters
----------
status : str
Status of tasks to be shown
"""
cur = get_fake_database_data('schema_tasks_test.sql').cursor()
cur.execute("SELECT COUNT(DISTINCT task_id) FROM tasks WHERE "
f"(status='{status}' COLLATE NOCASE)")
rows = cur.fetchone()
expected_total_items = rows[0]
filters = {'status': status}
result = task.get_task_status(filters=filters)
assert result.total_affected_items == expected_total_items
