From 4043672bbedeb6639a0d4d6af16c855f4c19dc6e Mon Sep 17 00:00:00 2001
From: Prabhat Sharma <ptsharma@amazon.com>
Date: Fri, 5 Apr 2024 13:52:21 +0530
Subject: [PATCH] Fixed tests

Signed-off-by: Prabhat Sharma <ptsharma@amazon.com>
---
 .../containers/CreateRollupForm/CreateRollupForm.tsx          | 4 ++--
 .../containers/CreateTransformForm/CreateTransformForm.tsx    | 4 ++--
 public/pages/Main/Main.tsx                                    | 1 -
 3 files changed, 4 insertions(+), 5 deletions(-)

diff --git a/public/pages/CreateRollup/containers/CreateRollupForm/CreateRollupForm.tsx b/public/pages/CreateRollup/containers/CreateRollupForm/CreateRollupForm.tsx
index d8263c6b6..c7cc2ea8b 100644
--- a/public/pages/CreateRollup/containers/CreateRollupForm/CreateRollupForm.tsx
+++ b/public/pages/CreateRollup/containers/CreateRollupForm/CreateRollupForm.tsx
@@ -195,7 +195,7 @@ export class CreateRollupForm extends Component<CreateRollupFormProps, CreateRol
     const dataSourceReadOnly = this.props.dataSourceReadOnly;
     const setDataSourceReadOnly = this.props.setDataSourceReadOnly;
 
-    if (!dataSourceReadOnly) {
+    if (this.props.multiDataSourceEnabled && !dataSourceReadOnly) {
       setDataSourceReadOnly(true);
     }
 
@@ -266,7 +266,7 @@ export class CreateRollupForm extends Component<CreateRollupFormProps, CreateRol
       const dataSourceReadOnly = this.props.dataSourceReadOnly;
       const setDataSourceReadOnly = this.props.setDataSourceReadOnly;
 
-      if (dataSourceReadOnly) {
+      if (this.props.multiDataSourceEnabled && dataSourceReadOnly) {
         setDataSourceReadOnly(false);
       }
     }
diff --git a/public/pages/CreateTransform/containers/CreateTransformForm/CreateTransformForm.tsx b/public/pages/CreateTransform/containers/CreateTransformForm/CreateTransformForm.tsx
index 7c0173f26..1c788e442 100644
--- a/public/pages/CreateTransform/containers/CreateTransformForm/CreateTransformForm.tsx
+++ b/public/pages/CreateTransform/containers/CreateTransformForm/CreateTransformForm.tsx
@@ -222,7 +222,7 @@ export class CreateTransformForm extends Component<CreateTransformFormProps, Cre
 
     const dataSourceReadOnly = this.props.dataSourceReadOnly;
     const setDataSourceReadOnly = this.props.setDataSourceReadOnly;
-    if (!dataSourceReadOnly) {
+    if (this.props.multiDataSourceEnabled && !dataSourceReadOnly) {
       setDataSourceReadOnly(true);
     }
 
@@ -282,7 +282,7 @@ export class CreateTransformForm extends Component<CreateTransformFormProps, Cre
       const dataSourceReadOnly = this.props.dataSourceReadOnly;
       const setDataSourceReadOnly = this.props.setDataSourceReadOnly;
 
-      if (dataSourceReadOnly) {
+      if (this.props.multiDataSourceEnabled && dataSourceReadOnly) {
         setDataSourceReadOnly(false);
       }
     }
diff --git a/public/pages/Main/Main.tsx b/public/pages/Main/Main.tsx
index c8fa7c398..c7da806dc 100644
--- a/public/pages/Main/Main.tsx
+++ b/public/pages/Main/Main.tsx
@@ -61,7 +61,6 @@ import CreateComposableTemplate from "../CreateComposableTemplate";
 import { DataSourceMenuContext, DataSourceMenuProperties } from "../../services/DataSourceMenuContext";
 import queryString from "query-string";
 import { DataSourceManagementPluginSetup } from "../../../../../src/plugins/data_source_management/public";
-import { bool } from "aws-sdk/clients/signer";
 
 enum Navigation {
   IndexManagement = "Index Management",