(t *testing.T, st ServerType)
| 87 | } |
| 88 | |
| 89 | func (ts *TestServer) NewServer(t *testing.T, st ServerType) *grpc.Server { |
| 90 | switch st { |
| 91 | case ReadServer: |
| 92 | return ts.Reg.ReadGRPCServer(t.Context()) |
| 93 | case WriteServer: |
| 94 | return ts.Reg.WriteGRPCServer(t.Context()) |
| 95 | case OplServer: |
| 96 | return ts.Reg.OplGRPCServer(t.Context()) |
| 97 | default: |
| 98 | panic(fmt.Sprintf("unknown ServerType %s", st)) |
| 99 | } |
| 100 | } |
| 101 | |
| 102 | func (ts *TestServer) Shutdown(t *testing.T) { |
| 103 | for _, s := range ts.Servers { |