From 2b2775a1b7cd8d6d597253654d6f72f74247d8fa Mon Sep 17 00:00:00 2001
From: Narek Hovhannisyan <narekgilmour@gmail.com>
Date: Mon, 4 Dec 2023 12:52:35 +0400
Subject: [PATCH] mocks: tune model universe mock.

---
 src/__mocks__/model-universe/index.ts | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/src/__mocks__/model-universe/index.ts b/src/__mocks__/model-universe/index.ts
index 5f0facb9f..ec8be30a3 100644
--- a/src/__mocks__/model-universe/index.ts
+++ b/src/__mocks__/model-universe/index.ts
@@ -12,7 +12,11 @@ export class MockModel implements ModelPluginInterface {
   configure(): Promise<ModelPluginInterface> {
     return Promise.resolve(this);
   }
-  execute(): Promise<any[]> {
+  execute(inputs: any): Promise<any[]> {
+    if (inputs[0].carbon) {
+      return Promise.resolve(inputs);
+    }
+
     return Promise.resolve([{data: 'mock-data'}]);
   }
 }