arangomigo_test.go

package main

import (
	"context"
	"log"
	"sort"
	"testing"

	driver "github.com/arangodb/go-driver"
	"github.com/stretchr/testify/assert"
)

func TestFullMigration(t *testing.T) {
	configFile := "testdata/complete/config.yaml"

	conf, err := loadConf(configFile)
	if e(err) {
		log.Fatal(err)
	}

	ctx := context.Background()

	cl, err := client(ctx, *conf)
	if e(err) {
		log.Fatal(err)
	}

	db, err := cl.Database(ctx, conf.Db)
	if err == nil {
		err := db.Remove(ctx)
		if e(err) {
			t.Fatal("Couldn't prepare for test")
		}
	}

	_, err = cl.Database(ctx, conf.Db)
	if !driver.IsNotFound(err) {
		t.Fatal("Could not connect to the Database", err)
	}

	triggerMigration(configFile)

	// Look to see if everything was made properly.
	db, err = cl.Database(ctx, conf.Db)
	assert.NoError(t, err, "Unable to find the database")

	recipes, err := db.Collection(ctx, "recipes")
	assert.NoError(t, err, "Could not find recipes collection")

	// Should find the custom recipe inserted by AQL.
	desiredKey := "hello"
	r := recipe{}
	md, err := recipes.ReadDocument(ctx, desiredKey, &r)
	assert.Equal(t, md.Key, desiredKey, "Meta data should match desired key.")
	assert.Equal(t, r.Key, desiredKey, "Document key should match desired key.")
	assert.Equal(t, "Lots of mayo", r.WithEscaped, "Should have updated the escaped var.")
	assert.Equal(t, "Fish", r.MeatType, "Should not have changed.")
	assert.Equal(t, "Taco Fishy", r.Name)

	// Can't really tell which indexes are available, just that recipes should have
	// 6: 1 for the PK and 5 others.
	idxs, err := recipes.Indexes(ctx)
	assert.Equal(t, 6, len(idxs), "Recipes should have 6 indexes")

	// Make sure wait for sync sticks.
	colprop, err := recipes.Properties(ctx)
	assert.True(t, colprop.WaitForSync, "Should wait for sync.")

	g, err := db.Graph(ctx, "testing_graph")
	assert.NoError(t, err, "Should have gotten a graph")
	vcs, err := g.VertexCollections(ctx)
	assert.NoError(t, err, "Should have gotten vertices")

	// Vertices include those in edges and oraphans.
	justNames := []string{}
	for _, k := range vcs {
		justNames = append(justNames, k.Name())
	}
	sort.Strings(justNames)

	assert.Equal(t, []string{"another", "recipes", "users"}, justNames)
}

type recipe struct {
	Name        string
	WithEscaped string
	MeatType    string
	Key         string `json:"_key"`
}