Mocking ApiGatewayManagementApi not working

[kstro21]

Here is a reduced example of what I'm trying to do.

I have a class with a function that looks like this:

// at the top of the file
const ApiGatewayManagementApi = require('aws-sdk/clients/apigatewaymanagementapi');

// function definition
async sendSummary(summary, { endpoint, connectionId }) {
    const apiGwManagementApi = new ApiGatewayManagementApi({ endpoint });

    return apiGwManagementApi
      .postToConnection({
        ConnectionId: connectionId,
        Data: JSON.stringify(summary),
      })
      .promise()
      .catch((err) => {
        console.error('Error while sending', err);
        throw err;
      });
  }

That works as expected. Now comes the part of the test

const AWS = require('aws-sdk-mock');

it('send summary', async () => {
    AWS.mock('ApiGatewayManagementApi', 'postToConnection', function (
      params,
      callback
    ) {
      expect(params).toEqual(
      jasmine.objectContaining({
         ConnectionId: connectionId,
        Data: JSON.stringify(summary),
      })
    );
      callback(null, {});
    });

    const { service } = require('../my-service');

    await expectAsync(
      service.sendSummary(summaryMock, connectionMock)
    ).toBeResolved();

    AWS.restore('ApiGatewayManagementApi');
  });

But when running the tests it throws the error

UnknownEndpoint: Inaccessible host: `some.execute-api.us-east-1.amazonaws.com'

Seems like the mock is not working. Any help is appreciated.

[sukruavcuoglu]

from READ.me

"Also note that if you initialise an AWS service inside a callback from an async function inside the handler function, that won't work either."

[kstro21]

So, @sukruavcuoglu, it is not possible?

[nmq64]

The solution is to replace this
const ApiGatewayManagementApi = require('aws-sdk/clients/apigatewaymanagementapi');
and this
const apiGwManagementApi = new ApiGatewayManagementApi({ endpoint });

with this
const AWS = require('aws-sdk');
and this
const apiGwManagementApi = new AWS.ApiGatewayManagementApi(({ endpoint });