Merge pull request #1 from chronicleprotocol/sc641-wsteth-integration-test

sc641, add mock for ETHRPC call for WSTETH

Author: jar-o

.github/workflows/lint_test.yml

@@ -12,21 +12,21 @@ jobs:
   test:
     strategy:
       matrix:
-        go-version: [1.16.x]
+        go-version: [1.17.x]
         os: [ubuntu-latest]
     runs-on: ${{ matrix.os }}
     steps:
       - name: Install Go
-        uses: actions/setup-go@v2
+        uses: actions/setup-go@v3
         with:
           go-version: ${{ matrix.go-version }}
       - name: Checkout code
-        uses: actions/checkout@v2
+        uses: actions/checkout@v3
       - name: Linting code
-        uses: golangci/golangci-lint-action@v2
+        uses: golangci/golangci-lint-action@v3
         with:
           # Optional: version of golangci-lint to use in form of v1.2 or v1.2.3 or `latest` to use the latest version
-          version: v1.43
+          version: v1.44
 
           # Optional: working directory, useful for monorepos
           # working-directory: somedir
@@ -50,4 +50,4 @@ jobs:
       - name: Run E2E Test
         env:
           SMOCKER_HOST: "http://localhost"
-        run: go test ./e2e
+        run: go test ./e2e

origin/balancerv2.go

@@ -0,0 +1,39 @@
+package origin
+
+// Simulate an ETHRPC node returning the price for STETH/ETH on BalancerV2
+// https://etherscan.io/address/0x32296969ef14eb0c6d29669c550d4a0449130230#code
+
+import (
+	"fmt"
+
+	"github.com/chronicleprotocol/infestor/smocker"
+)
+
+type BalancerV2 struct{}
+
+func (b BalancerV2) BuildMocks(e []ExchangeMock) ([]*smocker.Mock, error) {
+	return CombineMocks(e, b.build)
+}
+
+func (b BalancerV2) build(e ExchangeMock) (*smocker.Mock, error) {
+	m := smocker.NewSubstringMatcher("0x32296969ef14eb0c6d29669c550d4a0449130230")
+
+	return &smocker.Mock{
+		Request: smocker.MockRequest{
+			Method: smocker.NewStringMatcher("POST"),
+			Path:   smocker.NewStringMatcher("/"),
+			Body: &smocker.BodyMatcher{
+				BodyString: &m,
+			},
+		},
+		Response: &smocker.MockResponse{
+			Status: e.StatusCode,
+			Headers: map[string]smocker.StringSlice{
+				"Content-Type": []string{
+					"application/json",
+				},
+			},
+			Body: fmt.Sprintf(rpcJSONResult, "0x0000000000000000000000000000000000000000000000000dd22d6848e229b8"),
+		},
+	}, nil
+}

origin/curve.go

@@ -0,0 +1,39 @@
+package origin
+
+// Simulate an ETHRPC node returning the price for STETH/ETH on Curve
+// https://etherscan.io/address/0xdc24316b9ae028f1497c275eb9192a3ea0f67022#code
+
+import (
+	"fmt"
+
+	"github.com/chronicleprotocol/infestor/smocker"
+)
+
+type Curve struct{}
+
+func (b Curve) BuildMocks(e []ExchangeMock) ([]*smocker.Mock, error) {
+	return CombineMocks(e, b.build)
+}
+
+func (b Curve) build(e ExchangeMock) (*smocker.Mock, error) {
+	m := smocker.NewSubstringMatcher("0xdc24316b9ae028f1497c275eb9192a3ea0f67022")
+
+	return &smocker.Mock{
+		Request: smocker.MockRequest{
+			Method: smocker.NewStringMatcher("POST"),
+			Path:   smocker.NewStringMatcher("/"),
+			Body: &smocker.BodyMatcher{
+				BodyString: &m,
+			},
+		},
+		Response: &smocker.MockResponse{
+			Status: e.StatusCode,
+			Headers: map[string]smocker.StringSlice{
+				"Content-Type": []string{
+					"application/json",
+				},
+			},
+			Body: fmt.Sprintf(rpcJSONResult, "0x0000000000000000000000000000000000000000000000000dcfcc3d4023b410"),
+		},
+	}, nil
+}

origin/exchange.go

@@ -51,6 +51,9 @@ var exchanges = map[string]Mockable{
 	"okex":          Okex{},
 	"poloniex":      Poloniex{},
 	"upbit":         Upbit{},
+	"wsteth":        WSTETH{},
+	"balancerV2":    BalancerV2{},
+	"curve":         Curve{},
 }
 
 // Symbol represents an asset pair.

origin/wsteth.go

@@ -0,0 +1,84 @@
+package origin
+
+// Simulate an ETHRPC node returning the price for WstETH
+// https://etherscan.io/address/0x7f39C581F595B53c5cb19bD0b3f8dA6c935E2Ca0#code
+
+import (
+	"fmt"
+
+	"github.com/chronicleprotocol/infestor/smocker"
+)
+
+type WSTETH struct{}
+
+const rpcJSONResult = `{
+  "jsonrpc": "2.0",
+  "id": 1,
+  "result": "%s"
+}`
+
+func (b WSTETH) BuildMocks(e []ExchangeMock) ([]*smocker.Mock, error) {
+	mocks := make([]*smocker.Mock, 0)
+	fmt.Println(mocks)
+	m, err := CombineMocks(e, b.buildTokensPerStEth)
+	if err != nil {
+		return nil, err
+	}
+	mocks = append(mocks, m...)
+
+	m, err = CombineMocks(e, b.buildstEthPerToken)
+	if err != nil {
+		return nil, err
+	}
+	mocks = append(mocks, m...)
+
+	return mocks, nil
+}
+
+func (b WSTETH) buildTokensPerStEth(e ExchangeMock) (*smocker.Mock, error) {
+	// tokensPerStEth
+	m := smocker.NewSubstringMatcher("0x9576a0c8")
+
+	return &smocker.Mock{
+		Request: smocker.MockRequest{
+			Method: smocker.NewStringMatcher("POST"),
+			Path:   smocker.NewStringMatcher("/"),
+			Body: &smocker.BodyMatcher{
+				BodyString: &m,
+			},
+		},
+		Response: &smocker.MockResponse{
+			Status: e.StatusCode,
+			Headers: map[string]smocker.StringSlice{
+				"Content-Type": []string{
+					"application/json",
+				},
+			},
+			Body: fmt.Sprintf(rpcJSONResult, "0x0000000000000000000000000000000000000000000000000cf6c97c561e07bc"),
+		},
+	}, nil
+}
+
+func (b WSTETH) buildstEthPerToken(e ExchangeMock) (*smocker.Mock, error) {
+	// stEthPerToken
+	m := smocker.NewSubstringMatcher("0x035faf82")
+
+	return &smocker.Mock{
+		Request: smocker.MockRequest{
+			Method: smocker.NewStringMatcher("POST"),
+			Path:   smocker.NewStringMatcher("/"),
+			Body: &smocker.BodyMatcher{
+				BodyString: &m,
+			},
+		},
+		Response: &smocker.MockResponse{
+			Status: e.StatusCode,
+			Headers: map[string]smocker.StringSlice{
+				"Content-Type": []string{
+					"application/json",
+				},
+			},
+			Body: fmt.Sprintf(rpcJSONResult, "0x0000000000000000000000000000000000000000000000000edb20f642b06506"),
+		},
+	}, nil
+}

smocker/mock.go

@@ -88,3 +88,10 @@ func NewStringMatcher(value string) StringMatcher {
 		Value:   value,
 	}
 }
+
+func NewSubstringMatcher(value string) StringMatcher {
+	return StringMatcher{
+		Matcher: "ShouldContainSubstring",
+		Value:   value,
+	}
+}