102 lines
2.2 KiB
Go
Executable File
102 lines
2.2 KiB
Go
Executable File
package storage
|
|
|
|
import (
|
|
"context"
|
|
"crypto/rand"
|
|
"encoding/base64"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
func testMongoNew(t *testing.T) (*Mongo, func()) {
|
|
b := make([]byte, 5)
|
|
rand.Read(b)
|
|
ns := "gotest_" + base64.URLEncoding.EncodeToString(b)
|
|
m, err := NewMongo(10, ns, "mongodb://localhost:27017")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
return m, func() {
|
|
ctx, can := context.WithTimeout(context.Background(), time.Second*30)
|
|
defer can()
|
|
m.client.Database(ns).Drop(ctx)
|
|
m.Close()
|
|
}
|
|
}
|
|
|
|
func TestMongoNew(t *testing.T) {
|
|
_, can := testMongoNew(t)
|
|
can()
|
|
}
|
|
|
|
func TestMongoFind(t *testing.T) {
|
|
m, can := testMongoNew(t)
|
|
defer can()
|
|
|
|
c := m.Account()
|
|
if _, err := c.InsertMany(context.TODO(), []interface{}{
|
|
map[string]interface{}{"_id": "1", "a": "b"},
|
|
map[string]interface{}{"_id": "2", "a": "b", "c": "d"},
|
|
map[string]interface{}{"_id": "3", "c": "d"},
|
|
}); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
cnt := 0
|
|
inc := func() interface{} {
|
|
cnt += 1
|
|
var v interface{}
|
|
return &v
|
|
}
|
|
if err := m.Find(c, map[string]interface{}{"a": "b"}, inc); err != nil {
|
|
t.Fatal(err)
|
|
} else if cnt != 2 {
|
|
t.Fatal(cnt)
|
|
}
|
|
|
|
cnt = 0
|
|
if err := m.Find(c, map[string]interface{}{"_id": "1"}, inc); err != nil {
|
|
t.Fatal(err)
|
|
} else if cnt != 1 {
|
|
t.Fatal(cnt)
|
|
}
|
|
|
|
cnt = 0
|
|
if err := m.Find(c, map[string]interface{}{}, inc); err != nil {
|
|
t.Fatal(err)
|
|
} else if cnt != 3 {
|
|
t.Fatal(cnt)
|
|
}
|
|
}
|
|
|
|
func TestMongoUpsert(t *testing.T) {
|
|
m, can := testMongoNew(t)
|
|
defer can()
|
|
|
|
c := m.Account()
|
|
if _, err := c.InsertMany(context.TODO(), []interface{}{
|
|
map[string]interface{}{"_id": "1", "a": "b"},
|
|
map[string]interface{}{"_id": "2", "a": "b", "c": "d"},
|
|
map[string]interface{}{"_id": "3", "c": "d"},
|
|
}); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if n, err := c.EstimatedDocumentCount(context.TODO()); err != nil {
|
|
t.Fatal(err)
|
|
} else if n != 3 {
|
|
t.Fatal(n)
|
|
}
|
|
|
|
if err := m.Upsert(c, map[string]interface{}{"_id": "1"}, map[string]interface{}{"$set": map[string]interface{}{"c": "d"}}); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
mapped := map[string]string{}
|
|
if err := c.FindOne(context.TODO(), map[string]interface{}{"_id": "1"}).Decode(&mapped); err != nil {
|
|
t.Fatal(err)
|
|
} else if mapped["c"] != "d" {
|
|
t.Fatal(mapped)
|
|
}
|
|
}
|