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