Adjust test for edit_config_family_all

greenbone · Oct 31, 2023 · 5328e8c · 5328e8c
1 parent 1237e39
commit 5328e8c
Show file tree

Hide file tree

Showing 2 changed files with 31 additions and 18 deletions.
diff --git a/src/gmp/commands/__tests__/scanconfig.js b/src/gmp/commands/__tests__/scanconfig.js
@@ -18,6 +18,7 @@
 import {
  createEntityResponse,
  createHttp,
+ createHttpMany,
  createActionResultResponse,
  createResponse,
 } from '../testing';
@@ -296,27 +297,30 @@ describe('ScanConfigCommand tests', () => {
  },
  ],
  },
- all: {
- get_nvts_response: {
- nvt: [
- {
- _oid: 1,
- cvss_base: 1.1,
- },
- {
- _oid: 2,
- cvss_base: 2.2,
- },
- {
- _oid: 3,
- cvss_base: 3.3,
- },
- ],
- },
+ },
+ });
+ const responseAll = createResponse({
+ get_config_family_response: {
+ get_nvts_response: {
+ nvt: [
+ {
+ _oid: 1,
+ cvss_base: 1.1,
+ },
+ {
+ _oid: 2,
+ cvss_base: 2.2,
+ },
+ {
+ _oid: 3,
+ cvss_base: 3.3,
+ },
+ ],
  },
  },
  });
- const fakeHttp = createHttp(response);
+ const responses = [response, responseAll];
+ const fakeHttp = createHttpMany(responses);
 
  expect.hasAssertions();
 

diff --git a/src/gmp/commands/testing.js b/src/gmp/commands/testing.js
@@ -72,4 +72,13 @@ export const createHttp = response => ({
  request: jest.fn().mockReturnValue(Promise.resolve(response)),
 });
 
+export const createHttpMany = responses => {
+ let i = 0;
+ return {
+ request: jest
+ .fn()
+ .mockImplementation(() => Promise.resolve(responses[i++])),
+ };
+};
+
 // vim: set ts=2 sw=2 tw=80: