use TS for agent mock

elastic · Dec 6, 2021 · df6e05d · df6e05d
1 parent 5ed187b
commit df6e05d
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/packages/kbn-test/BUILD.bazel b/packages/kbn-test/BUILD.bazel
@@ -76,6 +76,7 @@ TYPES_DEPS = [
   "//packages/kbn-i18n-react:npm_module_types",
   "//packages/kbn-utils",
   "@npm//@elastic/elasticsearch",
+  "@npm//elastic-apm-node",
   "@npm//del",
   "@npm//form-data",
   "@npm//jest",

diff --git a/...kbn-test/src/jest/mocks/apm_agent_mock.js → ...kbn-test/src/jest/mocks/apm_agent_mock.ts b/...kbn-test/src/jest/mocks/apm_agent_mock.js → ...kbn-test/src/jest/mocks/apm_agent_mock.ts
@@ -6,14 +6,14 @@
  * Side Public License, v 1.
  */
 
+import type { Agent } from 'elastic-apm-node';
+
 /**
  * `elastic-apm-node` patches the runtime at import time
  *  causing memory leak with jest module sandbox, so it
  *  needs to be mocked for tests
  */
-
-/* eslint-disable no-undef */
-const agent = {
+const agent: jest.Mocked<Agent> = {
   start: jest.fn().mockImplementation(() => agent),
   isStarted: jest.fn().mockReturnValue(false),
   getServiceName: jest.fn().mockReturnValue('mock-service'),
@@ -59,4 +59,5 @@ const agent = {
   },
 };
 
-module.exports = agent;
+// eslint-disable-next-line import/no-default-export
+export default agent;