Function TestParseContractV2

app/cli/pkg/action/attestation_init_test.go:258–340

(t *testing.T)

}

func TestParseContractV2(t *testing.T) {
	testCases := []struct {
		name           string
		contractFile   string
		format         pb.WorkflowContractVersionItem_RawBody_Format
		expectV2Schema bool
		expectName     string
		expectError    bool
	}{
		{
			name:           "valid V2 YAML contract",
			contractFile:   "testdata/contract_v2.yaml",
			format:         pb.WorkflowContractVersionItem_RawBody_FORMAT_YAML,
			expectV2Schema: true,
			expectName:     "test-contract-v2",
		},
		{
			name:           "V1 contract should fail V2 parsing",
			contractFile:   "testdata/contract_v1.yaml",
			format:         pb.WorkflowContractVersionItem_RawBody_FORMAT_YAML,
			expectV2Schema: false,
		},
		{
			name:           "invalid contract data",
			contractFile:   "testdata/invalid_contract.yaml",
			format:         pb.WorkflowContractVersionItem_RawBody_FORMAT_YAML,
			expectV2Schema: false,
		},
		{
			name:           "nil raw contract",
			contractFile:   "",
			format:         pb.WorkflowContractVersionItem_RawBody_FORMAT_YAML,
			expectV2Schema: false,
		},
	}

	for _, tc := range testCases {
		t.Run(tc.name, func(t *testing.T) {
			var rawContract *pb.WorkflowContractVersionItem_RawBody

			if tc.contractFile != "" {
				// Load contract file
				data, err := os.ReadFile(tc.contractFile)
				if err != nil {
					if tc.expectV2Schema {
						require.NoError(t, err, "Failed to load contract file")
					}
					// For non-existent files, test with nil
					rawContract = nil
				} else {
					rawContract = &pb.WorkflowContractVersionItem_RawBody{
						Body:   data,
						Format: tc.format,
					}
				}
			}

			result := parseContractV2(rawContract)