Merge pull request #5 from chronicleprotocol/balancerv2_wsteth_fix

make balancerV2 mock configurable

Author: madumas

e2e/exchanges_test.go

@@ -41,6 +41,14 @@ type ExchangesE2ESuite struct {
 	url string
 }
 
+type jsonrpcMessage struct {
+	Version string          `json:"jsonrpc"`
+	ID      json.RawMessage `json:"id,omitempty"`
+	Method  string          `json:"method,omitempty"`
+	Params  json.RawMessage `json:"params,omitempty"`
+	Result  string          `json:"result,omitempty"`
+}
+
 func (s *ExchangesE2ESuite) SetupSuite() {
 	smockerHost, exist := os.LookupEnv("SMOCKER_HOST")
 	s.Require().True(exist, "SMOCKER_HOST env variable have to be set")
@@ -102,6 +110,79 @@ func (s *ExchangesE2ESuite) TestBalancer() {
 	defer func() { _ = resp.Body.Close() }()
 }
 
+func (s *ExchangesE2ESuite) TestBalancerV2_getLatest() {
+	ex := origin.NewExchange("balancerV2").WithSymbol("STETH/ETH").WithPrice(1)
+	// No `pool` field should fail
+	err := infestor.NewMocksBuilder().Reset().Add(ex).Deploy(s.api)
+	s.Require().Error(err)
+
+	// With set rate and price
+	price := "0x0000000000000000000000000000000000000000000000000dd22d6848e229b8"
+	ex = ex.WithCustom("rate", price).WithCustom("price", price)
+	err = infestor.NewMocksBuilder().Debug().Reset().Add(ex).Deploy(s.api)
+	s.Require().NoError(err)
+
+	url := fmt.Sprintf("%s/", s.url)
+	reqJSON := `{"method":"eth_call","params":[{"from":null,"to":"0x0000000000000000000000000000000000000000","data":"0xb10be7390000000000000000000000000000000000000000000000000000000000000000"}, "latest"],"id":1,"jsonrpc":"2.0"}`
+	jsonStr := []byte(fmt.Sprint(reqJSON))
+	resp, err := http.Post(url, "application/json", bytes.NewBuffer(jsonStr))
+	s.Require().NoError(err)
+	s.Require().Equal(http.StatusOK, resp.StatusCode)
+
+	var response jsonrpcMessage
+	err = parseBody(resp, &response)
+	s.Require().NoError(err)
+	s.Require().NotNil(response)
+	s.Require().Equal(price, response.Result)
+
+	// Test status code
+	ex = ex.WithStatusCode(http.StatusNotFound)
+	err = infestor.NewMocksBuilder().Reset().Add(ex).Deploy(s.api)
+	s.Require().NoError(err)
+
+	resp, err = http.Post(url, "application/json", bytes.NewBuffer(jsonStr))
+	s.Require().NoError(err)
+	s.Require().Equal(http.StatusNotFound, resp.StatusCode)
+	defer func() { _ = resp.Body.Close() }()
+}
+
+func (s *ExchangesE2ESuite) TestBalancerV2_getPriceRateCache() {
+	ex := origin.NewExchange("balancerV2").WithSymbol("STETH/ETH").WithPrice(1)
+	// No `pool` field should fail
+	err := infestor.NewMocksBuilder().Reset().Add(ex).Deploy(s.api)
+	s.Require().Error(err)
+
+	// With set rate and price
+	rate := "0x0000000000000000000000000000000000000000000000000dd22d6848e229b8"
+	ex = ex.WithCustom("rate", rate).WithCustom("price", rate)
+	err = infestor.NewMocksBuilder().Debug().Reset().Add(ex).Deploy(s.api)
+	s.Require().NoError(err)
+
+	url := fmt.Sprintf("%s/", s.url)
+	reqJSON := `{"method":"eth_call","params":[{"from":null,"to":"0x0000000000000000000000000000000000000000","data":"0xb867ee5a"}, "latest"],"id":1,"jsonrpc":"2.0"}`
+	jsonStr := []byte(fmt.Sprint(reqJSON))
+	resp, err := http.Post(url, "application/json", bytes.NewBuffer(jsonStr))
+	s.Require().NoError(err)
+	s.Require().Equal(http.StatusOK, resp.StatusCode)
+
+	var response jsonrpcMessage
+	err = parseBody(resp, &response)
+	s.Require().NoError(err)
+	s.Require().NotNil(response)
+	fmt.Println(response.Result)
+	s.Require().Equal(rate, response.Result)
+
+	// Test status code
+	ex = ex.WithStatusCode(http.StatusNotFound)
+	err = infestor.NewMocksBuilder().Reset().Add(ex).Deploy(s.api)
+	s.Require().NoError(err)
+
+	resp, err = http.Post(url, "application/json", bytes.NewBuffer(jsonStr))
+	s.Require().NoError(err)
+	s.Require().Equal(http.StatusNotFound, resp.StatusCode)
+	defer func() { _ = resp.Body.Close() }()
+}
+
 func (s *ExchangesE2ESuite) TestBinance() {
 	ex := origin.NewExchange("binance").WithSymbol("ETH/BTC").WithPrice(1)
 

origin/balancerv2.go

@@ -30,6 +30,11 @@ func (b BalancerV2) BuildMocks(e []ExchangeMock) ([]*smocker.Mock, error) {
 }
 
 func (b BalancerV2) buildGetLatest(e ExchangeMock) (*smocker.Mock, error) {
+	price, ok := e.Custom["price"]
+	if !ok {
+		return nil, fmt.Errorf("`price` custom field is requierd for balancerV2")
+	}
+
 	// getLatest(uint256)
 	m := smocker.ShouldContainSubstring("0xb10be739")
 
@@ -48,12 +53,17 @@ func (b BalancerV2) buildGetLatest(e ExchangeMock) (*smocker.Mock, error) {
 					"application/json",
 				},
 			},
-			Body: fmt.Sprintf(rpcJSONResult, "0x0000000000000000000000000000000000000000000000000dd22d6848e229b8"),
+			Body: fmt.Sprintf(rpcJSONResult, price),
 		},
 	}, nil
 }
 
 func (b BalancerV2) buildGetPriceRateCache(e ExchangeMock) (*smocker.Mock, error) {
+	rate, ok := e.Custom["rate"]
+	if !ok {
+		return nil, fmt.Errorf("`rate` custom field is requierd for balancerV2")
+	}
+
 	// getPriceRateCache(uint256,uint256,uint256)
 	m := smocker.ShouldContainSubstring("0xb867ee5a")
 
@@ -72,7 +82,7 @@ func (b BalancerV2) buildGetPriceRateCache(e ExchangeMock) (*smocker.Mock, error
 					"application/json",
 				},
 			},
-			Body: fmt.Sprintf(rpcJSONResult, "0x0000000000000000000000000000000000000000000000000dd22d6848e229b8"),
+			Body: fmt.Sprintf(rpcJSONResult, rate),
 		},
 	}, nil
 }