mocks.go

// Copyright (c) 2025 Bart Venter <bartventer@proton.me>
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//   http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

package internal

import (
	"net/http"
	"net/url"
	"time"
)

var _ Cache = (*MockCache)(nil)

type MockCache struct {
	GetFunc    func(key string) ([]byte, error)
	SetFunc    func(key string, entry []byte) error
	DeleteFunc func(key string) error
}

func (m *MockCache) Get(key string) ([]byte, error) {
	return m.GetFunc(key)
}

func (m *MockCache) Set(key string, entry []byte) error {
	return m.SetFunc(key, entry)
}

func (m *MockCache) Delete(key string) error {
	return m.DeleteFunc(key)
}

var _ ResponseCache = (*MockResponseCache)(nil)

type MockResponseCache struct {
	GetFunc     func(key string, req *http.Request) (*Response, error)
	SetFunc     func(key string, entry *Response) error
	DeleteFunc  func(key string) error
	GetRefsFunc func(key string) (ResponseRefs, error)
	SetRefsFunc func(key string, headers ResponseRefs) error
}

func (m *MockResponseCache) GetRefs(key string) (ResponseRefs, error) {
	return m.GetRefsFunc(key)
}

func (m *MockResponseCache) SetRefs(key string, headers ResponseRefs) error {
	return m.SetRefsFunc(key, headers)
}

func (m *MockResponseCache) Get(key string, req *http.Request) (*Response, error) {
	return m.GetFunc(key, req)
}
func (m *MockResponseCache) Set(key string, entry *Response) error {
	return m.SetFunc(key, entry)
}
func (m *MockResponseCache) Delete(key string) error {
	return m.DeleteFunc(key)
}

var _ RequestMethodChecker = (*MockRequestMethodChecker)(nil)

type MockRequestMethodChecker struct {
	IsRequestMethodUnderstoodFunc func(req *http.Request) bool
}

func (m *MockRequestMethodChecker) IsRequestMethodUnderstood(req *http.Request) bool {
	return m.IsRequestMethodUnderstoodFunc(req)
}

var _ VaryMatcher = (*MockVaryMatcher)(nil)

type MockVaryMatcher struct {
	VaryHeadersMatchFunc func(cachedHdrs ResponseRefs, reqHdr http.Header) (int, bool)
}

func (m *MockVaryMatcher) VaryHeadersMatch(
	cachedHdrs ResponseRefs,
	reqHdr http.Header,
) (int, bool) {
	return m.VaryHeadersMatchFunc(cachedHdrs, reqHdr)
}

var _ URLKeyer = (*MockCacheKeyer)(nil)

type MockCacheKeyer struct {
	CacheKeyFunc func(u *url.URL) string
}

func (m *MockCacheKeyer) URLKey(u *url.URL) string {
	return m.CacheKeyFunc(u)
}

var _ StaleIfErrorPolicy = (*MockStaleIfErrorPolicy)(nil)

type MockStaleIfErrorPolicy struct {
	CanStaleOnErrorFunc func(freshness *Freshness, sies ...StaleIfErrorer) bool
}

func (m *MockStaleIfErrorPolicy) CanStaleOnError(
	freshness *Freshness,
	sies ...StaleIfErrorer,
) bool {
	return m.CanStaleOnErrorFunc(freshness, sies...)
}

var _ FreshnessCalculator = (*MockFreshnessCalculator)(nil)

type MockFreshnessCalculator struct {
	CalculateFreshnessFunc func(resp *http.Response, reqCC CCRequestDirectives, resCC CCResponseDirectives) *Freshness
}

func (m *MockFreshnessCalculator) CalculateFreshness(
	resp *Response,
	reqCC CCRequestDirectives,
	resCC CCResponseDirectives,
) *Freshness {
	return m.CalculateFreshnessFunc(resp.Data, reqCC, resCC)
}

var _ Clock = (*MockClock)(nil)

type MockClock struct {
	NowResult   time.Time
	SinceResult time.Duration
}

func (m *MockClock) Now() time.Time                  { return m.NowResult }
func (m *MockClock) Since(t time.Time) time.Duration { return m.SinceResult }

var _ http.RoundTripper = (*MockRoundTripper)(nil)

type MockRoundTripper struct {
	RoundTripFunc func(req *http.Request) (*http.Response, error)
}

func (m *MockRoundTripper) RoundTrip(req *http.Request) (*http.Response, error) {
	return m.RoundTripFunc(req)
}

var _ ResponseStorer = (*MockResponseStorer)(nil)

type MockResponseStorer struct {
	StoreResponseFunc func(req *http.Request, resp *http.Response, key string, headers ResponseRefs, reqTime, respTime time.Time, refIndex int) error
}

func (m *MockResponseStorer) StoreResponse(
	req *http.Request,
	resp *http.Response,
	key string,
	headers ResponseRefs,
	reqTime, respTime time.Time,
	refIndex int,
) error {
	return m.StoreResponseFunc(req, resp, key, headers, reqTime, respTime, refIndex)
}

var _ CacheInvalidator = (*MockCacheInvalidator)(nil)

type MockCacheInvalidator struct {
	InvalidateCacheFunc func(reqURL *url.URL, respHeader http.Header, headers ResponseRefs, key string)
}

func (m *MockCacheInvalidator) InvalidateCache(
	reqURL *url.URL,
	respHeader http.Header,
	headers ResponseRefs,
	key string,
) {
	m.InvalidateCacheFunc(reqURL, respHeader, headers, key)
}

var _ ValidationResponseHandler = (*MockValidationResponseHandler)(nil)

type MockValidationResponseHandler struct {
	HandleValidationResponseFunc func(ctx RevalidationContext, req *http.Request, resp *http.Response) (*http.Response, error)
}

func (m *MockValidationResponseHandler) HandleValidationResponse(
	ctx RevalidationContext,
	req *http.Request,
	resp *http.Response,
) (*http.Response, error) {
	return m.HandleValidationResponseFunc(ctx, req, resp)
}

var _ VaryMatcher = (*MockVaryMatcher)(nil)