Implement executing queries using API

api/tests/documents_test.go

@@ -0,0 +1,89 @@
+package tests_test
+
+import (
+	"context"
+	"encoding/json"
+	"fmt"
+	"reflect"
+	"testing"
+
+	"github.com/Azure/azure-sdk-for-go/sdk/data/azcosmos"
+	"github.com/pikami/cosmium/internal/repositories"
+	"github.com/stretchr/testify/assert"
+)
+
+func testCosmosQuery(t *testing.T, collectionClient *azcosmos.ContainerClient, query string, expectedData []map[string]interface{}) {
+	pager := collectionClient.NewQueryItemsPager(
+		query,
+		azcosmos.PartitionKey{},
+		&azcosmos.QueryOptions{})
+
+	context := context.TODO()
+	items := make([]map[string]interface{}, 0)
+
+	for pager.More() {
+		response, err := pager.NextPage(context)
+		assert.Nil(t, err)
+
+		for _, bytes := range response.Items {
+			var item map[string]interface{}
+			err := json.Unmarshal(bytes, &item)
+			assert.Nil(t, err)
+
+			items = append(items, item)
+		}
+	}
+
+	assert.Equal(t, len(expectedData), len(items))
+	if !reflect.DeepEqual(items, expectedData) {
+		t.Errorf("executed query does not match expected data.\nExpected: %+v\nGot: %+v", expectedData, items)
+	}
+}
+
+func Test_Documents(t *testing.T) {
+	repositories.CreateCollection(testDatabaseName, repositories.Collection{
+		ID: testCollectionName,
+		PartitionKey: struct {
+			Paths   []string "json:\"paths\""
+			Kind    string   "json:\"kind\""
+			Version int      "json:\"Version\""
+		}{
+			Paths: []string{"/pk"},
+		},
+	})
+	repositories.CreateDocument(testDatabaseName, testCollectionName, map[string]interface{}{"id": "12345", "pk": "123", "isCool": false})
+	repositories.CreateDocument(testDatabaseName, testCollectionName, map[string]interface{}{"id": "67890", "pk": "456", "isCool": true})
+
+	ts := runTestServer()
+	defer ts.Close()
+
+	client, err := azcosmos.NewClientFromConnectionString(
+		fmt.Sprintf("AccountEndpoint=%s;AccountKey=%s", ts.URL, "asas"),
+		&azcosmos.ClientOptions{},
+	)
+	assert.Nil(t, err)
+
+	collectionClient, err := client.NewContainer(testDatabaseName, testCollectionName)
+	assert.Nil(t, err)
+
+	t.Run("Should query document", func(t *testing.T) {
+		testCosmosQuery(t, collectionClient,
+			"SELECT c.id, c[\"pk\"] FROM c",
+			[]map[string]interface{}{
+				{"id": "12345", "pk": "123"},
+				{"id": "67890", "pk": "456"},
+			},
+		)
+	})
+
+	t.Run("Should query document with single WHERE condition", func(t *testing.T) {
+		testCosmosQuery(t, collectionClient,
+			`select c.id
+			FROM c
+			WHERE c.isCool=true`,
+			[]map[string]interface{}{
+				{"id": "67890"},
+			},
+		)
+	})
+}