pkg/bufman/bufpkg/bufconnect/static_token_provider_test.go

// Licensed to the Apache Software Foundation (ASF) under one or more
// contributor license agreements.  See the NOTICE file distributed with
// this work for additional information regarding copyright ownership.
// The ASF licenses this file to You 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 bufconnect

import (
	"testing"

	"github.com/apache/dubbo-kubernetes/pkg/bufman/pkg/app"
	"github.com/stretchr/testify/assert"
)

func TestNewTokenProviderFromContainer(t *testing.T) {
	t.Parallel()
	tokenSet, err := NewTokenProviderFromContainer(app.NewEnvContainer(map[string]string{
		tokenEnvKey: "default",
	}))
	assert.NoError(t, err)
	token := tokenSet.RemoteToken("fake")
	assert.True(t, tokenSet.IsFromEnvVar())
	assert.Equal(t, "default", token)
}

func TestNewTokenProviderFromString(t *testing.T) {
	t.Parallel()
	tokenProvider, err := NewTokenProviderFromString("default")
	assert.NoError(t, err)
	assert.Equal(t, "default", tokenProvider.RemoteToken("host"))
	tokenProvider, err = NewTokenProviderFromString("token1@host1")
	assert.NoError(t, err)
	assert.Equal(t, "token1", tokenProvider.RemoteToken("host1"))
	tokenProvider, err = NewTokenProviderFromString("token1@remote1,token2@remote2")
	assert.NoError(t, err)
	assert.Equal(t, "token1", tokenProvider.RemoteToken("remote1"))
	assert.Equal(t, "token2", tokenProvider.RemoteToken("remote2"))
	_, err = NewTokenProviderFromString("")
	assert.NoError(t, err)
}

func TestInvalidTokens(t *testing.T) {
	t.Parallel()
	invalidTokens := []string{
		"user1@remote1,user2@remote1",
		"user1@remote1,user2@remote2,",
		",token1@host1",
		"token1@host1,",
		"token1@",
		"token1@host1@",
		"@token1",
		"token1@host1,token2",
		",",
		"token,",
		",token",
	}

	for _, token := range invalidTokens {
		_, err := NewTokenProviderFromString(token)
		assert.Error(t, err, "expected %s to be an invalid token, but it wasn't", token)
		_, err = NewTokenProviderFromContainer(app.NewEnvContainer(map[string]string{
			tokenEnvKey: token,
		}))
		assert.Error(t, err, "expected %s to be an invalid token, but it wasn't", token)
	}
}