test: update strict subset match test

Signed-off-by: mao3267 <[email protected]>
flyteorg · Nov 15, 2024 · b8f38a7 · b8f38a7
1 parent 56623e3
commit b8f38a7
Showing 1 changed file with 44 additions and 45 deletions.
diff --git a/flytepropeller/pkg/compiler/validators/typing_test.go b/flytepropeller/pkg/compiler/validators/typing_test.go
@@ -832,9 +832,8 @@ func TestMapCasting(t *testing.T) {
 				a: int
 
 			@dataclass
-			class A:
-				a: int
-				b: Optional[str]
+			class B(A):
+				b: Optional[str] = None
 		*/
 		castable := AreTypesCastable(
 			&core.LiteralType{
@@ -851,7 +850,7 @@ func TestMapCasting(t *testing.T) {
 							},
 						},
 						"title": &structpb.Value{
-							Kind: &structpb.Value_StringValue{StringValue: "A"},
+							Kind: &structpb.Value_StringValue{StringValue: "B"},
 						},
 						"properties": &structpb.Value{
 							Kind: &structpb.Value_StructValue{
@@ -868,6 +867,47 @@ func TestMapCasting(t *testing.T) {
 												},
 											},
 										},
+										"b": {
+											Kind: &structpb.Value_StructValue{
+												StructValue: &structpb.Struct{
+													Fields: map[string]*structpb.Value{
+														"default": {
+															Kind: &structpb.Value_NullValue{},
+														},
+														"anyOf": {
+															Kind: &structpb.Value_ListValue{
+																ListValue: &structpb.ListValue{
+																	Values: []*structpb.Value{
+																		{
+																			Kind: &structpb.Value_StructValue{
+																				StructValue: &structpb.Struct{
+																					Fields: map[string]*structpb.Value{
+																						"type": {
+																							Kind: &structpb.Value_StringValue{StringValue: "string"},
+																						},
+																					},
+																				},
+																			},
+																		},
+																		{
+																			Kind: &structpb.Value_StructValue{
+																				StructValue: &structpb.Struct{
+																					Fields: map[string]*structpb.Value{
+																						"type": {
+																							Kind: &structpb.Value_StringValue{StringValue: "null"},
+																						},
+																					},
+																				},
+																			},
+																		},
+																	},
+																},
+															},
+														},
+													},
+												},
+											},
+										},
 									},
 								},
 							},
@@ -912,47 +952,6 @@ func TestMapCasting(t *testing.T) {
 												},
 											},
 										},
-										"b": {
-											Kind: &structpb.Value_StructValue{
-												StructValue: &structpb.Struct{
-													Fields: map[string]*structpb.Value{
-														"default": {
-															Kind: &structpb.Value_NullValue{},
-														},
-														"anyOf": {
-															Kind: &structpb.Value_ListValue{
-																ListValue: &structpb.ListValue{
-																	Values: []*structpb.Value{
-																		{
-																			Kind: &structpb.Value_StructValue{
-																				StructValue: &structpb.Struct{
-																					Fields: map[string]*structpb.Value{
-																						"type": {
-																							Kind: &structpb.Value_StringValue{StringValue: "string"},
-																						},
-																					},
-																				},
-																			},
-																		},
-																		{
-																			Kind: &structpb.Value_StructValue{
-																				StructValue: &structpb.Struct{
-																					Fields: map[string]*structpb.Value{
-																						"type": {
-																							Kind: &structpb.Value_StringValue{StringValue: "null"},
-																						},
-																					},
-																				},
-																			},
-																		},
-																	},
-																},
-															},
-														},
-													},
-												},
-											},
-										},
 									},
 								},
 							},