fix: failing unit tests

Signed-off-by: Pianist038801 <[email protected]>
flyteorg · Aug 24, 2021 · 2d6eb33 · 2d6eb33
1 parent 7431975
commit 2d6eb33
Showing 1 changed file with 14 additions and 5 deletions.
diff --git a/src/components/Launch/LaunchForm/StructInput.tsx b/src/components/Launch/LaunchForm/StructInput.tsx
@@ -38,11 +38,20 @@ export const StructInput: React.FC<InputProps> = props => {
     let jsonFormRenderable = false;
     let parsedJson: PrimitiveType = {};
 
-    if (literalType?.metadata) {
-        parsedJson = protobufValueToPrimitive(
-            literalType.metadata.fields.definitions
+    if (literalType?.metadata?.fields?.definitions?.structValue?.fields) {
+        const keys = Object.keys(
+            literalType?.metadata?.fields?.definitions?.structValue?.fields
         );
-        if (parsedJson.DatumSchema) jsonFormRenderable = true;
+
+        if (keys[0]) {
+            parsedJson = protobufValueToPrimitive(
+                literalType.metadata.fields.definitions.structValue.fields[
+                    `${keys[0]}`
+                ]
+            );
+
+            if (parsedJson) jsonFormRenderable = true;
+        }
     }
 
     return jsonFormRenderable ? (
@@ -54,7 +63,7 @@ export const StructInput: React.FC<InputProps> = props => {
                 />
                 <CardContent>
                     <Form
-                        schema={parsedJson.DatumSchema}
+                        schema={JSON.parse(JSON.stringify(parsedJson))}
                         formData={paramData}
                         onChange={onFormChange}
                     >