package attestations

import (
	"context"
	"errors"
	"testing"

	"github.com/prysmaticlabs/prysm/shared/testutil/require"
)

func TestStop_OK(t *testing.T) {
	s, err := NewService(context.Background(), &Config{})
	require.NoError(t, 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())
	}
}