resource/resource_test.go

// 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)
	}
}