昔念
cd7583ba05
- 在 sqrt_test.go 中添加了 fastSqr1 测试函数，用于测试事件驱动模型 - 新增了 Event 和 Uint32AsyncEvent 类型用于测试 - 更新了 go.work、go.mod 和
82 lines
2.2 KiB
Go
82 lines
2.2 KiB
Go
package factory_request_reply
|
|
|
|
import (
|
|
"context"
|
|
"strings"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/badu/bus"
|
|
"github.com/badu/bus/test_scenarios/factory-request-reply/cart"
|
|
"github.com/badu/bus/test_scenarios/factory-request-reply/events"
|
|
"github.com/badu/bus/test_scenarios/factory-request-reply/inventory"
|
|
"github.com/badu/bus/test_scenarios/factory-request-reply/prices"
|
|
)
|
|
|
|
var sb strings.Builder
|
|
|
|
type pricesClientStub struct{}
|
|
|
|
func (s *pricesClientStub) GetPricesForProduct(ctx context.Context, in *prices.ProductIDRequest) (*prices.ProductPriceResponse, error) {
|
|
return &prices.ProductPriceResponse{Price: 10.30}, nil
|
|
}
|
|
|
|
type fakeCloser struct {
|
|
}
|
|
|
|
func (f *fakeCloser) Close() error {
|
|
return nil
|
|
}
|
|
|
|
func OnPricesGRPCClientStubRequest(e *events.PricesGRPCClientRequestEvent) {
|
|
sb.WriteString("OnPricesGRPCClientStubRequest\n")
|
|
e.Client = &pricesClientStub{}
|
|
e.Conn = &fakeCloser{}
|
|
<-time.After(300 * time.Millisecond)
|
|
e.Reply()
|
|
}
|
|
|
|
type inventoryClientStub struct{}
|
|
|
|
func (s *inventoryClientStub) GetStockForProduct(ctx context.Context, in *inventory.ProductIDRequest) (*inventory.ProductStockResponse, error) {
|
|
return &inventory.ProductStockResponse{Stock: 200}, nil
|
|
}
|
|
|
|
func OnInventoryGRPCClientStubRequest(e *events.InventoryGRPCClientRequestEvent) {
|
|
sb.WriteString("OnInventoryGRPCClientStubRequest\n")
|
|
e.Client = &inventoryClientStub{}
|
|
e.Conn = &fakeCloser{}
|
|
<-time.After(300 * time.Millisecond)
|
|
e.Reply()
|
|
}
|
|
|
|
func TestGRPCClientStub(t *testing.T) {
|
|
|
|
cartSvc := cart.NewService(&sb)
|
|
|
|
bus.Sub(OnInventoryGRPCClientStubRequest)
|
|
bus.Sub(OnPricesGRPCClientStubRequest)
|
|
|
|
err := cartSvc.AddProductToCart(context.Background(), "1")
|
|
if err != nil {
|
|
t.Fatalf("error adding product to cart : %#v", err)
|
|
}
|
|
|
|
err = cartSvc.AddProductToCart(context.Background(), "2")
|
|
if err != nil {
|
|
t.Fatalf("error adding product to cart : %#v", err)
|
|
}
|
|
|
|
const expecting = "OnInventoryGRPCClientStubRequest\n" +
|
|
"OnPricesGRPCClientStubRequest\n" +
|
|
"stock 200.00pcs @ price 10.30$\n" +
|
|
"OnInventoryGRPCClientStubRequest\n" +
|
|
"OnPricesGRPCClientStubRequest\n" +
|
|
"stock 200.00pcs @ price 10.30$\n"
|
|
|
|
got := sb.String()
|
|
if got != expecting {
|
|
t.Fatalf("expecting :\n%s but got : \n%s", expecting, got)
|
|
}
|
|
}
|