From a4f08e3fccaf247695aef9e9cd46e7b756df2500 Mon Sep 17 00:00:00 2001
From: Nikita Nek Dudnik <nikdudnik@gmail.com>
Date: Fri, 5 Jun 2020 16:29:46 +0200
Subject: [PATCH] fix: type checking

---
 .../RegisterComponentPage/RegisterComponentPage.test.tsx        | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/plugins/register-component/src/components/RegisterComponentPage/RegisterComponentPage.test.tsx b/plugins/register-component/src/components/RegisterComponentPage/RegisterComponentPage.test.tsx
index ad47c3a41b..ca463550d1 100644
--- a/plugins/register-component/src/components/RegisterComponentPage/RegisterComponentPage.test.tsx
+++ b/plugins/register-component/src/components/RegisterComponentPage/RegisterComponentPage.test.tsx
@@ -30,6 +30,8 @@ const catalogApi: jest.Mocked<typeof catalogApiRef.T> = {
   getEntities: jest.fn(),
   getEntityByName: jest.fn(),
   getLocationByEntity: jest.fn(),
+  getLocationById: jest.fn(),
+  getEntitiesByLocationId: jest.fn(),
 };
 
 const setup = () => ({