MealQuest/internal/dinners/export_test.go

package dinners

import (
	"context"
	"testing"
	"time"
)

// SetGroundTimeout sets the deadline of grounding by hand until t ends. A
// test calling it must not run in parallel.
func SetGroundTimeout(t *testing.T, d time.Duration) {
	old := groundTimeout
	groundTimeout = d
	t.Cleanup(func() { groundTimeout = old })
}

// SetOfferLifetime sets how long candidates ground a dinner until t ends.
// A test calling it must not run in parallel.
func SetOfferLifetime(t *testing.T, d time.Duration) {
	old := offerLifetime
	offerLifetime = d
	t.Cleanup(func() { offerLifetime = old })
}

// AdvanceClock moves the service's clock d ahead of real time until t
// ends, without firing any timer. A test calling it must not run in
// parallel.
func AdvanceClock(t *testing.T, d time.Duration) {
	old := now
	now = func() time.Time { return old().Add(d) }
	t.Cleanup(func() { now = old })
}

// HasOffers reports whether the service holds candidates offered for
// dinner id.
func (s *Service) HasOffers(id string) bool {
	s.offersMu.Lock()
	defer s.offersMu.Unlock()
	_, ok := s.offers[id]
	return ok
}

// OfferCount is the number of dinners whose candidates the service holds.
func (s *Service) OfferCount() int {
	s.offersMu.Lock()
	defer s.offersMu.Unlock()
	return len(s.offers)
}

// RunGroundJob runs the KindGroundDinner handler for p in the caller's
// goroutine, as the job runner would, and returns the grounding status it
// reports.
func (s *Service) RunGroundJob(ctx context.Context, p GroundJob) (string, error) {
	r, err := s.runGroundJob(ctx, p, nil)
	return r.GroundingStatus, err
}