Semaphore/db/bolt/migration_2_10_24_test.go

package bolt

import (
	"encoding/json"
	"go.etcd.io/bbolt"
	"testing"
)

func TestMigration_2_10_24_Apply(t *testing.T) {
	store := CreateTestStore()

	err := store.db.Update(func(tx *bbolt.Tx) error {
		b, err := tx.CreateBucketIfNotExists([]byte("project"))
		if err != nil {
			return err
		}

		err = b.Put([]byte("0000000001"), []byte("{}"))
		if err != nil {
			return err
		}

		r, err := tx.CreateBucketIfNotExists([]byte("project__template_0000000001"))
		if err != nil {
			return err
		}

		err = r.Put([]byte("0000000001"),
			[]byte("{\"id\":\"1\",\"project_id\":\"1\",\"vault_key_id\":\"1\"}"))

		return err
	})

	if err != nil {
		t.Fatal(err)
	}

	err = migration_2_10_24{migration{store.db}}.Apply()
	if err != nil {
		t.Fatal(err)
	}

	var template map[string]interface{}
	err = store.db.View(func(tx *bbolt.Tx) error {
		b := tx.Bucket([]byte("project__template_0000000001"))
		str := string(b.Get([]byte("0000000001")))
		return json.Unmarshal([]byte(str), &template)
	})
	if err != nil {
		t.Fatal(err)
	}

	var templateVault map[string]interface{}
	err = store.db.View(func(tx *bbolt.Tx) error {
		b := tx.Bucket([]byte("project__template_vault_0000000001"))
		str := string(b.Get([]byte("0000000001")))
		return json.Unmarshal([]byte(str), &templateVault)
	})
	if err != nil {
		t.Fatal(err)
	}

	if _, ok := template["vault_key_id"]; ok {
		t.Fatal("vault_key_id must be deleted")
	}

	if templateVault["vault_key_id"].(string) != "1" {
		t.Fatal("invalid vault_key_id: " + templateVault["vault_key_id"].(string))
	}
}

func TestMigration_2_10_24_Apply2(t *testing.T) {
	store := CreateTestStore()

	err := store.db.Update(func(tx *bbolt.Tx) error {
		b, err := tx.CreateBucketIfNotExists([]byte("project"))
		if err != nil {
			return err
		}

		err = b.Put([]byte("0000000001"), []byte("{}"))

		return err
	})

	if err != nil {
		t.Fatal(err)
	}

	err = migration_2_10_24{migration{store.db}}.Apply()
	if err != nil {
		t.Fatal(err)
	}
}