Initial cloud-services repo - gateway service + pkg modules

pkg/mongo/client_test.go

@@ -0,0 +1,81 @@
+package mongo_test
+
+import (
+	"testing"
+
+	"fiskerinc.com/modules/mongo"
+	"fiskerinc.com/modules/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)
+	}
+}