mirror of
https://github.com/AdguardTeam/AdGuardDNS.git
synced 2025-02-20 11:23:36 +08:00
104 lines
3.0 KiB
Go
104 lines
3.0 KiB
Go
package backendpb_test
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
|
|
"github.com/AdguardTeam/AdGuardDNS/internal/backendpb"
|
|
"google.golang.org/grpc"
|
|
"google.golang.org/protobuf/types/known/emptypb"
|
|
)
|
|
|
|
// testTimeout is the common timeout for tests.
|
|
const testTimeout = 1 * time.Second
|
|
|
|
// testDNSServiceServer is the [backendpb.DNSServiceServer] for tests.
|
|
//
|
|
// TODO(d.kolyshev): Use this to remove as much as possible from the internal
|
|
// test.
|
|
type testDNSServiceServer struct {
|
|
backendpb.UnimplementedDNSServiceServer
|
|
|
|
//lint:ignore ST1003 Keep in sync with the generated code.
|
|
OnCreateDeviceByHumanId func(
|
|
ctx context.Context,
|
|
req *backendpb.CreateDeviceRequest,
|
|
) (resp *backendpb.CreateDeviceResponse, err error)
|
|
|
|
OnGetDNSProfiles func(
|
|
req *backendpb.DNSProfilesRequest,
|
|
srv grpc.ServerStreamingServer[backendpb.DNSProfile],
|
|
) (err error)
|
|
|
|
OnSaveDevicesBillingStat func(
|
|
srv grpc.ClientStreamingServer[backendpb.DeviceBillingStat, emptypb.Empty],
|
|
) (err error)
|
|
}
|
|
|
|
// type check
|
|
var _ backendpb.DNSServiceServer = (*testDNSServiceServer)(nil)
|
|
|
|
// CreateDeviceByHumanId implements the [backendpb.DNSServiceServer] interface
|
|
// for *testDNSServiceServer.
|
|
//
|
|
//lint:ignore ST1003 Keep in sync with the generated code.
|
|
func (s *testDNSServiceServer) CreateDeviceByHumanId(
|
|
ctx context.Context,
|
|
req *backendpb.CreateDeviceRequest,
|
|
) (resp *backendpb.CreateDeviceResponse, err error) {
|
|
return s.OnCreateDeviceByHumanId(ctx, req)
|
|
}
|
|
|
|
// GetDNSProfiles implements the [backendpb.DNSServiceServer] interface for
|
|
// *testDNSServiceServer
|
|
func (s *testDNSServiceServer) GetDNSProfiles(
|
|
req *backendpb.DNSProfilesRequest,
|
|
srv grpc.ServerStreamingServer[backendpb.DNSProfile],
|
|
) (err error) {
|
|
return s.OnGetDNSProfiles(req, srv)
|
|
}
|
|
|
|
// SaveDevicesBillingStat implements the [backendpb.DNSServiceServer] interface
|
|
// for *testDNSServiceServer
|
|
func (s *testDNSServiceServer) SaveDevicesBillingStat(
|
|
srv grpc.ClientStreamingServer[backendpb.DeviceBillingStat, emptypb.Empty],
|
|
) (err error) {
|
|
return s.OnSaveDevicesBillingStat(srv)
|
|
}
|
|
|
|
// testRemoteKVServiceServer is the [backendpb.RemoteKVServiceServer] for tests.
|
|
type testRemoteKVServiceServer struct {
|
|
backendpb.UnimplementedRemoteKVServiceServer
|
|
|
|
OnGet func(
|
|
ctx context.Context,
|
|
req *backendpb.RemoteKVGetRequest,
|
|
) (resp *backendpb.RemoteKVGetResponse, err error)
|
|
|
|
OnSet func(
|
|
ctx context.Context,
|
|
req *backendpb.RemoteKVSetRequest,
|
|
) (resp *backendpb.RemoteKVSetResponse, err error)
|
|
}
|
|
|
|
// type check
|
|
var _ backendpb.RemoteKVServiceServer = (*testRemoteKVServiceServer)(nil)
|
|
|
|
// Get implements the [backendpb.RemoteKVServiceServer] interface for
|
|
// *testRemoteKVServiceServer.
|
|
func (s *testRemoteKVServiceServer) Get(
|
|
ctx context.Context,
|
|
req *backendpb.RemoteKVGetRequest,
|
|
) (resp *backendpb.RemoteKVGetResponse, err error) {
|
|
return s.OnGet(ctx, req)
|
|
}
|
|
|
|
// Set implements the [backendpb.RemoteKVServiceServer] interface for
|
|
// *testRemoteKVServiceServer.
|
|
func (s *testRemoteKVServiceServer) Set(
|
|
ctx context.Context,
|
|
req *backendpb.RemoteKVSetRequest,
|
|
) (resp *backendpb.RemoteKVSetResponse, err error) {
|
|
return s.OnSet(ctx, req)
|
|
}
|