internal/test/client.go

// Copyright 2022 The Go Authors. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.

package test

import (
	"context"

	"golang.org/x/vuln/client"
	"golang.org/x/vuln/osv"
)

// MockClient is an in-memory vulnerability
// database client.
type MockClient struct {
	client.Client
	Ret map[string][]*osv.Entry
}

func (mc *MockClient) GetByModule(ctx context.Context, a string) ([]*osv.Entry, error) {
	return mc.Ret[a], nil
}