| // Copyright 2015 The Vanadium 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 fake |
| |
| import ( |
| "time" |
| |
| "v.io/v23/context" |
| "v.io/v23/flow" |
| "v.io/v23/rpc" |
| "v.io/v23/security" |
| "v.io/x/ref/lib/apilog" |
| ) |
| |
| // SetClientFactory can be used to inject a mock Client implementation |
| // into the context. When v23.WithNewClient is called passed function |
| // will be invoked. |
| func SetClientFactory(ctx *context.T, factory func(ctx *context.T, opts ...rpc.ClientOpt) rpc.Client) *context.T { |
| client := factory(ctx) |
| ctx = context.WithValue(ctx, clientFactoryKey, factory) |
| return context.WithValue(ctx, clientKey, client) |
| } |
| func (r *Runtime) WithNewClient(ctx *context.T, opts ...rpc.ClientOpt) (*context.T, rpc.Client, error) { |
| defer apilog.LogCallf(ctx, "opts...=%v", opts)(ctx, "") // gologcop: DO NOT EDIT, MUST BE FIRST STATEMENT |
| factory, ok := ctx.Value(clientFactoryKey).(func(ctx *context.T, opts ...rpc.ClientOpt) rpc.Client) |
| if !ok { |
| panic("Calling WithNewClient on the fake runtime, but no factory has been set.") |
| } |
| client := factory(ctx, opts...) |
| return context.WithValue(ctx, clientKey, client), client, nil |
| } |
| func (r *Runtime) GetClient(ctx *context.T) rpc.Client { |
| defer apilog.LogCall(ctx)(ctx) // gologcop: DO NOT EDIT, MUST BE FIRST STATEMENT |
| c, _ := ctx.Value(clientKey).(rpc.Client) |
| return c |
| } |
| |
| func (r *Runtime) GetListenSpec(ctx *context.T) rpc.ListenSpec { |
| defer apilog.LogCall(ctx)(ctx) // gologcop: DO NOT EDIT, MUST BE FIRST STATEMENT |
| ls, _ := ctx.Value(listenSpecKey).(rpc.ListenSpec) |
| return ls |
| } |
| |
| func (r *Runtime) WithListenSpec(ctx *context.T, ls rpc.ListenSpec) *context.T { |
| defer apilog.LogCall(ctx)(ctx) // gologcop: DO NOT EDIT, MUST BE FIRST STATEMENT |
| return context.WithValue(ctx, listenSpecKey, ls) |
| return ctx |
| } |
| |
| // SetFlowManagerFactory can be used to inject a mock FlowManager |
| // implementation into the context. When v23.NewFlowManager is called |
| // passed function will be invoked. |
| func SetFlowManagerFactory(ctx *context.T, factory func(ctx *context.T, channelTimeout time.Duration) flow.Manager) *context.T { |
| return context.WithValue(ctx, flowFactoryKey, factory) |
| } |
| |
| func (r *Runtime) NewFlowManager(ctx *context.T, channelTimeout time.Duration) (flow.Manager, error) { |
| defer apilog.LogCall(ctx)(ctx) // gologcop: DO NOT EDIT, MUST BE FIRST STATEMENT |
| |
| factory, ok := ctx.Value(flowFactoryKey).(func(ctx *context.T, channelTimeout time.Duration) flow.Manager) |
| if !ok { |
| panic("Calling NewFlowManager on the fake runtime, but no factory has been set.") |
| } |
| return factory(ctx, channelTimeout), nil |
| } |
| |
| func (r *Runtime) WithNewServer(ctx *context.T, name string, object interface{}, auth security.Authorizer, opts ...rpc.ServerOpt) (*context.T, rpc.Server, error) { |
| defer apilog.LogCall(ctx)(ctx) // gologcop: DO NOT EDIT, MUST BE FIRST STATEMENT |
| panic("unimplemented") |
| } |
| |
| func (r *Runtime) WithNewDispatchingServer(ctx *context.T, name string, disp rpc.Dispatcher, opts ...rpc.ServerOpt) (*context.T, rpc.Server, error) { |
| defer apilog.LogCall(ctx)(ctx) // gologcop: DO NOT EDIT, MUST BE FIRST STATEMENT |
| panic("unimplemented") |
| } |