From a5ee50cfc982a565bcfb11c55203c73b0e71f40b Mon Sep 17 00:00:00 2001
From: Mateusz Kwasniewski <kwasniewski.mateusz@gmail.com>
Date: Wed, 21 Jun 2023 08:17:53 +0200
Subject: [PATCH] test: advanced playground error (#4023)

---
 .../Playground/AdvancedPlayground.test.tsx    | 43 +++++++++++++++++--
 .../Playground/AdvancedPlayground.tsx         |  2 +-
 2 files changed, 41 insertions(+), 4 deletions(-)

diff --git a/frontend/src/component/playground/Playground/AdvancedPlayground.test.tsx b/frontend/src/component/playground/Playground/AdvancedPlayground.test.tsx
index e0927bc9b6..b423c14f5d 100644
--- a/frontend/src/component/playground/Playground/AdvancedPlayground.test.tsx
+++ b/frontend/src/component/playground/Playground/AdvancedPlayground.test.tsx
@@ -3,8 +3,10 @@ import { render } from 'utils/testRenderer';
 import { UIProviderContainer } from '../../providers/UIProvider/UIProviderContainer';
 import AdvancedPlayground from './AdvancedPlayground';
 import { createLocalStorage } from 'utils/createLocalStorage';
+import { testServerRoute, testServerSetup } from 'utils/testServer';
+import userEvent from '@testing-library/user-event';
 
-const testComponent = (
+const testDisplayComponent = (
     <UIProviderContainer>
         <AdvancedPlayground
             FormComponent={props => (
@@ -22,6 +24,18 @@ const testComponent = (
     </UIProviderContainer>
 );
 
+const testEvaluateComponent = (
+    <UIProviderContainer>
+        <AdvancedPlayground
+            FormComponent={props => (
+                <form onSubmit={props.onSubmit}>
+                    <button>Submit</button>
+                </form>
+            )}
+        />
+    </UIProviderContainer>
+);
+
 afterEach(() => {
     const { setValue } = createLocalStorage('AdvancedPlayground:v1', {});
     setValue({});
@@ -35,7 +49,7 @@ test('should fetch initial form data from local storage', async () => {
         context: { userId: '1' },
     });
 
-    render(testComponent);
+    render(testDisplayComponent);
 
     expect(screen.getByText('Unleash playground')).toBeInTheDocument();
     expect(screen.getByText('["projectA","projectB"]')).toBeInTheDocument();
@@ -53,7 +67,7 @@ test('should fetch initial form data from url', async () => {
         context: { userId: '1' },
     });
 
-    render(testComponent, {
+    render(testDisplayComponent, {
         route: '/playground?context=customContext&environments=customEnv&projects=urlProject&sort=name',
     });
 
@@ -62,3 +76,26 @@ test('should fetch initial form data from url', async () => {
     expect(screen.getByText('["customEnv"]')).toBeInTheDocument();
     expect(screen.getByText('"customContext"')).toBeInTheDocument();
 });
+
+const server = testServerSetup();
+
+test('should display error on submit', async () => {
+    testServerRoute(
+        server,
+        '/api/admin/playground/advanced',
+        {
+            name: 'BadDataError',
+            details: [{ message: 'some error about too many items' }],
+        },
+        'post',
+        400
+    );
+
+    render(testEvaluateComponent);
+
+    const user = userEvent.setup();
+    const submitButton = screen.getByText('Submit');
+    user.click(submitButton);
+
+    await screen.findByText('some error about too many items');
+});
diff --git a/frontend/src/component/playground/Playground/AdvancedPlayground.tsx b/frontend/src/component/playground/Playground/AdvancedPlayground.tsx
index 23f94e5743..0e1d662c9b 100644
--- a/frontend/src/component/playground/Playground/AdvancedPlayground.tsx
+++ b/frontend/src/component/playground/Playground/AdvancedPlayground.tsx
@@ -59,7 +59,7 @@ export const AdvancedPlayground: VFC<{
     const { evaluateAdvancedPlayground, loading } = usePlaygroundApi();
 
     useEffect(() => {
-        if (environments.length === 0) {
+        if (environments?.length === 0) {
             setEnvironments([resolveDefaultEnvironment(availableEnvironments)]);
         }
     }, [availableEnvironments]);