You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
92 lines
2.0 KiB
92 lines
2.0 KiB
3 years ago
|
package db
|
||
|
3 years ago
|
|
||
|
|
import (
|
||
|
|
"database/sql"
|
||
|
|
"fmt"
|
||
|
|
"io/ioutil"
|
||
|
|
"log"
|
||
|
|
"os"
|
||
|
|
"testing"
|
||
|
|
)
|
||
|
|
|
||
|
|
func assert_string(t *testing.T, expected string, actual string) {
|
||
|
|
if actual != expected {
|
||
|
|
t.Fatalf("(%s, %s)", expected, actual)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
func assert_int(t *testing.T, expected int, actual int) {
|
||
|
|
if actual != expected {
|
||
|
|
t.Fatalf("(%d, %d)", expected, actual)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestInitBasic(t *testing.T) {
|
||
|
|
var db_location = ":memory:"
|
||
|
|
ctx, err := Init(db_location)
|
||
|
|
if err != nil {
|
||
|
|
t.Fatalf(err.Error())
|
||
|
|
}
|
||
|
|
// check that the tables exist
|
||
|
|
var table_name string
|
||
|
|
|
||
|
|
ctx.db.QueryRow(
|
||
|
|
"SELECT name FROM sqlite_master WHERE type='table' AND name='eng_log_version';",
|
||
|
|
).Scan(&table_name)
|
||
|
|
assert_string(t, "eng_log_version", table_name)
|
||
|
|
|
||
|
|
ctx.db.QueryRow(
|
||
|
|
"SELECT name FROM sqlite_master WHERE type='table' AND name='log';",
|
||
|
|
).Scan(&table_name)
|
||
|
|
assert_string(t, "log", table_name)
|
||
|
|
|
||
|
|
// Check that the version stored is correct
|
||
|
|
var version int
|
||
|
|
ctx.db.QueryRow("SELECT id FROM eng_log_version").Scan(&version)
|
||
|
|
assert_int(t, __version, ctx.version)
|
||
|
|
assert_int(t, __version, version)
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestInitLogVersionTableExists(t *testing.T) {
|
||
|
|
var db_location = ":memory:"
|
||
|
|
db, err := sql.Open("sqlite3", db_location)
|
||
|
|
|
||
|
|
db.Exec(`
|
||
|
|
CREATE TABLE eng_log_version (id INTEGER NOT NULL);
|
||
|
|
INSERT INTO eng_log_version (id) VALUES (1);
|
||
|
|
`)
|
||
|
|
_, err = Init(db_location)
|
||
|
|
if err != nil {
|
||
|
|
t.Fatalf(err.Error())
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestInitWrongLogVersionExists(t *testing.T) {
|
||
|
|
file, err := ioutil.TempFile("", "log_test_init_wrong.*.db")
|
||
|
|
defer os.Remove(file.Name()) // clean up
|
||
|
|
|
||
|
|
db, err := sql.Open("sqlite3", file.Name())
|
||
|
|
if err != nil {
|
||
|
|
log.Fatal(err.Error())
|
||
|
|
}
|
||
|
|
|
||
|
|
db.Exec(`
|
||
|
|
CREATE TABLE eng_log_version (id INTEGER NOT NULL);
|
||
|
|
`)
|
||
|
|
db.Exec(`
|
||
|
|
INSERT INTO eng_log_version (id) VALUES (?);
|
||
|
|
`, __version+1)
|
||
|
|
|
||
|
|
var version int
|
||
|
|
db.QueryRow("SELECT id FROM eng_log_version LIMIT 1").Scan(&version)
|
||
|
|
|
||
|
|
_, err = Init(file.Name())
|
||
|
|
if err == nil {
|
||
|
|
t.Fatal("No error when error expected")
|
||
|
|
}
|
||
|
|
assert_string(
|
||
|
|
t,
|
||
|
|
fmt.Sprintf("Wrong version. Expected %d got %d", __version, __version+1),
|
||
|
|
err.Error(),
|
||
|
|
)
|
||
|
|
}
|