|
1 | | -package handlers |
| 1 | +package handlers_test |
2 | 2 |
|
3 | 3 | import ( |
4 | | - "context" |
5 | 4 | "fmt" |
| 5 | + "github.com/rsksmart/liquidity-provider-server/internal/adapters/entrypoints/rest/handlers" |
6 | 6 | "github.com/rsksmart/liquidity-provider-server/internal/usecases/reports" |
7 | 7 | "github.com/rsksmart/liquidity-provider-server/test/mocks" |
8 | 8 | "github.com/stretchr/testify/assert" |
| 9 | + "github.com/stretchr/testify/mock" |
9 | 10 | "math/big" |
| 11 | + "net/http" |
| 12 | + "net/http/httptest" |
| 13 | + "strings" |
10 | 14 | "testing" |
11 | 15 | ) |
12 | 16 |
|
@@ -35,26 +39,28 @@ func TestNewGetReportsAssetsHandler(t *testing.T) { |
35 | 39 | } |
36 | 40 |
|
37 | 41 | t.Run("should return 200 on success", func(t *testing.T) { |
38 | | - ctx := context.Background() |
39 | | - useCase := mocks.NewAssetsReportUseCaseMock(t) |
| 42 | + useCase := new(mocks.GetAssetsReportUseCaseMock) |
| 43 | + useCase.EXPECT().Run(mock.Anything).Return(successReturn, nil) |
40 | 44 |
|
41 | | - useCase.On("Run", ctx).Return(successReturn, nil) |
42 | | - |
43 | | - handler := NewGetReportsAssetsHandler(useCase) |
44 | | - assert.HTTPSuccess(t, handler, verb, path, nil) |
| 45 | + handler := handlers.NewGetReportsAssetsHandler(useCase) |
| 46 | + reqBody := `{}` |
| 47 | + req := httptest.NewRequest(http.MethodPost, "/reports/assets", strings.NewReader(reqBody)) |
| 48 | + req.Header.Set("Content-Type", "application/json") |
| 49 | + w := httptest.NewRecorder() |
| 50 | + handler(w, req) |
| 51 | + assert.Equal(t, http.StatusOK, w.Code) |
45 | 52 | assert.HTTPBodyContains(t, handler, verb, path, nil, fmt.Sprintf(`"btcBalance":%d`, successReturn.BtcBalance)) |
46 | 53 | assert.HTTPBodyContains(t, handler, verb, path, nil, fmt.Sprintf(`"rbtcBalance":%d`, successReturn.RbtcBalance)) |
47 | 54 | assert.HTTPBodyContains(t, handler, verb, path, nil, fmt.Sprintf(`"btcLocked":%d`, successReturn.BtcLocked)) |
48 | 55 | assert.HTTPBodyContains(t, handler, verb, path, nil, fmt.Sprintf(`"rbtcLocked":%d`, successReturn.RbtcLocked)) |
49 | 56 | assert.HTTPBodyContains(t, handler, verb, path, nil, fmt.Sprintf(`"btcLiquidity":%d`, successReturn.BtcLiquidity)) |
50 | 57 | assert.HTTPBodyContains(t, handler, verb, path, nil, fmt.Sprintf(`"rbtcLiquidity":%d`, successReturn.RbtcLiquidity)) |
| 58 | + useCase.AssertExpectations(t) |
51 | 59 | }) |
52 | 60 | t.Run("Should return an error when use case fail", func(t *testing.T) { |
53 | | - ctx := context.Background() |
54 | | - useCase := mocks.NewAssetsReportUseCaseMock(t) |
55 | | - |
56 | | - useCase.On("Run", ctx).Return(failReturn, assert.AnError) |
57 | | - handler := NewGetReportsAssetsHandler(useCase) |
| 61 | + useCase := new(mocks.GetAssetsReportUseCaseMock) |
| 62 | + useCase.EXPECT().Run(mock.Anything).Return(failReturn, assert.AnError) |
| 63 | + handler := handlers.NewGetReportsAssetsHandler(useCase) |
58 | 64 | assert.HTTPError(t, handler, verb, path, nil) |
59 | 65 | assert.HTTPBodyContains(t, handler, verb, path, nil, `{"error":"assert.AnError general error for testing"}`) |
60 | 66 | }) |
|
0 commit comments