Files
threshold/internal/servers/controller/handlers/mask_manager_test.go
T
Todd KnightandJeff Mitchell 168ae3a726 Create MaskManager to guarantee wire/storage update mask consistency (#121)
* Feature: Create MaskManager logic to handle project -> scope update mask logic.

* Updating resource protos and services to use new mask manager setup.

Co-authored-by: Jeff Mitchell <jeffrey.mitchell@gmail.com>
2020-07-20 11:17:09 -07:00

30 lines
1.2 KiB
Go

package handlers
import (
"testing"
pb "github.com/hashicorp/watchtower/internal/gen/controller/protooptions"
"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/require"
)
func TestMaskManager(t *testing.T) {
mm, err := NewMaskManager(&pb.TestBase{}, &pb.TestProperlyNamedFields{})
require.NoError(t, err)
assert.Equal(t, []string(nil), mm.Translate([]string{"doesnt_exist"}))
assert.Equal(t, []string{"OtherFirstField"}, mm.Translate([]string{"first_field"}))
assert.Equal(t, []string{"OtherFirstField"}, mm.Translate([]string{"first_field", "doesnt_exist"}))
assert.Equal(t, []string(nil), mm.Translate([]string{"FiRsT_fIeLd"}))
assert.Equal(t, []string{"other_second_field"}, mm.Translate([]string{"strangly_formatted_field"}))
assert.Equal(t, []string{"other_second_field", "other_field_3"}, mm.Translate([]string{"strangly_formatted_field", "field3"}))
}
func TestMaskManager_errors(t *testing.T) {
_, err := NewMaskManager(&pb.TestBase{}, &pb.TestManyToOneMappings{})
assert.Error(t, err)
_, err = NewMaskManager(&pb.TestBase{}, &pb.TestNameDoesntMap{})
assert.Error(t, err)
_, err = NewMaskManager(&pb.TestBase{}, &pb.TestNotEnoughFields{})
assert.Error(t, err)
}