Function TestNamespaceMutator

controller/mutators/etcd/namespace_mutator_test.go:27–71

(t *testing.T)

)

func TestNamespaceMutator(t *testing.T) {
	t.Run("list namespaces should work", func(t *testing.T) {
		etcdStore := mem.NewStore()
		_, err := etcdStore.Set(utils.NamespaceListKey(), &pb.EntityList{
			Entities: []*pb.EntityName{
				{
					Name: "ns1",
				},
			},
		})
		assert.NoError(t, err)

		// test
		namespaceMutator := NewNamespaceMutator(etcdStore)
		res, err := namespaceMutator.ListNamespaces()
		assert.NoError(t, err)
		assert.Equal(t, []string{"ns1"}, res)
	})

	t.Run("create namespace should work", func(t *testing.T) {
		etcdStore := mem.NewStore()

		_, err := etcdStore.Set(utils.NamespaceListKey(), &pb.EntityList{
			Entities: []*pb.EntityName{
				{
					Name: "ns1",
				},
			},
		})
		assert.NoError(t, err)

		// test
		namespaceMutator := NewNamespaceMutator(etcdStore)
		err = namespaceMutator.CreateNamespace("ns1")
		assert.EqualError(t, err, common.ErrNamespaceAlreadyExists.Error())

		err = namespaceMutator.CreateNamespace("ns2")
		assert.NoError(t, err)

		_, err = etcdStore.Get(utils.JobListKey("ns2"))
		assert.NoError(t, err)
		_, err = etcdStore.Get(utils.SchemaListKey("ns2"))
		assert.NoError(t, err)
	})
}