blob: 0dd4d111aafb6baf5eca0e64f48fdaf90e5a0e1b [file] [log] [blame]
// Copyright 2018, OpenCensus Authors
//
// 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 resource
import (
"context"
"errors"
"fmt"
"reflect"
"testing"
)
func TestMerge(t *testing.T) {
cases := []struct {
a, b, want *Resource
}{
{
a: &Resource{
Type: "t1",
Labels: map[string]string{"a": "1", "b": "2"},
},
b: &Resource{
Type: "t2",
Labels: map[string]string{"a": "1", "b": "3", "c": "4"},
},
want: &Resource{
Type: "t1",
Labels: map[string]string{"a": "1", "b": "2", "c": "4"},
},
},
{
a: nil,
b: &Resource{
Type: "t1",
Labels: map[string]string{"a": "1"},
},
want: &Resource{
Type: "t1",
Labels: map[string]string{"a": "1"},
},
},
{
a: &Resource{
Type: "t1",
Labels: map[string]string{"a": "1"},
},
b: nil,
want: &Resource{
Type: "t1",
Labels: map[string]string{"a": "1"},
},
},
}
for i, c := range cases {
t.Run(fmt.Sprintf("case-%d", i), func(t *testing.T) {
res := merge(c.a, c.b)
if !reflect.DeepEqual(res, c.want) {
t.Fatalf("unwanted result: want %+v, got %+v", c.want, res)
}
})
}
}
func TestDecodeLabels(t *testing.T) {
cases := []struct {
encoded string
wantLabels map[string]string
wantFail bool
}{
{
encoded: `example.org/test-1="test $ \"" , Abc="Def"`,
wantLabels: map[string]string{"example.org/test-1": "test $ \"", "Abc": "Def"},
}, {
encoded: `single="key"`,
wantLabels: map[string]string{"single": "key"},
},
{encoded: `invalid-char-ü="test"`, wantFail: true},
{encoded: `invalid-char="ü-test"`, wantFail: true},
{encoded: `missing="trailing-quote`, wantFail: true},
{encoded: `missing=leading-quote"`, wantFail: true},
{encoded: `extra="chars", a`, wantFail: true},
}
for i, c := range cases {
t.Run(fmt.Sprintf("case-%d", i), func(t *testing.T) {
res, err := DecodeLabels(c.encoded)
if err != nil && !c.wantFail {
t.Fatalf("unwanted error: %s", err)
}
if c.wantFail && err == nil {
t.Fatalf("wanted failure but got none, result: %v", res)
}
if !reflect.DeepEqual(res, c.wantLabels) {
t.Fatalf("wanted result %v, got %v", c.wantLabels, res)
}
})
}
}
func TestEncodeLabels(t *testing.T) {
got := EncodeLabels(map[string]string{
"example.org/test-1": "test ¥ \"",
"un": "quøted",
"Abc": "Def",
})
if want := `Abc="Def",example.org/test-1="test ¥ \"",un="quøted"`; got != want {
t.Fatalf("got %q, want %q", got, want)
}
}
func TestMultiDetector(t *testing.T) {
got, err := MultiDetector(
func(context.Context) (*Resource, error) {
return &Resource{
Type: "t1",
Labels: map[string]string{"a": "1", "b": "2"},
}, nil
},
func(context.Context) (*Resource, error) {
return &Resource{
Type: "t2",
Labels: map[string]string{"a": "11", "c": "3"},
}, nil
},
)(context.Background())
if err != nil {
t.Fatalf("unexpected error: %s", err)
}
want := &Resource{
Type: "t1",
Labels: map[string]string{"a": "1", "b": "2", "c": "3"},
}
if !reflect.DeepEqual(got, want) {
t.Fatalf("unexpected resource: want %v, got %v", want, got)
}
wantErr := errors.New("err1")
_, err = MultiDetector(
func(context.Context) (*Resource, error) {
return &Resource{
Type: "t1",
Labels: map[string]string{"a": "1", "b": "2"},
}, nil
},
func(context.Context) (*Resource, error) {
return nil, wantErr
},
)(context.Background())
if err != wantErr {
t.Fatalf("unexpected error: want %v, got %v", wantErr, err)
}
}