func main() { flag.Parse() s, err := common.NewServer(*flAddr) if err != nil { log.Fatal(err) } log.Printf("run rpc server") err = s.Run() if err != nil { log.Fatal(err) } }
func TestClientGet(t *testing.T) { var client *BalanceStorageClient addr := "127.0.0.1:17090" server, err := common.NewServer(addr) if err != nil { log.Fatal(err) } go func() { if err = server.Run(); err != nil { log.Fatalf("server run err: %s", err) } }() for i := 0; i < 50; i++ { time.Sleep(time.Millisecond * 100) client, err = Connect(addr) if err != nil { continue } } if err != nil { log.Fatalf("client connect err: %s", err) } else { log.Printf("connected") } value, err := client.IncrementValue(1, 10) if err != nil { t.Fatal(err) } value, err = client.IncrementValue(1, -20) if err != nil { t.Fatal(err) } if value != 90 { t.Fatalf("balance not equals to 90: %d", value) } t.Logf("balance tested") }
func TestMain(m *testing.M) { server, err := rpcserver.NewServer(RpcAddr) if err != nil { log.Fatalf("rpc init err: %s", err) } go func() { if err := server.Run(); err != nil { log.Fatalf("rpc run err: %s", err) } }() log.Printf("rpc runned") app := httpserver.App{} if err = app.Init(RpcAddr); err != nil { log.Fatalf("http init error: %s", err) } go func() { if err := app.Run(HttpAddr); err != nil { log.Fatalf("http run error: %s", err) } }() log.Printf("http runned") time.Sleep(time.Second) m.Run() }