136 lines
3.3 KiB
Go
136 lines
3.3 KiB
Go
package aghtest
|
|
|
|
import (
|
|
"io/fs"
|
|
"net"
|
|
|
|
"github.com/AdguardTeam/AdGuardHome/internal/aghos"
|
|
"github.com/AdguardTeam/dnsproxy/upstream"
|
|
"github.com/miekg/dns"
|
|
)
|
|
|
|
// Interface Mocks
|
|
//
|
|
// Keep entities in this file in alphabetic order.
|
|
|
|
// Standard Library
|
|
|
|
// type check
|
|
var _ fs.FS = &FS{}
|
|
|
|
// FS is a mock [fs.FS] implementation for tests.
|
|
type FS struct {
|
|
OnOpen func(name string) (fs.File, error)
|
|
}
|
|
|
|
// Open implements the [fs.FS] interface for *FS.
|
|
func (fsys *FS) Open(name string) (fs.File, error) {
|
|
return fsys.OnOpen(name)
|
|
}
|
|
|
|
// type check
|
|
var _ fs.GlobFS = &GlobFS{}
|
|
|
|
// GlobFS is a mock [fs.GlobFS] implementation for tests.
|
|
type GlobFS struct {
|
|
// FS is embedded here to avoid implementing all it's methods.
|
|
FS
|
|
OnGlob func(pattern string) ([]string, error)
|
|
}
|
|
|
|
// Glob implements the [fs.GlobFS] interface for *GlobFS.
|
|
func (fsys *GlobFS) Glob(pattern string) ([]string, error) {
|
|
return fsys.OnGlob(pattern)
|
|
}
|
|
|
|
// type check
|
|
var _ fs.StatFS = &StatFS{}
|
|
|
|
// StatFS is a mock [fs.StatFS] implementation for tests.
|
|
type StatFS struct {
|
|
// FS is embedded here to avoid implementing all it's methods.
|
|
FS
|
|
OnStat func(name string) (fs.FileInfo, error)
|
|
}
|
|
|
|
// Stat implements the [fs.StatFS] interface for *StatFS.
|
|
func (fsys *StatFS) Stat(name string) (fs.FileInfo, error) {
|
|
return fsys.OnStat(name)
|
|
}
|
|
|
|
// type check
|
|
var _ net.Listener = (*Listener)(nil)
|
|
|
|
// Listener is a mock [net.Listener] implementation for tests.
|
|
type Listener struct {
|
|
OnAccept func() (conn net.Conn, err error)
|
|
OnAddr func() (addr net.Addr)
|
|
OnClose func() (err error)
|
|
}
|
|
|
|
// Accept implements the [net.Listener] interface for *Listener.
|
|
func (l *Listener) Accept() (conn net.Conn, err error) {
|
|
return l.OnAccept()
|
|
}
|
|
|
|
// Addr implements the [net.Listener] interface for *Listener.
|
|
func (l *Listener) Addr() (addr net.Addr) {
|
|
return l.OnAddr()
|
|
}
|
|
|
|
// Close implements the [net.Listener] interface for *Listener.
|
|
func (l *Listener) Close() (err error) {
|
|
return l.OnClose()
|
|
}
|
|
|
|
// Module dnsproxy
|
|
|
|
// type check
|
|
var _ upstream.Upstream = (*UpstreamMock)(nil)
|
|
|
|
// UpstreamMock is a mock [upstream.Upstream] implementation for tests.
|
|
//
|
|
// TODO(a.garipov): Replace with all uses of Upstream with UpstreamMock and
|
|
// rename it to just Upstream.
|
|
type UpstreamMock struct {
|
|
OnAddress func() (addr string)
|
|
OnExchange func(req *dns.Msg) (resp *dns.Msg, err error)
|
|
}
|
|
|
|
// Address implements the [upstream.Upstream] interface for *UpstreamMock.
|
|
func (u *UpstreamMock) Address() (addr string) {
|
|
return u.OnAddress()
|
|
}
|
|
|
|
// Exchange implements the [upstream.Upstream] interface for *UpstreamMock.
|
|
func (u *UpstreamMock) Exchange(req *dns.Msg) (resp *dns.Msg, err error) {
|
|
return u.OnExchange(req)
|
|
}
|
|
|
|
// Module AdGuardHome
|
|
|
|
// type check
|
|
var _ aghos.FSWatcher = (*FSWatcher)(nil)
|
|
|
|
// FSWatcher is a mock [aghos.FSWatcher] implementation for tests.
|
|
type FSWatcher struct {
|
|
OnEvents func() (e <-chan struct{})
|
|
OnAdd func(name string) (err error)
|
|
OnClose func() (err error)
|
|
}
|
|
|
|
// Events implements the [aghos.FSWatcher] interface for *FSWatcher.
|
|
func (w *FSWatcher) Events() (e <-chan struct{}) {
|
|
return w.OnEvents()
|
|
}
|
|
|
|
// Add implements the [aghos.FSWatcher] interface for *FSWatcher.
|
|
func (w *FSWatcher) Add(name string) (err error) {
|
|
return w.OnAdd(name)
|
|
}
|
|
|
|
// Close implements the [aghos.FSWatcher] interface for *FSWatcher.
|
|
func (w *FSWatcher) Close() (err error) {
|
|
return w.OnClose()
|
|
}
|