depot/packages/networking/ipfs-cluster/informer/pinqueue/pinqueue_test.go

79 lines
1.5 KiB
Go

package pinqueue
import (
"context"
"testing"
rpc "github.com/libp2p/go-libp2p-gorpc"
)
type mockService struct{}
func (mock *mockService) PinQueueSize(ctx context.Context, in struct{}, out *int64) error {
*out = 42
return nil
}
func mockRPCClient(t *testing.T) *rpc.Client {
s := rpc.NewServer(nil, "mock")
c := rpc.NewClientWithServer(nil, "mock", s)
err := s.RegisterName("PinTracker", &mockService{})
if err != nil {
t.Fatal(err)
}
return c
}
func Test(t *testing.T) {
ctx := context.Background()
cfg := &Config{}
cfg.Default()
cfg.WeightBucketSize = 0
inf, err := New(cfg)
if err != nil {
t.Fatal(err)
}
metrics := inf.GetMetrics(ctx)
if len(metrics) != 1 {
t.Fatal("expected 1 metric")
}
m := metrics[0]
if m.Valid {
t.Error("metric should be invalid")
}
inf.SetClient(mockRPCClient(t))
metrics = inf.GetMetrics(ctx)
if len(metrics) != 1 {
t.Fatal("expected 1 metric")
}
m = metrics[0]
if !m.Valid {
t.Error("metric should be valid")
}
if m.Value != "42" {
t.Error("bad metric value", m.Value)
}
if m.Partitionable {
t.Error("should not be a partitionable metric")
}
if m.Weight != -42 {
t.Error("weight should be -42")
}
cfg.WeightBucketSize = 5
inf, err = New(cfg)
if err != nil {
t.Fatal(err)
}
inf.SetClient(mockRPCClient(t))
metrics = inf.GetMetrics(ctx)
if len(metrics) != 1 {
t.Fatal("expected 1 metric")
}
m = metrics[0]
if m.Weight != -8 {
t.Error("weight should be -8, not", m.Weight)
}
}