Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
41 changes: 41 additions & 0 deletions src/gmp/commands/__tests__/tasks.test.ts
Original file line number Diff line number Diff line change
Expand Up @@ -49,6 +49,24 @@ describe('TasksCommand tests', () => {
expect(result.data).toEqual([new Task({id: '3', name: 'Custom Task'})]);
});

test('should fetch tasks with schedules only', async () => {
const response = createEntitiesResponse('task', [
{_id: '3', name: 'Custom Task'},
]);
const fakeHttp = createHttp(response);

const cmd = new TasksCommand(fakeHttp);
const result = await cmd.get({schedulesOnly: true});
expect(fakeHttp.request).toHaveBeenCalledWith('get', {
args: {
cmd: 'get_tasks',
usage_type: 'scan',
schedules_only: 1,
},
});
expect(result.data).toEqual([new Task({id: '3', name: 'Custom Task'})]);
});

test('should fetch all tasks', async () => {
const response = createEntitiesResponse('task', [
{_id: '4', name: 'All Tasks 1'},
Expand All @@ -67,6 +85,29 @@ describe('TasksCommand tests', () => {
]);
});

test('should fetch all tasks with schedules only', async () => {
const response = createEntitiesResponse('task', [
{_id: '4', name: 'All Tasks 1'},
{_id: '5', name: 'All Tasks 2'},
]);
const fakeHttp = createHttp(response);

const cmd = new TasksCommand(fakeHttp);
const result = await cmd.getAll({schedulesOnly: true});
expect(fakeHttp.request).toHaveBeenCalledWith('get', {
args: {
cmd: 'get_tasks',
filter: 'first=1 rows=-1',
usage_type: 'scan',
schedules_only: 1,
},
});
expect(result.data).toEqual([
new Task({id: '4', name: 'All Tasks 1'}),
new Task({id: '5', name: 'All Tasks 2'}),
]);
});

test('should fetch severity aggregates', async () => {
const response = createAggregatesResponse({});
const fakeHttp = createHttp(response);
Expand Down
14 changes: 11 additions & 3 deletions src/gmp/commands/tasks.ts
Original file line number Diff line number Diff line change
Expand Up @@ -10,7 +10,8 @@ import type Http from 'gmp/http/http';
import type Filter from 'gmp/models/filter';
import {type Element} from 'gmp/models/model';
import Task, {type TaskElement} from 'gmp/models/task';
import {type YesNo} from 'gmp/parser';
import {parseYesNo, type YesNo} from 'gmp/parser';
import {isDefined} from 'gmp/utils/identity';

interface GetTasksResponse extends Element {
apply_overrides: YesNo;
Expand Down Expand Up @@ -39,6 +40,7 @@ interface TasksCommandWithFilterParam {

interface TasksCommandGetParams {
filter?: Filter | string;
schedulesOnly?: boolean;
}

class TasksCommand extends EntitiesCommand<Task, GetTasksResponse> {
Expand All @@ -52,10 +54,16 @@ class TasksCommand extends EntitiesCommand<Task, GetTasksResponse> {
}

async get(
{filter}: TasksCommandGetParams = {},
{filter, schedulesOnly}: TasksCommandGetParams = {},
options?: HttpCommandOptions,
) {
const params = {filter, usage_type: 'scan'};
const params = {
filter,
usage_type: 'scan',
schedules_only: isDefined(schedulesOnly)
? parseYesNo(schedulesOnly)
: undefined,
};
const response = await this.httpGetWithTransform(params, options);
const {
entities,
Expand Down
Loading