128 lines
2.5 KiB
Go
128 lines
2.5 KiB
Go
|
package disk
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"errors"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/ipfs-cluster/ipfs-cluster/api"
|
||
|
|
"github.com/ipfs-cluster/ipfs-cluster/test"
|
||
|
|
|
||
|
|
rpc "github.com/libp2p/go-libp2p-gorpc"
|
||
|
|
)
|
||
|
|
|
||
|
|
type badRPCService struct {
|
||
|
|
}
|
||
|
|
|
||
|
|
func badRPCClient(t *testing.T) *rpc.Client {
|
||
|
|
s := rpc.NewServer(nil, "mock")
|
||
|
|
c := rpc.NewClientWithServer(nil, "mock", s)
|
||
|
|
err := s.RegisterName("IPFSConnector", &badRPCService{})
|
||
|
|
if err != nil {
|
||
|
|
t.Fatal(err)
|
||
|
|
}
|
||
|
|
return c
|
||
|
|
}
|
||
|
|
|
||
|
|
func (mock *badRPCService) RepoStat(ctx context.Context, in struct{}, out *api.IPFSRepoStat) error {
|
||
|
|
return errors.New("fake error")
|
||
|
|
}
|
||
|
|
|
||
|
|
// Returns the first metric
|
||
|
|
func getMetrics(t *testing.T, inf *Informer) api.Metric {
|
||
|
|
t.Helper()
|
||
|
|
metrics := inf.GetMetrics(context.Background())
|
||
|
|
if len(metrics) != 1 {
|
||
|
|
t.Fatal("expected 1 metric")
|
||
|
|
}
|
||
|
|
return metrics[0]
|
||
|
|
}
|
||
|
|
|
||
|
|
func Test(t *testing.T) {
|
||
|
|
ctx := context.Background()
|
||
|
|
cfg := &Config{}
|
||
|
|
cfg.Default()
|
||
|
|
inf, err := NewInformer(cfg)
|
||
|
|
if err != nil {
|
||
|
|
t.Fatal(err)
|
||
|
|
}
|
||
|
|
defer inf.Shutdown(ctx)
|
||
|
|
m := getMetrics(t, inf)
|
||
|
|
if m.Valid {
|
||
|
|
t.Error("metric should be invalid")
|
||
|
|
}
|
||
|
|
inf.SetClient(test.NewMockRPCClient(t))
|
||
|
|
m = getMetrics(t, inf)
|
||
|
|
if !m.Valid {
|
||
|
|
t.Error("metric should be valid")
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestFreeSpace(t *testing.T) {
|
||
|
|
ctx := context.Background()
|
||
|
|
cfg := &Config{}
|
||
|
|
cfg.Default()
|
||
|
|
cfg.MetricType = MetricFreeSpace
|
||
|
|
|
||
|
|
inf, err := NewInformer(cfg)
|
||
|
|
if err != nil {
|
||
|
|
t.Fatal(err)
|
||
|
|
}
|
||
|
|
defer inf.Shutdown(ctx)
|
||
|
|
m := getMetrics(t, inf)
|
||
|
|
if m.Valid {
|
||
|
|
t.Error("metric should be invalid")
|
||
|
|
}
|
||
|
|
inf.SetClient(test.NewMockRPCClient(t))
|
||
|
|
m = getMetrics(t, inf)
|
||
|
|
if !m.Valid {
|
||
|
|
t.Error("metric should be valid")
|
||
|
|
}
|
||
|
|
// The mock client reports 100KB and 2 pins of 1 KB
|
||
|
|
if m.Value != "98000" {
|
||
|
|
t.Error("bad metric value")
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestRepoSize(t *testing.T) {
|
||
|
|
ctx := context.Background()
|
||
|
|
cfg := &Config{}
|
||
|
|
cfg.Default()
|
||
|
|
cfg.MetricType = MetricRepoSize
|
||
|
|
|
||
|
|
inf, err := NewInformer(cfg)
|
||
|
|
if err != nil {
|
||
|
|
t.Fatal(err)
|
||
|
|
}
|
||
|
|
defer inf.Shutdown(ctx)
|
||
|
|
m := getMetrics(t, inf)
|
||
|
|
if m.Valid {
|
||
|
|
t.Error("metric should be invalid")
|
||
|
|
}
|
||
|
|
inf.SetClient(test.NewMockRPCClient(t))
|
||
|
|
m = getMetrics(t, inf)
|
||
|
|
if !m.Valid {
|
||
|
|
t.Error("metric should be valid")
|
||
|
|
}
|
||
|
|
// The mock client reports 100KB and 2 pins of 1 KB
|
||
|
|
if m.Value != "2000" {
|
||
|
|
t.Error("bad metric value")
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestWithErrors(t *testing.T) {
|
||
|
|
ctx := context.Background()
|
||
|
|
cfg := &Config{}
|
||
|
|
cfg.Default()
|
||
|
|
inf, err := NewInformer(cfg)
|
||
|
|
if err != nil {
|
||
|
|
t.Fatal(err)
|
||
|
|
}
|
||
|
|
defer inf.Shutdown(ctx)
|
||
|
|
inf.SetClient(badRPCClient(t))
|
||
|
|
m := getMetrics(t, inf)
|
||
|
|
if m.Valid {
|
||
|
|
t.Errorf("metric should be invalid")
|
||
|
|
}
|
||
|
|
}
|