plugins/fallbacker/timer/timer_fallbacker_test.go

// Licensed to 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. Apache Software Foundation (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 timer

import (
	"errors"
	"reflect"
	"testing"

	"github.com/apache/skywalking-satellite/internal/pkg/log"
	"github.com/apache/skywalking-satellite/internal/pkg/plugin"
	"github.com/apache/skywalking-satellite/internal/satellite/event"
	_ "github.com/apache/skywalking-satellite/internal/satellite/test"
	"github.com/apache/skywalking-satellite/plugins/fallbacker/api"
)

func initFallbacker(cfg plugin.Config) *Fallbacker {
	plugin.RegisterPluginCategory(reflect.TypeOf((*api.Fallbacker)(nil)).Elem())
	plugin.RegisterPlugin(new(Fallbacker))
	cfg[plugin.NameField] = Name
	q := api.GetFallbacker(cfg)
	if q == nil {
		log.Logger.Errorf("cannot get a default config fallbacker from the registry")
		return nil
	}
	return q.(*Fallbacker)
}

func TestFallbacker_FallBack1(t1 *testing.T) {
	count := 0
	mockForwarderFunc := func(_ event.BatchEvents) error {
		count++
		if count < 4 {
			return errors.New("mock error")
		}
		return nil
	}
	tests := []struct {
		name      string
		args      plugin.Config
		want      bool
		wantCount int
	}{
		{
			name:      "default-fallbacker",
			args:      plugin.Config{},
			want:      false,
			wantCount: 2,
		},
		{
			name: "test-recach-max_times",
			args: plugin.Config{
				"max_times":        5,
				"latency_factor":   200,
				"max_latency_time": 3000,
			},
			want:      true,
			wantCount: 4,
		},
		{
			name: "test-unrecach-max_times",
			args: plugin.Config{
				"max_times":        10,
				"latency_factor":   20,
				"max_latency_time": 30000000,
			},
			want:      true,
			wantCount: 4,
		},
	}
	for _, tt := range tests {
		t1.Run(tt.name, func(t1 *testing.T) {
			f := initFallbacker(tt.args)
			count = 0
			if got := f.FallBack(make(event.BatchEvents, 0), mockForwarderFunc); got != tt.want {
				t1.Errorf("FallBack() = %v, want %v", got, tt.want)
			}
			if count != tt.wantCount {
				t1.Errorf("Fallback count = %v, want %v", count, tt.wantCount)
			}
		})
	}
}