You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
58 lines
1.3 KiB
58 lines
1.3 KiB
// Copyright (c) Tailscale Inc & contributors
|
|
// SPDX-License-Identifier: BSD-3-Clause
|
|
|
|
package localclient
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
"tailscale.com/ipn"
|
|
)
|
|
|
|
type FakeLocalClient struct {
|
|
FakeIPNBusWatcher
|
|
SetServeCalled bool
|
|
EditPrefsCalls []*ipn.MaskedPrefs
|
|
GetPrefsResult *ipn.Prefs
|
|
}
|
|
|
|
func (m *FakeLocalClient) SetServeConfig(ctx context.Context, cfg *ipn.ServeConfig) error {
|
|
m.SetServeCalled = true
|
|
return nil
|
|
}
|
|
|
|
func (m *FakeLocalClient) EditPrefs(ctx context.Context, mp *ipn.MaskedPrefs) (*ipn.Prefs, error) {
|
|
m.EditPrefsCalls = append(m.EditPrefsCalls, mp)
|
|
if m.GetPrefsResult == nil {
|
|
return &ipn.Prefs{}, nil
|
|
}
|
|
return m.GetPrefsResult, nil
|
|
}
|
|
|
|
func (m *FakeLocalClient) GetPrefs(ctx context.Context) (*ipn.Prefs, error) {
|
|
if m.GetPrefsResult == nil {
|
|
return &ipn.Prefs{}, nil
|
|
}
|
|
return m.GetPrefsResult, nil
|
|
}
|
|
|
|
func (f *FakeLocalClient) WatchIPNBus(ctx context.Context, mask ipn.NotifyWatchOpt) (IPNBusWatcher, error) {
|
|
return &f.FakeIPNBusWatcher, nil
|
|
}
|
|
|
|
func (f *FakeLocalClient) CertPair(ctx context.Context, domain string) ([]byte, []byte, error) {
|
|
return nil, nil, fmt.Errorf("CertPair not implemented")
|
|
}
|
|
|
|
type FakeIPNBusWatcher struct {
|
|
NotifyChan chan ipn.Notify
|
|
}
|
|
|
|
func (f *FakeIPNBusWatcher) Close() error {
|
|
return nil
|
|
}
|
|
|
|
func (f *FakeIPNBusWatcher) Next() (ipn.Notify, error) {
|
|
return <-f.NotifyChan, nil
|
|
}
|
|
|