package main

import (
	"database/sql"
	"testing"

	_ "github.com/mattn/go-sqlite3"
)

func TestMigrations(t *testing.T) {
	db, err := sql.Open("sqlite3", ":memory:")
	if err != nil {
		t.Fatal(err)
	}
	defer db.Close()

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

	var count int
	row := db.QueryRow("select count(name) from migrations")
	row.Scan(&count)
	if count != 1 {
		t.Fatalf("Unexpected migration count: %d", count)
	}
}