Function TestEnumMutator

controller/mutators/etcd/enum_mutator_test.go:28–95

(t *testing.T)

)

func TestEnumMutator(t *testing.T) {
	testTable := metaCom.Table{
		Name: "test",
		Columns: []metaCom.Column{
			{
				Name:    "c1",
				Type:    metaCom.BigEnum,
				Deleted: false,
			},
		},
	}

	t.Run("Extend and get enum cases", func(t *testing.T) {
		// test setup
		txnStore := mem.NewStore()

		_, err := txnStore.Set(utils.EnumNodeListKey("ns1", "test", 0, 0), &pb.EnumNodeList{
			NumEnumNodes: 1,
		})
		assert.NoError(t, err)
		_, err = txnStore.Set(utils.EnumNodeKey("ns1", "test", 0, 0, 0), &pb.EnumCases{
			Cases: []string{},
		})
		assert.NoError(t, err)

		schemaMutator := &mocks.TableSchemaMutator{}
		// test
		enumMutator := NewEnumMutator(txnStore, schemaMutator)
		schemaMutator.On("GetTable", "ns1", "test").Return(&testTable, nil)
		enumCases, err := enumMutator.GetEnumCases("ns1", "test", "c1")
		assert.NoError(t, err)
		assert.Empty(t, enumCases)

		enumIDs, err := enumMutator.ExtendEnumCases("ns1", "test", "c1", []string{"a", "b"})
		assert.NoError(t, err)
		assert.Equal(t, enumIDs, []int{0, 1})

		enumIDs, err = enumMutator.ExtendEnumCases("ns1", "test", "c1", []string{"c", "b", "a"})
		assert.NoError(t, err)
		assert.Equal(t, enumIDs, []int{2, 1, 0})

		enumCases, err = enumMutator.GetEnumCases("ns1", "test", "c1")
		assert.NoError(t, err)
		assert.Equal(t, enumCases, []string{"a", "b", "c"})

		existingNumEnums := len(enumCases)

		// test more than 1 node
		newCases := make([]string, 0)
		for i := 0; i < maxEnumCasePerNode; i++ {
			newCases = append(newCases, strconv.Itoa(existingNumEnums+i))
		}

		enumIDs, err = enumMutator.ExtendEnumCases("ns1", "test", "c1", newCases)
		if assert.NoError(t, err) {
			for i := 0; i < maxEnumCasePerNode; i++ {
				assert.Equal(t, enumIDs[i], existingNumEnums+i)
			}