github.com/hasnat/dolt/go@v0.0.0-20210628190320-9eb5d843fbb7/libraries/doltcore/rowconv/field_mapping_test.go

// Copyright 2019 Dolthub, Inc.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

package rowconv

import (
	"reflect"
	"testing"

	"github.com/dolthub/dolt/go/libraries/doltcore/schema"
	"github.com/dolthub/dolt/go/libraries/utils/filesys"
	"github.com/dolthub/dolt/go/store/types"
)

var fieldsA = schema.NewColCollection(
	schema.NewColumn("a", 0, types.StringKind, true),
	schema.NewColumn("b", 1, types.StringKind, false),
	schema.NewColumn("c", 2, types.StringKind, false))

var fieldsB = schema.NewColCollection(
	schema.NewColumn("a", 0, types.StringKind, true),
	schema.NewColumn("b", 1, types.StringKind, false))

var fieldsC = schema.NewColCollection(
	schema.NewColumn("key", 3, types.UUIDKind, true),
	schema.NewColumn("value", 4, types.StringKind, false))

var fieldsCNoPK = schema.NewColCollection(
	schema.NewColumn("key", 3, types.UUIDKind, true),
	schema.NewColumn("value", 4, types.StringKind, false))

var fieldsD = schema.NewColCollection(
	schema.NewColumn("key", 3, types.StringKind, true),
	schema.NewColumn("value", 4, types.StringKind, false))

var schemaA = schema.MustSchemaFromCols(fieldsA)
var schemaB = schema.MustSchemaFromCols(fieldsB)
var schemaC = schema.MustSchemaFromCols(fieldsC)
var schemaCNoPK = schema.MustSchemaFromCols(fieldsCNoPK)
var schemaD = schema.MustSchemaFromCols(fieldsD)

func TestFieldMapping(t *testing.T) {
	tests := []struct {
		mappingJSON string
		inSch       schema.Schema
		outSch      schema.Schema
		expectErr   bool
		expected    map[uint64]uint64
		identity    bool
	}{
		{"", schemaA, schemaA, false, map[uint64]uint64{0: 0, 1: 1, 2: 2}, true},
		{"", schemaA, schemaB, false, map[uint64]uint64{0: 0, 1: 1}, false},
		{"", schemaB, schemaA, false, map[uint64]uint64{0: 0, 1: 1}, false},
		{"", schemaA, schemaC, true, nil, false},
		{`{"invalid_json": }`, schemaA, schemaC, true, nil, false},
		{`{"b": "value"}`, schemaA, schemaC, false, map[uint64]uint64{1: 4}, false},
		{`{"c": "key", "b": "value"}`, schemaA, schemaC, false, map[uint64]uint64{2: 3, 1: 4}, false},
		{`{"a": "key", "b": "value"}`, schemaA, schemaC, false, map[uint64]uint64{0: 3, 1: 4}, false},
		{`{"a": "key", "b": "value"}`, schemaB, schemaC, false, map[uint64]uint64{0: 3, 1: 4}, false},
		{`{"a": "key", "b": "value"}`, schemaB, schemaCNoPK, false, map[uint64]uint64{0: 3, 1: 4}, false},
		{`{"a": "key", "b": "value"}`, schemaB, schemaD, false, map[uint64]uint64{0: 3, 1: 4}, true},
	}

	for _, test := range tests {
		fs := filesys.NewInMemFS([]string{"/"}, nil, "/")

		mappingFile := ""
		if test.mappingJSON != "" {
			mappingFile = "mapping.json"
			fs.WriteFile(mappingFile, []byte(test.mappingJSON))
		}

		var mapping *FieldMapping
		var err error
		if mappingFile != "" {
			var nm NameMapper
			nm, err = NameMapperFromFile(mappingFile, fs)
			if err == nil {
				mapping, err = NameMapping(test.inSch, test.outSch, nm)
			}
		} else {
			mapping, err = TagMapping(test.inSch, test.outSch)
		}

		if (err != nil) != test.expectErr {
			if test.expectErr {
				t.Fatal("Expected an error that didn't come.")
			} else {
				t.Fatal("Unexpected error creating mapping.", err)
			}
		}

		if !test.expectErr {
			if !reflect.DeepEqual(mapping.SrcToDest, test.expected) {
				t.Error("Mapping does not match expected.  Expected:", test.expected, "Actual:", mapping.SrcToDest)
			}

			//if test.identity != mapping.IsIdentityMapping() {
			//	t.Error("identity expected", test.identity, "actual:", !test.identity)
			//}
		}
	}
}