Update CI settings

Author: ipfans

.travis.yml

@@ -14,3 +14,6 @@ before_install:
 script:
   - go test -v ./...
   - go tool vet .
+
+after_success:
+- bash <(curl -s https://codecov.io/bash)

README.md

@@ -2,15 +2,16 @@
 
 [![GoDoc](http://img.shields.io/badge/godoc-reference-5272B4.svg?style=flat-square)](https://godoc.org/github.com/yeeuu/mongo)
 [![Build Status](https://travis-ci.org/yeeuu/mongo.svg?branch=master)](https://travis-ci.org/yeeuu/mongo)
+[![codecov](https://codecov.io/gh/yeeuu/mongo/branch/master/graph/badge.svg)](https://codecov.io/gh/yeeuu/mongo)
 [![Go Report Card](https://goreportcard.com/badge/github.com/yeeuu/mongo)](https://goreportcard.com/report/github.com/yeeuu/mongo)
 
 
 Simple ODM wraps mgo with interface.
 
 ## TODO
- * [ ] 0.1.0 release
- * [ ] Test coverage
- * [ ] Travis-ci support
+ - [ ] 0.1.0 release
+ - [ ] Test coverage
+ - [x] Travis-ci support
 
 ## Limits
 

storage_test.go

@@ -0,0 +1,49 @@
+package mongo
+
+import (
+	"fmt"
+	"testing"
+	"time"
+
+	"github.com/globalsign/mgo"
+)
+
+func TestStorage(t *testing.T) {
+	sess, err := mgo.DialWithTimeout("127.0.0.1", 2*time.Second)
+	if err != nil {
+		panic(err)
+	}
+	defer sess.Close()
+	sess.DB("test").DropDatabase()
+	store := NewStorage(sess)
+	err = store.Insert(&userSelector{}, user{Name: "hello"})
+	if err != nil {
+		panic(err)
+	}
+	var u user
+	name := "hello"
+	qs, err := store.Query(&userSelector{
+		Name: &name,
+	})
+	if err != nil {
+		panic(err)
+	}
+	err = qs.One(&u)
+	if err != nil {
+		panic(err)
+	}
+	qs.Close()
+	fmt.Println(u)
+	err = store.Insert(&userSelector{}, user{Name: "world"})
+	qs, err = store.Query(&userSelector{})
+	if err != nil {
+		panic(err)
+	}
+	var users []user
+	err = qs.All(&users)
+	if err != nil {
+		panic(err)
+	}
+	qs.Close()
+	fmt.Println(users)
+}

util_test.go

@@ -0,0 +1,165 @@
+package mongo
+
+import (
+	"reflect"
+	"testing"
+
+	"github.com/globalsign/mgo"
+	"github.com/globalsign/mgo/bson"
+)
+
+type user struct {
+	ID   bson.ObjectId `bson:"_id,omitempty"`
+	Name string        `bson:"name"`
+}
+
+type userSelector struct {
+	ID   *bson.ObjectId   `bson:"_id"`
+	IDs  *[]bson.ObjectId `bson:"_id"`
+	Name *string          `bson:"name"`
+}
+
+func (us *userSelector) Database() string {
+	return "test"
+}
+
+func (us *userSelector) Collection() string {
+	return "users"
+}
+
+func Test_structToBsonM(t *testing.T) {
+	type args struct {
+		s interface{}
+	}
+	id := bson.NewObjectId()
+	ids := []bson.ObjectId{bson.NewObjectId()}
+	name := "hello"
+	tests := []struct {
+		name    string
+		args    args
+		want    bson.M
+		wantErr bool
+	}{
+		// TODO: Add test cases.
+		{
+			name: "nil",
+			args: args{
+				nil,
+			},
+			want:    bson.M{},
+			wantErr: true,
+		},
+		{
+			name: "struct",
+			args: args{
+				user{},
+			},
+			want:    bson.M{},
+			wantErr: true,
+		},
+		{
+			name: "empty",
+			args: args{
+				&userSelector{},
+			},
+			want:    bson.M{},
+			wantErr: false,
+		},
+		{
+			name: "name",
+			args: args{
+				&userSelector{Name: &name},
+			},
+			want:    bson.M{"name": name},
+			wantErr: false,
+		},
+		{
+			name: "id",
+			args: args{
+				&userSelector{ID: &id},
+			},
+			want:    bson.M{"_id": id},
+			wantErr: false,
+		},
+		{
+			name: "ids",
+			args: args{
+				&userSelector{IDs: &ids},
+			},
+			want:    bson.M{"_id": bson.M{"$in": ids}},
+			wantErr: false,
+		},
+	}
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			got, err := structToBsonM(tt.args.s)
+			if (err != nil) != tt.wantErr {
+				t.Errorf("structToBsonM() error = %v, wantErr %v", err, tt.wantErr)
+				return
+			}
+			if !reflect.DeepEqual(got, tt.want) {
+				t.Errorf("structToBsonM() = %v, want %v", got, tt.want)
+			}
+		})
+	}
+}
+
+func Test_getCollection(t *testing.T) {
+	sess, err := mgo.Dial("127.0.0.1")
+	if err != nil {
+		t.Skip("Cannot connect to mongodb.")
+		return
+	}
+	defer sess.Close()
+	type args struct {
+		sess  *mgo.Session
+		query interface{}
+	}
+	tests := []struct {
+		name    string
+		args    args
+		wantC   *mgo.Collection
+		wantErr bool
+	}{
+		// TODO: Add test cases.
+		{
+			name: "nil",
+			args: args{
+				sess:  sess,
+				query: nil,
+			},
+			wantC:   nil,
+			wantErr: true,
+		},
+		{
+			name: "struct",
+			args: args{
+				sess:  sess,
+				query: user{},
+			},
+			wantC:   nil,
+			wantErr: true,
+		},
+		{
+			name: "struct pointer",
+			args: args{
+				sess:  sess,
+				query: &user{},
+			},
+			wantC:   nil,
+			wantErr: true,
+		},
+	}
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			gotC, err := getCollection(tt.args.sess, tt.args.query)
+			if (err != nil) != tt.wantErr {
+				t.Errorf("getCollection() error = %v, wantErr %v", err, tt.wantErr)
+				return
+			}
+			if !reflect.DeepEqual(gotC, tt.wantC) {
+				t.Errorf("getCollection() = %v, want %v", gotC, tt.wantC)
+			}
+		})
+	}
+}