talm

1
// This Source Code Form is subject to the terms of the Mozilla Public
2
// License, v. 2.0. If a copy of the MPL was not distributed with this
3
// file, You can obtain one at http://mozilla.org/MPL/2.0/.
4

5
package director_test
6

7
import (
8
	"context"
9

10
	"github.com/siderolabs/grpc-proxy/proxy"
11
	"google.golang.org/grpc"
12
)
13

14
type mockBackend struct {
15
	target string
16
}
17

18
func (m *mockBackend) String() string {
19
	return m.target
20
}
21

22
func (m *mockBackend) GetConnection(ctx context.Context, fullMethodName string) (context.Context, *grpc.ClientConn, error) {
23
	return ctx, nil, nil
24
}
25

26
func (m *mockBackend) AppendInfo(streaming bool, resp []byte) ([]byte, error) {
27
	return resp, nil
28
}
29

30
func (m *mockBackend) BuildError(streaming bool, err error) ([]byte, error) {
31
	return nil, nil
32
}
33

34
func mockBackendFactory(target string) (proxy.Backend, error) {
35
	return &mockBackend{target: target}, nil
36
}
37

38
type mockLocalAddressProvider struct {
39
	local map[string]struct{}
40
}
41

42
func (m *mockLocalAddressProvider) IsLocalTarget(t string) bool {
43
	_, ok := m.local[t]
44

45
	return ok
46
}
47