SM-856 - unit test (#111)

* unit test

* Update test

* Remove vars

---------

Co-authored-by: ricardo <chenton147@hotmail.com>

internal/repositories/devicedefinition/devicedefinition_test.go

@@ -0,0 +1,133 @@
+package devicedefinition
+
+import (
+	"context"
+	"encoding/json"
+	"github.com/DIMO-Network/identity-api/graph/model"
+	"github.com/DIMO-Network/identity-api/internal/config"
+	"github.com/DIMO-Network/identity-api/internal/helpers"
+	"github.com/DIMO-Network/identity-api/internal/repositories/base"
+	"github.com/DIMO-Network/identity-api/internal/services"
+	"github.com/DIMO-Network/identity-api/models"
+	"github.com/ethereum/go-ethereum/common"
+	"github.com/jarcoal/httpmock"
+	"github.com/rs/zerolog"
+	"github.com/stretchr/testify/assert"
+	"github.com/volatiletech/null/v8"
+	"github.com/volatiletech/sqlboiler/v4/boil"
+	"net/http"
+	"testing"
+)
+
+const migrationsDir = "../../../migrations"
+
+func Test_GetDeviceDefinitions_Query(t *testing.T) {
+	ctx := context.Background()
+
+	pdb, _ := helpers.StartContainerDatabase(ctx, t, migrationsDir)
+
+	mfr := models.Manufacturer{
+		ID:      137,
+		Name:    "Toyota",
+		Owner:   common.FromHex("0xaba3A41bd932244Dd08186e4c19F1a7E48cbcDf4"),
+		Slug:    "toyota",
+		TableID: null.IntFrom(1),
+	}
+	err := mfr.Insert(ctx, pdb.DBS().Writer, boil.Infer())
+	assert.NoError(t, err)
+
+	logger := zerolog.Nop()
+
+	const baseURL = "http://local"
+
+	repo := base.NewRepository(pdb, config.Settings{DIMORegistryChainID: 30001, TablelandAPIGateway: baseURL}, &logger)
+
+	tablelandAPI := services.NewTablelandApiService(&logger, &config.Settings{
+		TablelandAPIGateway: baseURL,
+	})
+
+	adController := Repository{Repository: repo, TablelandApiService: tablelandAPI}
+	last := 2
+	before := "MQ=="
+
+	httpmock.Activate()
+	defer httpmock.DeactivateAndReset()
+
+	countURL := "api/v1/query?statement=SELECT+COUNT%28%2A%29+FROM+%22_30001_1%22"
+	respCountBody := `[{"count(*)": 4}]`
+	var modelCountTablelandResponse []DeviceDefinitionTablelandCountModel
+	_ = json.Unmarshal([]byte(respCountBody), &modelCountTablelandResponse)
+
+	httpmock.RegisterResponder(http.MethodGet, baseURL+countURL, httpmock.NewStringResponder(200, respCountBody))
+
+	queryURL := "api/v1/query?statement=SELECT+%2A+FROM+%22_30001_1%22+WHERE+%28%22id%22+%3C+%271%27%29+ORDER+BY+%22id%22+DESC+LIMIT+3"
+	respQueryBody := `[
+	  {
+		"id": "alfa-romeo_147_2007",
+		"deviceType": "vehicle",
+		"imageURI": "https://image",
+		"ksuid": "26G3iFH7Xc9Wvsw7pg6sD7uzoSS",
+		"metadata": {
+		  "device_attributes": [
+			{
+			  "name": "powertrain_type",
+			  "value": "ICE"
+			}
+		  ]
+		}
+	  },
+	  {
+		"id": "bmw_x5_2019",
+		"deviceType": "vehicle",
+		"imageURI": "https://image",
+		"ksuid": "12G3iFH7Xc9Wvsw7pg6sD7uzoKK",
+		"metadata": {
+		  "device_attributes": [
+			{
+			  "name": "powertrain_type",
+			  "value": "EV"
+			}
+		  ]
+		}
+	  }
+	]`
+	var modelQueryTablelandResponse []DeviceDefinitionTablelandModel
+	_ = json.Unmarshal([]byte(respQueryBody), &modelQueryTablelandResponse)
+
+	httpmock.RegisterResponder(http.MethodGet, baseURL+queryURL, httpmock.NewStringResponder(200, respQueryBody))
+
+	res, err := adController.GetDeviceDefinitions(ctx, &mfr.TableID.Int, nil, nil, &last, &before, &model.DeviceDefinitionFilter{})
+	assert.NoError(t, err)
+
+	assert.Len(t, res.Edges, 2)
+	assert.Equal(t, res.TotalCount, 4)
+
+	deviceType := "vehicle"
+	legacyID1 := "26G3iFH7Xc9Wvsw7pg6sD7uzoSS"
+	legacyID2 := "12G3iFH7Xc9Wvsw7pg6sD7uzoKK"
+
+	expected := []*model.DeviceDefinitionEdge{
+		{
+			Node: &model.DeviceDefinition{
+				DeviceDefinitionID: "bmw_x5_2019",
+				LegacyID:           &legacyID2,
+				DeviceType:         &deviceType,
+			},
+			Cursor: "Mg==",
+		},
+		{
+			Node: &model.DeviceDefinition{
+				DeviceDefinitionID: "alfa-romeo_147_2007",
+				LegacyID:           &legacyID1,
+				DeviceType:         &deviceType,
+			},
+			Cursor: "Mw==",
+		},
+	}
+
+	for i, e := range expected {
+		assert.Equal(t, e.Node.DeviceDefinitionID, res.Edges[i].Node.DeviceDefinitionID)
+		assert.Equal(t, e.Node.LegacyID, res.Edges[i].Node.LegacyID)
+		assert.Equal(t, e.Node.DeviceType, res.Edges[i].Node.DeviceType)
+	}
+}

internal/services/tableland_api_service_test.go

@@ -0,0 +1,54 @@
+package services
+
+import (
+	"context"
+	"net/http"
+	"os"
+	"testing"
+
+	"github.com/DIMO-Network/identity-api/internal/config"
+	"github.com/jarcoal/httpmock"
+	"github.com/rs/zerolog"
+	"github.com/stretchr/testify/require"
+)
+
+func Test_Query_Tableland_Success(t *testing.T) {
+	logger := zerolog.New(os.Stderr).With().Timestamp().Logger()
+
+	const baseURL = "http://local"
+
+	tablelandAPI := NewTablelandApiService(&logger, &config.Settings{
+		TablelandAPIGateway: baseURL,
+	})
+
+	httpmock.Activate()
+	defer httpmock.DeactivateAndReset()
+
+	statement := "SELECT * FROM table"
+	expectedURL := "api/v1/query?statement=SELECT+%2A+FROM+table"
+
+	respBody := `[
+		  {
+			"id": "alfa-romeo_147_2007",
+			"deviceType": "vehicle",
+			"imageURI": "https://image",
+			"ksuid": "26G3iFH7Xc9Wvsw7pg6sD7uzoSS",
+			"metadata": {
+			  "device_attributes": [
+				{
+				  "name": "powertrain_type",
+				  "value": "ICE"
+				}
+			  ]
+			}
+		  }
+		]`
+
+	httpmock.RegisterResponder(http.MethodGet, baseURL+expectedURL, httpmock.NewStringResponder(200, respBody))
+
+	var result []map[string]interface{}
+	err := tablelandAPI.Query(context.Background(), statement, &result)
+
+	require.NoError(t, err)
+	require.NotEmpty(t, result)
+}