api: add test for migration 25

shellhub-io · Jun 11, 2021 · dbcd00b · dbcd00b
1 parent a4d6113
commit dbcd00b
Showing 1 changed file with 89 additions and 0 deletions.
diff --git a/api/store/mongo/migrations/migration_25_test.go b/api/store/mongo/migrations/migration_25_test.go
@@ -0,0 +1,89 @@
+package migrations
+
+import (
+	"context"
+	"testing"
+	"time"
+
+	"github.com/shellhub-io/shellhub/api/pkg/dbtest"
+	"github.com/shellhub-io/shellhub/pkg/models"
+	"github.com/stretchr/testify/assert"
+	migrate "github.com/xakep666/mongo-migrate"
+	"go.mongodb.org/mongo-driver/bson"
+	"go.mongodb.org/mongo-driver/mongo"
+)
+
+func TestMigration25(t *testing.T) {
+	db := dbtest.DBServer{}
+	defer db.Stop()
+
+	migrations := GenerateMigrations()[:24]
+
+	migrates := migrate.NewMigrate(db.Client().Database("test"), migrations...)
+	err := migrates.Up(migrate.AllAvailable)
+	assert.NoError(t, err)
+
+	version, _, err := migrates.Version()
+	assert.NoError(t, err)
+	assert.Equal(t, uint64(24), version)
+
+	namespace := models.Namespace{
+		Name:     "name",
+		Owner:    "owner",
+		TenantID: "tenant",
+	}
+	_, err = db.Client().Database("test").Collection("namespaces").InsertOne(context.TODO(), namespace)
+	assert.NoError(t, err)
+
+	device := models.Device{
+		Name:     "device",
+		UID:      "uid",
+		Identity: &models.DeviceIdentity{MAC: "mac"},
+		TenantID: "tenant",
+		LastSeen: time.Now(),
+	}
+	_, err = db.Client().Database("test").Collection("devices").InsertOne(context.TODO(), device)
+	assert.NoError(t, err)
+
+	device = models.Device{
+		Name:     "device2",
+		UID:      "uid2",
+		Identity: &models.DeviceIdentity{MAC: "mac"},
+		TenantID: "tenant2",
+		LastSeen: time.Now(),
+	}
+	_, err = db.Client().Database("test").Collection("devices").InsertOne(context.TODO(), device)
+	assert.NoError(t, err)
+
+	device = models.Device{
+		Name:     "device3",
+		UID:      "uid3",
+		Identity: &models.DeviceIdentity{MAC: "mac"},
+		TenantID: "tenant3",
+		LastSeen: time.Now(),
+	}
+	_, err = db.Client().Database("test").Collection("devices").InsertOne(context.TODO(), device)
+	assert.NoError(t, err)
+
+	migration := GenerateMigrations()[24]
+
+	migrates = migrate.NewMigrate(db.Client().Database("test"), migration)
+	err = migrates.Up(migrate.AllAvailable)
+	assert.NoError(t, err)
+
+	version, _, err = migrates.Version()
+	assert.NoError(t, err)
+	assert.Equal(t, uint64(25), version)
+
+	var migratedDevice *models.Device
+	err = db.Client().Database("test").Collection("devices").FindOne(context.TODO(), bson.M{"tenant_id": "tenant"}).Decode(&migratedDevice)
+	assert.NoError(t, err)
+	assert.Equal(t, "device", migratedDevice.Name)
+
+	err = db.Client().Database("test").Collection("devices").FindOne(context.TODO(), bson.M{"tenant_id": "tenant2"}).Decode(&models.Namespace{})
+	assert.EqualError(t, mongo.ErrNoDocuments, err.Error())
+
+	err = db.Client().Database("test").Collection("devices").FindOne(context.TODO(), bson.M{"tenant_id": "tenant3"}).Decode(&models.Namespace{})
+	assert.EqualError(t, mongo.ErrNoDocuments, err.Error())
+
+}