Function TestAllDocsQuery

db/query_test.go:183–249

(t *testing.T)

}

func TestAllDocsQuery(t *testing.T) {

	db, ctx := setupTestDB(t)
	defer db.Close(ctx)
	collection, ctx := GetSingleDatabaseCollectionWithUser(ctx, t, db)

	// Add docs with channel assignment
	for i := 1; i <= 10; i++ {
		_, _, err := collection.Put(ctx, fmt.Sprintf("allDocsTest%d", i), Body{"channels": []string{"ABC"}})
		assert.NoError(t, err, "Put allDocsTest doc")
	}

	// Standard query
	startKey := "a"
	endKey := ""
	results, queryErr := collection.QueryAllDocs(base.TestCtx(t), startKey, endKey)
	assert.NoError(t, queryErr, "Query error")
	var row map[string]interface{}
	rowCount := 0
	for results.Next(ctx, &row) {
		t.Logf("row[%d]: %v", rowCount, row)
		rowCount++
	}
	assert.Equal(t, 10, rowCount)
	assert.NoError(t, results.Close())

	// Attempt to invalidate standard query
	startKey = "a' AND 1=0\x00"
	endKey = ""
	results, queryErr = collection.QueryAllDocs(base.TestCtx(t), startKey, endKey)
	assert.NoError(t, queryErr, "Query error")
	rowCount = 0
	for results.Next(ctx, &row) {
		t.Logf("row[%d]: %v", rowCount, row)
		rowCount++
	}
	assert.Equal(t, 10, rowCount)
	assert.NoError(t, results.Close())

	// Attempt to invalidate statement to add row to resultset
	startKey = `a' UNION ALL SELECT TOSTRING(BASE64_DECODE("SW52YWxpZERhdGE=")) as id;` + "\x00"
	endKey = ""
	results, queryErr = collection.QueryAllDocs(base.TestCtx(t), startKey, endKey)
	assert.NoError(t, queryErr, "Query error")
	rowCount = 0
	for results.Next(ctx, &row) {
		t.Logf("row[%d]: %v", rowCount, row)
		assert.NotEqual(t, row["id"], "InvalidData")
		rowCount++
	}
	assert.Equal(t, 10, rowCount)
	assert.NoError(t, results.Close())

	// Attempt to create syntax error
	startKey = `a'1`
	endKey = ""
	results, queryErr = collection.QueryAllDocs(base.TestCtx(t), startKey, endKey)
	assert.NoError(t, queryErr, "Query error")