77 lines
2.3 KiB
Go
77 lines
2.3 KiB
Go
|
//
|
||
|
|
// schema_test.go
|
||
|
|
// Copyright (C) 2023 tiglog <me@tiglog.com>
|
||
|
|
//
|
||
|
|
// Distributed under terms of the MIT license.
|
||
|
|
//
|
||
|
|
|
||
|
|
package orm
|
||
|
|
|
||
|
|
import (
|
||
|
|
"database/sql"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/stretchr/testify/assert"
|
||
|
|
)
|
||
|
|
|
||
|
|
func setup(t *testing.T) {
|
||
|
|
db, err := sql.Open("sqlite3", ":memory:")
|
||
|
|
err = SetupConnections(ConnectionConfig{
|
||
|
|
Name: "default",
|
||
|
|
DB: db,
|
||
|
|
Dialect: Dialects.SQLite3,
|
||
|
|
})
|
||
|
|
// orm.Schematic()
|
||
|
|
_, err = GetConnection("default").DB.Exec(`CREATE TABLE IF NOT EXISTS posts (id INTEGER PRIMARY KEY, body text, created_at TIMESTAMP, updated_at TIMESTAMP, deleted_at TIMESTAMP)`)
|
||
|
|
_, err = GetConnection("default").DB.Exec(`CREATE TABLE IF NOT EXISTS emails (id INTEGER PRIMARY KEY, post_id INTEGER, email text)`)
|
||
|
|
_, err = GetConnection("default").DB.Exec(`CREATE TABLE IF NOT EXISTS header_pictures (id INTEGER PRIMARY KEY, post_id INTEGER, link text)`)
|
||
|
|
_, err = GetConnection("default").DB.Exec(`CREATE TABLE IF NOT EXISTS comments (id INTEGER PRIMARY KEY, post_id INTEGER, body text)`)
|
||
|
|
_, err = GetConnection("default").DB.Exec(`CREATE TABLE IF NOT EXISTS categories (id INTEGER PRIMARY KEY, title text)`)
|
||
|
|
_, err = GetConnection("default").DB.Exec(`CREATE TABLE IF NOT EXISTS post_categories (post_id INTEGER, category_id INTEGER, PRIMARY KEY(post_id, category_id))`)
|
||
|
|
assert.NoError(t, err)
|
||
|
|
}
|
||
|
|
|
||
|
|
type Object struct {
|
||
|
|
ID int64
|
||
|
|
Name string
|
||
|
|
Timestamps
|
||
|
|
}
|
||
|
|
|
||
|
|
func (o Object) ConfigureEntity(e *EntityConfigurator) {
|
||
|
|
e.Table("objects").Connection("default")
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestGenericFieldsOf(t *testing.T) {
|
||
|
|
t.Run("fields of with id and timestamps embedded", func(t *testing.T) {
|
||
|
|
fs := genericFieldsOf(&Object{})
|
||
|
|
assert.Len(t, fs, 5)
|
||
|
|
assert.Equal(t, "id", fs[0].Name)
|
||
|
|
assert.True(t, fs[0].IsPK)
|
||
|
|
assert.Equal(t, "name", fs[1].Name)
|
||
|
|
assert.Equal(t, "created_at", fs[2].Name)
|
||
|
|
assert.Equal(t, "updated_at", fs[3].Name)
|
||
|
|
assert.Equal(t, "deleted_at", fs[4].Name)
|
||
|
|
})
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestGenericValuesOf(t *testing.T) {
|
||
|
|
t.Run("values of", func(t *testing.T) {
|
||
|
|
|
||
|
|
setup(t)
|
||
|
|
vs := genericValuesOf(Object{}, true)
|
||
|
|
assert.Len(t, vs, 5)
|
||
|
|
})
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestEntityConfigurator(t *testing.T) {
|
||
|
|
t.Run("test has many with user provided values", func(t *testing.T) {
|
||
|
|
setup(t)
|
||
|
|
var ec EntityConfigurator
|
||
|
|
ec.Table("users").Connection("default").HasMany(Object{}, HasManyConfig{
|
||
|
|
"objects", "user_id",
|
||
|
|
})
|
||
|
|
|
||
|
|
})
|
||
|
|
|
||
|
|
}
|