view_test.go

package sofa

import (
	"fmt"
	"testing"

	"github.com/nbio/st"
	"github.com/h2non/gock"
)

const TestViewFunc = `function(doc) {
    if (doc.type !== undefined && doc.name !== undefined) {
        emit(doc.type, doc.name)
    }
}`

func TestTemporaryView(t *testing.T) {
	defer gock.Off()

	// TODO: Check the data which gets sent
	gock.New(fmt.Sprintf("https://%s", globalTestConnections.Version1MockHost)).
		Post("/view_test_db/_temp_view").
		BodyString(`{"map":"function(doc) {\n    if (doc.type !== undefined \u0026\u0026 doc.name !== undefined) {\n        emit(doc.type, doc.name)\n    }\n}"}`).
		Reply(200).
		JSON(map[string]interface{}{
			"total_rows": 2,
			"offset":     0,
			"rows": []map[string]interface{}{
				{
					"id":    "fruit1",
					"key":   "fruit",
					"value": "apple",
				},
				{
					"id":    "fruit2",
					"key":   "fruit",
					"value": "apple",
				},
			},
		})

	con := globalTestConnections.Version1(t, true)
	db := con.Database("view_test_db")

	view := db.TemporaryView(TestViewFunc)
	result, err := view.Execute(ViewParams{})
	st.Assert(t, err, nil)

	st.Assert(t, result.TotalRows, float64(2))
	st.Assert(t, result.Offset, float64(0))
}

func TestNamedView(t *testing.T) {
	defer gock.Off()

	gock.New(fmt.Sprintf("https://%s", globalTestConnections.Version1MockHost)).
		Get("/view_test_db/_design/things/_view/byType").
		MatchParam("reduce", "false").
		Reply(200).
		JSON(map[string]interface{}{
			"total_rows": 2,
			"offset":     0,
			"rows": []map[string]interface{}{
				{
					"id":    "fruit1",
					"key":   "fruit",
					"value": "apple",
				},
				{
					"id":    "fruit2",
					"key":   "fruit",
					"value": "apple",
				},
			},
		})

	con := globalTestConnections.Version1(t, true)
	db := con.Database("view_test_db")
	view := db.NamedView("things", "byType")

	result, err := view.Execute(ViewParams{Reduce: False})
	st.Assert(t, err, nil)

	st.Assert(t, result.TotalRows, float64(2))
	st.Assert(t, result.Offset, float64(0))
}

// TODO: Currently only tests a temporary view. Would be nice to also create a
//       named view and test that too.
func TestViewReal(t *testing.T) {
	con := globalTestConnections.Version1(t, false)

	// Delete the DB if it currently exists
	if err := con.DeleteDatabase("feed_test_db"); err != nil {
		if !ErrorStatus(err, 404) {
			t.Fatal(err)
		}
	}

	db, err := con.CreateDatabase("feed_test_db")
	st.Assert(t, err, nil)

	defer func() {
		if err := con.DeleteDatabase("feed_test_db"); err != nil {
			t.Fatal(err)
		}
	}()

	appleDoc := getDefaultTestDoc()

	kiwiDoc := &struct {
		DocumentMetadata
		Name string `json:"name"`
		Type string `json:"type"`
	}{
		DocumentMetadata: DocumentMetadata{
			ID: "fruit2",
		},
		Name: "kiwi",
		Type: "fruit",
	}

	_, err = db.Put(appleDoc)
	st.Assert(t, err, nil)

	_, err = db.Put(kiwiDoc)
	st.Assert(t, err, nil)

	view := db.TemporaryView(TestViewFunc)
	result, err := view.Execute(ViewParams{})
	st.Assert(t, err, nil)

	st.Assert(t, result.TotalRows, float64(2))
	st.Assert(t, result.Offset, float64(0))
}