prysm-pulse/beacon-chain/operations/attestations/service_test.go

package attestations

import (
	"context"
	"errors"
	"testing"
)

func TestStop_OK(t *testing.T) {
	s, err := NewService(context.Background(), &Config{})
	if err != nil {
		t.Fatal(err)
	}

	if err := s.Stop(); err != nil {
		t.Fatalf("Unable to stop attestation pool service: %v", err)
	}

	if s.ctx.Err() != context.Canceled {
		t.Error("context was not canceled")
	}
}

func TestStatus_Error(t *testing.T) {
	err := errors.New("bad bad bad")
	s := &Service{err: err}

	if err := s.Status(); err != s.err {
		t.Errorf("Wanted: %v, got: %v", s.err, s.Status())
	}
}