collection_test.go

// Copyright 2011 Gary Burd
//
// Licensed under the Apache License, Version 2.0 (the "License"): you may
// not use this file except in compliance with the License. You may obtain
// a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
// WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
// License for the specific language governing permissions and limitations
// under the License.

package mongo

import (
	"testing"
)

func TestFindAndModify(t *testing.T) {

	c := dialAndDrop(t, "go-mongo-test", "test")
	defer c.Conn.Close()

	var m M
	err := c.FindAndUpdate(
		M{"_id": "users"},
		M{"$inc": M{"seq": 1}},
		&FindAndModifyOptions{New: true, Upsert: true},
		&m)
	if err != nil {
		t.Fatal("FindAndModify", err)
	}

	t.Log("foo", m)
	if m["seq"] != 1 {
		t.Fatalf("m[seq]=%v, want 1", m["seq"])
	}

	m = nil
	err = c.FindAndUpdate(M{"_id": "users"}, M{"$inc": M{"seq": 1}}, nil, &m)
	if err != nil {
		t.Fatal("findAndUpdate", err)
	}

	t.Log("bar", m)
	if m["seq"] != 1 {
		t.Fatalf("m[seq]=%v, want 1", m["seq"])
	}

	m = nil
	err = c.FindAndRemove(M{"_id": "users"}, nil, &m)
	if err != nil {
		t.Fatal("findAndRemove", err)
	}

	if m["seq"] != 2 {
		t.Fatalf("expect m[seq]=%v, want 2", m["seq"])
	}

	m = nil
	err = c.Find(M{"_id": "users"}).One(&m)
	if err != EOF {
		t.Fatal("findone, expect EOF, got", err)
	}
}

var indexNameTests = []struct {
	keys D
	name string
}{
	{D{{"up", 1}, {"down", -1}, {"geo", "2d"}}, "up_1_down_-1_geo_2d"},
}

func TestIndexName(t *testing.T) {
	for _, tt := range indexNameTests {
		name := IndexName(tt.keys)
		if name != tt.name {
			t.Errorf("%v, name=%s, want %s\n", tt.keys, name, tt.name)
		}
	}
}