Add associative list key schema evolution changes

Author: jpbetz

merge/schema_change_test.go

@@ -252,3 +252,244 @@ func TestAtomicToGranularSchemaChanges(t *testing.T) {
 		})
 	}
 }
+
+var associativeListParserOld = func() *typed.Parser {
+	oldParser, err := typed.NewParser(`types:
+- name: v1
+  map:
+    fields:
+      - name: list
+        type:
+          namedType: associativeList
+- name: associativeList
+  list:
+    elementType:
+      namedType: myElement
+    elementRelationship: associative
+    keys:
+    - name
+- name: myElement
+  map:
+    fields:
+    - name: name
+      type:
+        scalar: string
+    - name: id
+      type:
+        scalar: numeric
+    - name: value
+      type:
+        scalar: numeric
+`)
+	if err != nil {
+		panic(err)
+	}
+	return oldParser
+}()
+
+var associativeListParserNewOptionalKey = func() *typed.Parser {
+	newParser, err := typed.NewParser(`types:
+- name: v1
+  map:
+    fields:
+      - name: list
+        type:
+          namedType: associativeList
+- name: associativeList
+  list:
+    elementType:
+      namedType: myElement
+    elementRelationship: associative
+    keys:
+    - name
+    - id
+- name: myElement
+  map:
+    fields:
+    - name: name
+      type:
+        scalar: string
+    - name: id
+      type:
+        scalar: numeric
+    - name: value
+      type:
+        scalar: numeric
+`)
+	if err != nil {
+		panic(err)
+	}
+	return newParser
+}()
+
+var associativeListParserNewKeyWithDefault = func() *typed.Parser {
+	newParser, err := typed.NewParser(`types:
+- name: v1
+  map:
+    fields:
+      - name: list
+        type:
+          namedType: associativeList
+- name: associativeList
+  list:
+    elementType:
+      namedType: myElement
+    elementRelationship: associative
+    keys:
+    - name
+    - id
+- name: myElement
+  map:
+    fields:
+    - name: name
+      type:
+        scalar: string
+    - name: id
+      type:
+        scalar: numeric
+      default: 1
+    - name: value
+      type:
+        scalar: numeric
+`)
+	if err != nil {
+		panic(err)
+	}
+	return newParser
+}()
+
+func TestAssociativeListSchemaChanges(t *testing.T) {
+	tests := map[string]TestCase{
+		"new required key with default": {
+			Ops: []Operation{
+				Apply{
+					Manager: "one",
+					Object: `
+						list:
+						- name: a
+						  value: 1
+						- name: b
+						  value: 1
+						- name: c
+						  value: 1
+					`,
+					APIVersion: "v1",
+				},
+				ChangeParser{Parser: associativeListParserNewKeyWithDefault},
+				Apply{
+					Manager: "one",
+					Object: `
+						list:
+						- name: a
+						  value: 2
+						- name: b
+						  id: 1
+						  value: 2
+						- name: c
+						  value: 1
+						- name: c
+						  id: 2
+						  value: 2
+						- name: c
+						  id: 3
+						  value: 3
+					`,
+					APIVersion: "v1",
+				},
+			},
+			Object: `
+				list:
+				- name: a
+				  value: 2
+				- name: b
+				  id: 1
+				  value: 2
+				- name: c
+				  value: 1
+				- name: c
+				  id: 2
+				  value: 2
+				- name: c
+				  id: 3
+				  value: 3
+			`,
+			APIVersion: "v1",
+			Managed: fieldpath.ManagedFields{
+				"one": fieldpath.NewVersionedSet(_NS(
+					_P("list", _KBF("name", "a", "id", float64(1))),
+					_P("list", _KBF("name", "a", "id", float64(1)), "name"),
+					_P("list", _KBF("name", "a", "id", float64(1)), "value"),
+					_P("list", _KBF("name", "b", "id", float64(1))),
+					_P("list", _KBF("name", "b", "id", float64(1)), "name"),
+					_P("list", _KBF("name", "b", "id", float64(1)), "id"),
+					_P("list", _KBF("name", "b", "id", float64(1)), "value"),
+					_P("list", _KBF("name", "c", "id", float64(1))),
+					_P("list", _KBF("name", "c", "id", float64(1)), "name"),
+					_P("list", _KBF("name", "c", "id", float64(1)), "value"),
+					_P("list", _KBF("name", "c", "id", float64(2))),
+					_P("list", _KBF("name", "c", "id", float64(2)), "name"),
+					_P("list", _KBF("name", "c", "id", float64(2)), "id"),
+					_P("list", _KBF("name", "c", "id", float64(2)), "value"),
+					_P("list", _KBF("name", "c", "id", float64(3))),
+					_P("list", _KBF("name", "c", "id", float64(3)), "name"),
+					_P("list", _KBF("name", "c", "id", float64(3)), "id"),
+					_P("list", _KBF("name", "c", "id", float64(3)), "value"),
+				), "v1", true),
+			},
+		},
+		"new optional key": {
+			Ops: []Operation{
+				Apply{
+					Manager: "one",
+					Object: `
+						list:
+						- name: a
+						  value: 1
+					`,
+					APIVersion: "v1",
+				},
+				ChangeParser{Parser: associativeListParserNewOptionalKey},
+				Apply{
+					Manager: "one",
+					Object: `
+						list:
+						- name: a
+						  value: 2
+						- name: a
+						  id: 1
+						  value: 1
+					`,
+					APIVersion: "v1",
+				},
+			},
+			Object: `
+				list:
+				- name: a
+				  value: 2
+				- name: a
+				  id: 1
+				  value: 1
+			`,
+			APIVersion: "v1",
+			Managed: fieldpath.ManagedFields{
+				"one": fieldpath.NewVersionedSet(_NS(
+					_P("list", _KBF("name", "a")),
+					_P("list", _KBF("name", "a"), "name"),
+					_P("list", _KBF("name", "a"), "value"),
+					_P("list", _KBF("name", "a", "id", float64(1))),
+					_P("list", _KBF("name", "a", "id", float64(1)), "name"),
+					_P("list", _KBF("name", "a", "id", float64(1)), "id"),
+					_P("list", _KBF("name", "a", "id", float64(1)), "value"),
+				), "v1", true),
+			},
+		},
+	}
+
+	for name, test := range tests {
+		t.Run(name, func(t *testing.T) {
+			if err := test.Test(associativeListParserOld); err != nil {
+				t.Fatal(err)
+			}
+		})
+	}
+}