// StartActionLoop will start the action loop for a fake tablet,
// using ft.FakeMysqlDaemon as the backing mysqld.
func (ft *FakeTablet) StartActionLoop(t *testing.T, wr *wrangler.Wrangler) {
if ft.Agent != nil {
t.Fatalf("Agent for %v is already running", ft.Tablet.Alias)
}
// Listen on a random port
var err error
ft.Listener, err = net.Listen("tcp", ":0")
if err != nil {
t.Fatalf("Cannot listen: %v", err)
}
port := ft.Listener.Addr().(*net.TCPAddr).Port
// create a test agent on that port, and re-read the record
// (it has new ports and IP)
ft.Agent = tabletmanager.NewTestActionAgent(wr.TopoServer(), ft.Tablet.Alias, port, ft.FakeMysqlDaemon)
ft.Tablet = ft.Agent.Tablet().Tablet
// create the RPC server
server := rpcplus.NewServer()
gorpctmserver.RegisterForTest(server, ft.Agent)
// create the HTTP server, serve the server from it
handler := http.NewServeMux()
bsonrpc.ServeTestRPC(handler, server)
httpServer := http.Server{
Handler: handler,
}
go httpServer.Serve(ft.Listener)
}
// the test here creates a fake server implementation, a fake client
// implementation, and runs the test suite against the setup.
func TestGoRpcTMServer(t *testing.T) {
// Listen on a random port
listener, err := net.Listen("tcp", ":0")
if err != nil {
t.Fatalf("Cannot listen: %v", err)
}
port := listener.Addr().(*net.TCPAddr).Port
// Create a Go Rpc server and listen on the port
server := rpcplus.NewServer()
server.Register(&TabletManager{agentrpctest.NewFakeRpcAgent(t)})
// create the HTTP server, serve the server from it
handler := http.NewServeMux()
bsonrpc.ServeTestRPC(handler, server)
httpServer := http.Server{
Handler: handler,
}
go httpServer.Serve(listener)
// Create a Go Rpc client to talk to the fake tablet
client := &gorpctmclient.GoRpcTabletManagerClient{}
ti := topo.NewTabletInfo(&topo.Tablet{
Alias: topo.TabletAlias{
Cell: "test",
Uid: 123,
},
Hostname: "localhost",
Portmap: map[string]int{
"vt": port,
},
}, 0)
// and run the test suite
agentrpctest.AgentRpcTestSuite(t, client, ti)
}