82 lines
1.9 KiB
Go
82 lines
1.9 KiB
Go
package mongo_test
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/fiskerinc/cloud-services/pkg/mongo"
|
|
"github.com/fiskerinc/cloud-services/pkg/testhelper"
|
|
)
|
|
|
|
func TestNewClient(t *testing.T) {
|
|
_, err := mongo.NewClient(mongo.StandardDB)
|
|
if err != nil {
|
|
t.Errorf(testhelper.TestErrorTemplate, "TestNewClient", nil, err)
|
|
}
|
|
}
|
|
|
|
func TestClientGetVehicles(t *testing.T) {
|
|
client, err := mongo.NewClient(mongo.StandardDB)
|
|
if err != nil {
|
|
t.Errorf(testhelper.TestErrorTemplate, "TestClientGetVehicles", nil, err)
|
|
}
|
|
|
|
client.GetVehicles()
|
|
}
|
|
|
|
func TestClientSetVehicles(t *testing.T) {
|
|
client, err := mongo.NewClient(mongo.StandardDB)
|
|
if err != nil {
|
|
t.Errorf(testhelper.TestErrorTemplate, "TestClientSetVehicles", nil, err)
|
|
}
|
|
|
|
client.SetVehicles(mongo.NewVehiclesCollection(
|
|
mongo.NewCollection(client.Collection("vehicles")),
|
|
))
|
|
}
|
|
|
|
func TestClientGetFleets(t *testing.T) {
|
|
client, err := mongo.NewClient(mongo.StandardDB)
|
|
if err != nil {
|
|
t.Errorf(testhelper.TestErrorTemplate, "TestClientGetFleets", nil, err)
|
|
}
|
|
|
|
client.GetFleets()
|
|
}
|
|
|
|
func TestClientSetFleets(t *testing.T) {
|
|
client, err := mongo.NewClient(mongo.StandardDB)
|
|
if err != nil {
|
|
t.Errorf(testhelper.TestErrorTemplate, "TestClientSetFleets", nil, err)
|
|
}
|
|
|
|
client.SetFleets(mongo.NewFleetsCollection(
|
|
mongo.NewCollection(client.Collection("fleets")),
|
|
))
|
|
}
|
|
|
|
func TestClientCollection(t *testing.T) {
|
|
client, err := mongo.NewClient(mongo.StandardDB)
|
|
if err != nil {
|
|
t.Errorf(testhelper.TestErrorTemplate, "TestClientCollection", nil, err)
|
|
return
|
|
}
|
|
|
|
c := client.Collection("test")
|
|
if c == nil {
|
|
t.Errorf(testhelper.TestErrorTemplate, "TestClientCollection", "*mongo.Collection", c)
|
|
}
|
|
}
|
|
|
|
func TestClientClose(t *testing.T) {
|
|
client, err := mongo.NewClient(mongo.StandardDB)
|
|
if err != nil {
|
|
t.Errorf(testhelper.TestErrorTemplate, "TestClientClose", nil, err)
|
|
return
|
|
}
|
|
|
|
err = client.Close()
|
|
if err != nil {
|
|
t.Errorf(testhelper.TestErrorTemplate, "TestClientClose", nil, err)
|
|
}
|
|
}
|