Repository impl & tests, move tests, comp. pk insert support

2026-05-17 22:44:04 +02:00
parent 1642a01aad
commit e98dbe3338
7 changed files with 357 additions and 5 deletions
@@ -0,0 +1,486 @@
+package simpleorm_test
+
+import (
+	"os"
+	"testing"
+	"time"
+
+	simpleorm "git.gdulai.com/gdulai/simpleorm"
+	"git.gdulai.com/gdulai/simpleorm/exec"
+	log "gitlab.com/gdulai/simpleloglvl"
+)
+
+func cleanUp(dbFile string, conn *simpleorm.DBConnection) {
+	conn.Close()
+	os.Remove(dbFile)
+}
+
+func TestSelectEmpty(t *testing.T) {
+	// GIVEN
+	conn := simpleorm.OpenConnection("sqlite3", "file:test.db")
+	defer cleanUp("test.db", conn)
+
+	orm := simpleorm.NewORM(Test{}, TestWithFk{}, TestWithFkAndFkId{},
+		TestWithCompositePk{}, TestWithCompositeFk{})
+
+	exec.ExecuteDDL(conn, orm)
+
+	// WHEN
+	selectExec, err := exec.CreateSelect[Test](orm, "")
+
+	// THEN
+
+	if err != nil {
+		log.LogError("Failed to create select. %s", err)
+		t.Fail()
+		return
+	}
+
+	res, err := selectExec.Execute(conn)
+
+	if err != nil {
+		log.LogError("Select failure. %s", err)
+		t.Fail()
+		return
+	}
+	if len(res) > 0 {
+		log.LogError("Expected empty result.")
+		t.Fail()
+		return
+	}
+}
+
+func TestInsertAndSelectSingle(t *testing.T) {
+	// GIVEN
+	conn := simpleorm.OpenConnection("sqlite3", "file:test.db")
+	defer cleanUp("test.db", conn)
+	orm := simpleorm.NewORM(Test{}, TestWithFk{}, TestWithFkAndFkId{},
+		TestWithCompositePk{}, TestWithCompositeFk{})
+
+	exec.ExecuteDDL(conn, orm)
+
+	testObj := Test{Int64Field: 54, IntField: 12, StringField: "asdf"}
+
+	insertExec, err := exec.NewInsert(orm, testObj)
+	if err != nil {
+		log.LogError("TestInsertAndSelectSingle setup failed: %s", err)
+		t.Fail()
+		return
+	}
+
+	selectExec, err := exec.CreateSelect[Test](orm, "")
+	if err != nil {
+		log.LogError("TestInsertAndSelectSingle setup failed: %s", err)
+		t.Fail()
+		return
+	}
+
+	// WHEN
+	_, err = insertExec.Execute(conn)
+	if err != nil {
+		log.LogError("TestInsertAndSelectSingle insert failed: %s", err)
+		t.Fail()
+		return
+	}
+
+	res, err := selectExec.Execute(conn)
+
+	// THEN
+	if err != nil {
+		log.LogError("TestInsertAndSelectSingle failure. %s", err)
+		t.Fail()
+		return
+	}
+	if len(res) != 1 {
+		log.LogError("TestInsertAndSelectSingle test failed. Expected: 1, Actual: %s", len(res))
+		t.Fail()
+		return
+	}
+
+	singleRes := res[0]
+	if singleRes.ID != 1 || singleRes.Int64Field != 54 || singleRes.IntField != 12 || singleRes.StringField != "asdf" {
+		log.LogError("TestInsertAndSelectSingle invalid result. Expected: %s, Actual: %s", testObj, singleRes)
+		t.Fail()
+	}
+
+}
+
+func TestInsertSelectWithCompositePk(t *testing.T) {
+	// GIVEN
+	conn := simpleorm.OpenConnection("sqlite3", "file:test.db")
+	defer cleanUp("test.db", conn)
+	orm := simpleorm.NewORM(Test{}, TestWithFk{}, TestWithFkAndFkId{},
+		TestWithCompositePk{}, TestWithCompositeFk{})
+
+	exec.ExecuteDDL(conn, orm)
+
+	testObj := TestWithCompositePk{ID: 12, Name: "Test"}
+
+	insertExec, err := exec.NewInsert(orm, testObj)
+	if err != nil {
+		log.LogError("TestInsertSelectWithCompositePk setup failed: %s", err)
+		t.Fail()
+		return
+	}
+
+	selectExec, err := exec.CreateSelect[TestWithCompositePk](orm, "")
+	if err != nil {
+		log.LogError("TestInsertSelectWithCompositePk setup failed: %s", err)
+		t.Fail()
+		return
+	}
+
+	// WHEN
+	_, err = insertExec.Execute(conn)
+	if err != nil {
+		log.LogError("TestInsertSelectWithCompositePk insert failed: %s", err)
+		t.Fail()
+		return
+	}
+
+	res, err := selectExec.Execute(conn)
+
+	// THEN
+	if err != nil {
+		log.LogError("TestInsertSelectWithCompositePk failure. %s", err)
+		t.Fail()
+		return
+	}
+	if len(res) != 1 {
+		log.LogError("TestInsertSelectWithCompositePk test failed. Expected: 1, Actual: %s", len(res))
+		t.Fail()
+		return
+	}
+
+	singleRes := res[0]
+	if singleRes.ID != 12 || singleRes.Name != "Test" {
+		log.LogError("TestInsertSelectWithCompositePk invalid result. Expected: %s, Actual: %s", testObj, singleRes)
+		t.Fail()
+	}
+
+}
+
+func TestInsertMultipleAndSelectWithParam(t *testing.T) {
+	// GIVEN
+	conn := simpleorm.OpenConnection("sqlite3", "file:test.db")
+	defer cleanUp("test.db", conn)
+
+	orm := simpleorm.NewORM(Test{}, TestWithFk{}, TestWithFkAndFkId{},
+		TestWithCompositePk{}, TestWithCompositeFk{})
+
+	exec.ExecuteDDL(conn, orm)
+
+	testObj := Test{Int64Field: 54, IntField: 12, StringField: "asdf"}
+	testObjSecond := Test{Int64Field: 12, IntField: 54, StringField: "fdsa"}
+
+	insertExec, err := exec.NewInsert(orm, testObj)
+	if err != nil {
+		log.LogError("TestInsertMultipleAndSelectWithParam setup failed: %s", err)
+		t.Fail()
+		return
+	}
+
+	insertSecond, err := exec.NewInsert(orm, testObjSecond)
+	if err != nil {
+		log.LogError("TestInsertMultipleAndSelectWithParam setup failed: %s", err)
+		t.Fail()
+		return
+	}
+
+	selectExec, err := exec.CreateSelect[Test](orm, "string_field = ?", "fdsa")
+	if err != nil {
+		log.LogError("TestInsertMultipleAndSelectWithParam setup failed: %s", err)
+		t.Fail()
+		return
+	}
+
+	// WHEN
+
+	_, err = insertExec.Execute(conn)
+	if err != nil {
+		log.LogError("TestInsertMultipleAndSelectWithParam first insert failed: %s", err)
+		t.Fail()
+		return
+	}
+
+	_, err = insertSecond.Execute(conn)
+	if err != nil {
+		log.LogError("TestInsertMultipleAndSelectWithParam second insert failed: %s", err)
+		t.Fail()
+		return
+	}
+
+	res, err := selectExec.Execute(conn)
+
+	// THEN
+	if err != nil {
+		log.LogError("TestInsertMultipleAndSelectWithParam failure. %s", err)
+		t.Fail()
+		return
+	}
+	if len(res) != 1 {
+		log.LogError(" TestInsertMultipleAndSelectWithParam test failed. Expected: 1, Actual: %s", len(res))
+		t.Fail()
+		return
+	}
+
+	singleRes := res[0]
+	if singleRes.ID != 2 || singleRes.Int64Field != 12 || singleRes.IntField != 54 || singleRes.StringField != "fdsa" {
+		log.LogError("TestInsertMultipleAndSelectWithParam invalid result. Expected: %s, Actual: %s", testObjSecond, singleRes)
+		t.Fail()
+		return
+	}
+}
+
+func TestInsertAndUpdate(t *testing.T) {
+	// GIVEN
+	conn := simpleorm.OpenConnection("sqlite3", "file:test.db")
+	defer cleanUp("test.db", conn)
+	orm := simpleorm.NewORM(Test{}, TestWithFk{}, TestWithFkAndFkId{},
+		TestWithCompositePk{}, TestWithCompositeFk{})
+
+	exec.ExecuteDDL(conn, orm)
+
+	testObj := Test{Int64Field: 54, IntField: 12, StringField: "asdf"}
+
+	insertExec, err := exec.NewInsert(orm, testObj)
+	if err != nil {
+		log.LogError("TestInsertAndUpdate setup failed: %s", err)
+		t.Fail()
+		return
+	}
+
+	selectExec, err := exec.CreateSelect[Test](orm, "string_field = ?", "asdf")
+	if err != nil {
+		log.LogError("TestInsertMultipleAndSelectWithParam setup failed: %s", err)
+		t.Fail()
+		return
+	}
+
+	// WHEN
+
+	_, err = insertExec.Execute(conn)
+	if err != nil {
+		log.LogError("TestInsertAndUpdate insert failed: %s", err)
+		t.Fail()
+		return
+	}
+
+	res, err := selectExec.Execute(conn)
+	if err != nil {
+		log.LogError("TestInsertAndUpdate re-select failed: %s", err)
+		t.Fail()
+		return
+	}
+
+	testObj = res[0]
+
+	testObj.IntField = 42069
+	testObj.Int64Field = 69420
+
+	updateExec, err := exec.NewUpdate(orm, testObj)
+	if err != nil {
+		log.LogError("TestInsertAndUpdate setup failed: %s", err)
+		t.Fail()
+		return
+	}
+
+	_, err = updateExec.Execute(conn)
+	res, selectErr := selectExec.Execute(conn)
+
+	// THEN
+	if err != nil {
+		log.LogError("TestInsertAndUpdate update failure. %s", err)
+		t.Fail()
+		return
+	}
+
+	if selectErr != nil {
+		log.LogError("TestInsertAndUpdate select failure. %s", err)
+		t.Fail()
+		return
+	}
+
+	if len(res) != 1 {
+		log.LogError("TestInsertAndUpdate test failed. Expected: 1, Actual: %s", len(res))
+		t.Fail()
+		return
+	}
+
+	singleRes := res[0]
+	if singleRes.ID != 1 || singleRes.Int64Field != 69420 || singleRes.IntField != 42069 || singleRes.StringField != "asdf" {
+		log.LogError("TestInsertAndUpdate invalid result. Expected: %s, Actual: %s", testObj, singleRes)
+		t.Fail()
+	}
+
+}
+
+func TestInsertAndDelete(t *testing.T) {
+	// GIVEN
+	conn := simpleorm.OpenConnection("sqlite3", "file:test.db")
+	defer cleanUp("test.db", conn)
+	orm := simpleorm.NewORM(Test{}, TestWithFk{}, TestWithFkAndFkId{},
+		TestWithCompositePk{}, TestWithCompositeFk{})
+
+	exec.ExecuteDDL(conn, orm)
+
+	testObj := Test{Int64Field: 54, IntField: 12, StringField: "asdf"}
+
+	insertExec, err := exec.NewInsert(orm, testObj)
+	if err != nil {
+		log.LogError("TestInsertAndDelete setup failed: %s", err)
+		t.Fail()
+		return
+	}
+
+	_, err = insertExec.Execute(conn)
+	if err != nil {
+		log.LogError("TestInsertAndDelete setup failed: %s", err)
+		t.Fail()
+		return
+	}
+
+	selectExec, err := exec.CreateSelect[Test](orm, "")
+	if err != nil {
+		log.LogError("TestInsertAndDelete setup failed: %s", err)
+		t.Fail()
+		return
+	}
+
+	selectRes, err := selectExec.Execute(conn)
+
+	if err != nil {
+		log.LogError("TestInsertAndDelete setup failed. %s", err)
+		t.Fail()
+		return
+	}
+
+	deleteExec, err := exec.NewDelete(orm, selectRes)
+
+	// WHEN
+	_, err = deleteExec.Execute(conn)
+
+	// THEN
+	if err != nil {
+		log.LogError("TestInsertAndDelete delete failure. %s", err)
+		t.Fail()
+		return
+	}
+
+	selectRes, err = selectExec.Execute(conn)
+	if len(selectRes) != 0 {
+		log.LogError("TestInsertAndDelete test failed. Expected: 0, Actual: %s", len(selectRes))
+		t.Fail()
+		return
+	}
+
+}
+
+func TestBoolInsertAndSelectSingle(t *testing.T) {
+	// GIVEN
+	conn := simpleorm.OpenConnection("sqlite3", "file:test.db")
+	defer cleanUp("test.db", conn)
+	orm := simpleorm.NewORM(TestWithBool{})
+
+	exec.ExecuteDDL(conn, orm)
+
+	testObj := TestWithBool{BoolField: true}
+
+	insertExec, err := exec.NewInsert(orm, testObj)
+	if err != nil {
+		log.LogError("TestBoolInsertAndSelectSingle setup failed: %s", err)
+		t.Fail()
+		return
+	}
+
+	selectExec, err := exec.CreateSelect[TestWithBool](orm, "")
+	if err != nil {
+		log.LogError("TestBoolInsertAndSelectSingle setup failed: %s", err)
+		t.Fail()
+		return
+	}
+
+	// WHEN
+	_, err = insertExec.Execute(conn)
+	if err != nil {
+		log.LogError("TestBoolInsertAndSelectSingle insert failed: %s", err)
+		t.Fail()
+		return
+	}
+
+	res, err := selectExec.Execute(conn)
+
+	// THEN
+	if err != nil {
+		log.LogError("TestBoolInsertAndSelectSingle failure. %s", err)
+		t.Fail()
+		return
+	}
+	if len(res) != 1 {
+		log.LogError("TestBoolInsertAndSelectSingle test failed. Expected: 1, Actual: %s", len(res))
+		t.Fail()
+		return
+	}
+
+	singleRes := res[0]
+	if singleRes.ID != 1 || !singleRes.BoolField {
+		log.LogError("TestBoolInsertAndSelectSingle invalid result. Expected: %s, Actual: %s", testObj, singleRes)
+		t.Fail()
+	}
+
+}
+
+func TestTimeInsertAndSelectSingle(t *testing.T) {
+	// GIVEN
+	conn := simpleorm.OpenConnection("sqlite3", "file:test.db")
+	defer cleanUp("test.db", conn)
+	orm := simpleorm.NewORM(TestWithTime{})
+
+	exec.ExecuteDDL(conn, orm)
+
+	now := time.Now().UnixMilli()
+	testObj := TestWithTime{TimeField: now}
+
+	insertExec, err := exec.NewInsert(orm, testObj)
+	if err != nil {
+		log.LogError("TestTimeInsertAndSelectSingle setup failed: %s", err)
+		t.Fail()
+		return
+	}
+
+	selectExec, err := exec.CreateSelect[TestWithTime](orm, "")
+	if err != nil {
+		log.LogError("TestTimeInsertAndSelectSingle setup failed: %s", err)
+		t.Fail()
+		return
+	}
+
+	// WHEN
+	_, err = insertExec.Execute(conn)
+	if err != nil {
+		log.LogError("TestTimeInsertAndSelectSingle insert failed: %s", err)
+		t.Fail()
+		return
+	}
+
+	res, err := selectExec.Execute(conn)
+
+	// THEN
+	if err != nil {
+		log.LogError("TestTimeInsertAndSelectSingle failure. %s", err)
+		t.Fail()
+		return
+	}
+	if len(res) != 1 {
+		log.LogError("TestTimeInsertAndSelectSingle test failed. Expected: 1, Actual: %s", len(res))
+		t.Fail()
+		return
+	}
+
+	singleRes := res[0]
+	if singleRes.ID != 1 || singleRes.TimeField != now {
+		log.LogError("TestTimeInsertAndSelectSingle invalid result. Expected: %s, Actual: %s", testObj, singleRes)
+		t.Fail()
+	}
+
+}
@@ -0,0 +1,63 @@
+package simpleorm_test
+
+import (
+	"os"
+	"testing"
+
+	_ "github.com/mattn/go-sqlite3"
+	log "gitlab.com/gdulai/simpleloglvl"
+)
+
+type Test struct {
+	ID          int    `sql:"pk"`
+	Int64Field  int64  `sql:"nn"`
+	IntField    int    `sql:"nn"`
+	StringField string `sql:"nn"`
+}
+
+func (t Test) IsInsertable() bool {
+	return t.ID == 0
+}
+
+type TestWithFk struct {
+	ID     int `sql:"pk"`
+	TestID int `sql:"nn;fk=Test.ID"`
+}
+
+type TestWithFkAndFkId struct {
+	ID     int `sql:"pk"`
+	TestID int `sql:"nn;fk=Test.ID;fk_id=custom_fk"`
+}
+
+type TestWithCompositePk struct {
+	ID   int    `sql:"nn;pk;"`
+	Name string `sql:"nn;pk"`
+}
+
+func (t TestWithCompositePk) IsInsertable() bool {
+	return t.ID == 0 && t.Name == ""
+}
+
+type TestWithCompositeFk struct {
+	ID              int    `sql:"nn;pk"`
+	CompositeFkId   int    `sql:"nn;fk=TestWithCompositePk.ID;"`
+	CompositeFkName string `sql:"nn;fk=TestWithCompositePk.Name;"`
+}
+
+type TestWithBool struct {
+	ID        int  `sql:"pk"`
+	BoolField bool `sql:"nn"`
+}
+
+type TestWithTime struct {
+	ID        int   `sql:"pk"`
+	TimeField int64 `sql:"nn"`
+}
+
+func TestMain(m *testing.M) {
+	log.SetupLogs("Info")
+
+	code := m.Run()
+
+	os.Exit(code)
+}
@@ -0,0 +1,148 @@
+package simpleorm_test
+
+import (
+	"testing"
+
+	simpleorm "git.gdulai.com/gdulai/simpleorm"
+	"git.gdulai.com/gdulai/simpleorm/exec"
+	"git.gdulai.com/gdulai/simpleorm/repository"
+	log "gitlab.com/gdulai/simpleloglvl"
+)
+
+func repoTestSetup() (*simpleorm.ORM, *simpleorm.DBConnection) {
+	conn := simpleorm.OpenConnection("sqlite3", "file:test.db")
+
+	orm := simpleorm.NewORM(Test{}, TestWithFk{}, TestWithFkAndFkId{},
+		TestWithCompositePk{}, TestWithCompositeFk{})
+
+	exec.ExecuteDDL(conn, orm)
+
+	return orm, conn
+}
+
+func TestRepoSelectAll(t *testing.T) {
+	// GIVEN
+	orm, conn := repoTestSetup()
+	defer cleanUp("test.db", conn)
+
+	testObj := Test{Int64Field: 54, IntField: 12, StringField: "asdf"}
+	insertExec, err := exec.NewInsert(orm, testObj)
+	if err != nil {
+		log.LogError("TestRepoSelectAll setup failed: %s", err)
+		t.Fail()
+		return
+	}
+	_, err = insertExec.Execute(conn)
+	repo := repository.NewRepository[Test](conn, orm)
+
+	// WHEN
+	res := repo.SelectAll()
+
+	// THEN
+	if len(res) != 1 {
+		log.LogError("TestRepoSelectAll test failed. Expected: 1, Actual: %s", len(res))
+		t.Fail()
+		return
+	}
+}
+
+func TestRepoSelectByPk(t *testing.T) {
+	// GIVEN
+	orm, conn := repoTestSetup()
+	defer cleanUp("test.db", conn)
+
+	testObj := Test{Int64Field: 54, IntField: 12, StringField: "asdf"}
+	insertExec, err := exec.NewInsert(orm, testObj)
+	if err != nil {
+		log.LogError("TestRepoSelectByPk setup failed: %s", err)
+		t.Fail()
+		return
+	}
+	_, err = insertExec.Execute(conn)
+	repo := repository.NewRepository[Test](conn, orm)
+
+	// WHEN
+	res := repo.SelectByPk(1)
+
+	// THEN
+	if res == nil {
+		log.LogError("TestRepoSelectByPk test failed. Result is nil!")
+		t.Fail()
+		return
+	}
+}
+
+func TestRepoSelectByCompundPk(t *testing.T) {
+	// GIVEN
+	orm, conn := repoTestSetup()
+	defer cleanUp("test.db", conn)
+
+	testObj := TestWithCompositePk{ID: 42069, Name: "Test"}
+	insertExec, err := exec.NewInsert(orm, testObj)
+	if err != nil {
+		log.LogError("TestRepoSelectByCompundPk setup failed: %s", err)
+		t.Fail()
+		return
+	}
+	_, err = insertExec.Execute(conn)
+	repo := repository.NewRepository[TestWithCompositePk](conn, orm)
+
+	// WHEN
+	res := repo.SelectByPk(42069, "Test")
+
+	// THEN
+	if res == nil {
+		log.LogError("TestRepoSelectByCompundPk test failed. Result is nil!")
+		t.Fail()
+		return
+	}
+}
+
+func TestRepoSave(t *testing.T) {
+	// GIVEN
+	orm, conn := repoTestSetup()
+	defer cleanUp("test.db", conn)
+
+	testObj := Test{Int64Field: 54, IntField: 12, StringField: "asdf"}
+	repo := repository.NewRepository[Test](conn, orm)
+
+	// WHEN
+	repo.Save(testObj)
+
+	// THEN
+	res := repo.SelectByPk(1)
+	if res == nil {
+		log.LogError("TestRepoSave test failed. Result is nil!")
+		t.Fail()
+		return
+	}
+}
+
+func TestRepoDelete(t *testing.T) {
+	// GIVEN
+	orm, conn := repoTestSetup()
+	defer cleanUp("test.db", conn)
+
+	testObj := Test{Int64Field: 54, IntField: 12, StringField: "asdf"}
+	repo := repository.NewRepository[Test](conn, orm)
+
+	repo.Save(testObj)
+
+	res := repo.SelectByPk(1)
+	if res == nil {
+		log.LogError("TestRepoDelete setup failed! Test data not saved!")
+		t.Fail()
+		return
+	}
+
+	// WHEN
+	repo.Delete(*res)
+
+	// THEN
+	res = repo.SelectByPk(1)
+	if res != nil {
+		log.LogError("TestRepoDelete test failed. Result is not nil!")
+		t.Fail()
+		return
+	}
+}
@@ -0,0 +1,115 @@
+package simpleorm_test
+
+import (
+	"strconv"
+	"testing"
+
+	simpleorm "git.gdulai.com/gdulai/simpleorm"
+	log "gitlab.com/gdulai/simpleloglvl"
+)
+
+func TestParseSimple(t *testing.T) {
+	log.LogInfo("Test start...")
+	// GIVEN
+	orm := simpleorm.NewORM(Test{})
+	// WHEN
+	ddl, err := orm.CreateSchema()
+	// THEN
+	if err != nil {
+		log.LogError("Failed to parse schema! %s", err)
+		t.Fail()
+	}
+
+	expectedDdl := "CREATE TABLE IF NOT EXISTS TEST (ID INTEGER, INT64_FIELD BIGINT NOT NULL, INT_FIELD INTEGER NOT NULL, STRING_FIELD TEXT NOT NULL, CONSTRAINT pk_test PRIMARY KEY(ID));"
+	if ddl != expectedDdl {
+		log.LogError("Incorrect DDL.\nExpected\n%s\nActual\n%s", expectedDdl, ddl)
+		log.LogError("\nExpected size: %s\nActual size: %s", strconv.Itoa(len(expectedDdl)), strconv.Itoa(len(ddl)))
+		t.Fail()
+	}
+	log.LogInfo("Test finished!")
+}
+
+func TestParseFk(t *testing.T) {
+	log.LogInfo("Test start...")
+	// GIVEN
+	orm := simpleorm.NewORM(Test{}, TestWithFk{})
+	// WHEN
+	ddl, err := orm.CreateSchema()
+	// THEN
+	if err != nil {
+		log.LogError("Failed to parse schema! %s", err)
+		t.Fail()
+	}
+
+	expectedDdl :=
+		"CREATE TABLE IF NOT EXISTS TEST (ID INTEGER, INT64_FIELD BIGINT NOT NULL, INT_FIELD INTEGER NOT NULL, STRING_FIELD TEXT NOT NULL, CONSTRAINT pk_test PRIMARY KEY(ID));\n" +
+			"CREATE TABLE IF NOT EXISTS TEST_WITH_FK (ID INTEGER, TEST_ID INTEGER NOT NULL, CONSTRAINT pk_test_with_fk PRIMARY KEY(ID), CONSTRAINT fk_test FOREIGN KEY(TEST_ID) REFERENCES TEST(ID));"
+	if ddl != expectedDdl {
+		log.LogError("Incorrect DDL.\nExpected\n%s\nActual\n%s", expectedDdl, ddl)
+		log.LogError("\nExpected size: %s\nActual size: %s", strconv.Itoa(len(expectedDdl)), strconv.Itoa(len(ddl)))
+		t.Fail()
+	}
+	log.LogInfo("Test finished!")
+}
+
+func TestParseFkAndFkId(t *testing.T) {
+	// GIVEN
+	orm := simpleorm.NewORM(Test{}, TestWithFkAndFkId{})
+	// WHEN
+	// WHEN
+	ddl, err := orm.CreateSchema()
+	// THEN
+	if err != nil {
+		log.LogError("Failed to parse schema! %s", err)
+		t.Fail()
+	}
+
+	expectedDdl :=
+		"CREATE TABLE IF NOT EXISTS TEST (ID INTEGER, INT64_FIELD BIGINT NOT NULL, INT_FIELD INTEGER NOT NULL, STRING_FIELD TEXT NOT NULL, CONSTRAINT pk_test PRIMARY KEY(ID));\n" +
+			"CREATE TABLE IF NOT EXISTS TEST_WITH_FK_AND_FK_ID (ID INTEGER, TEST_ID INTEGER NOT NULL, CONSTRAINT pk_test_with_fk_and_fk_id PRIMARY KEY(ID), CONSTRAINT custom_fk FOREIGN KEY(TEST_ID) REFERENCES TEST(ID));"
+	if ddl != expectedDdl {
+		log.LogError("Incorrect DDL.\nExpected\n%s\nActual\n%s", expectedDdl, ddl)
+		log.LogError("\nExpected size: %s\nActual size: %s", strconv.Itoa(len(expectedDdl)), strconv.Itoa(len(ddl)))
+		t.Fail()
+	}
+}
+
+func TestParseCompositePk(t *testing.T) {
+	// GIVEN
+	orm := simpleorm.NewORM(TestWithCompositePk{})
+	// WHEN
+	ddl, err := orm.CreateSchema()
+	// THEN
+	if err != nil {
+		log.LogError("Failed to parse schema! %s", err)
+		t.Fail()
+	}
+
+	expectedDdl := "CREATE TABLE IF NOT EXISTS TEST_WITH_COMPOSITE_PK (ID INTEGER NOT NULL, NAME TEXT NOT NULL, CONSTRAINT pk_test_with_composite_pk PRIMARY KEY(ID, NAME));"
+	if ddl != expectedDdl {
+		log.LogError("Incorrect DDL.\nExpected\n%s\nActual\n%s", expectedDdl, ddl)
+		log.LogError("\nExpected size: %s\nActual size: %s", strconv.Itoa(len(expectedDdl)), strconv.Itoa(len(ddl)))
+		t.Fail()
+	}
+}
+
+func TestParseCompositeFk(t *testing.T) {
+	// GIVEN
+	orm := simpleorm.NewORM(TestWithCompositePk{}, TestWithCompositeFk{})
+	// WHEN
+	ddl, err := orm.CreateSchema()
+	// THEN
+	if err != nil {
+		log.LogError("Failed to parse schema! %s", err)
+		t.Fail()
+	}
+
+	expectedDdl :=
+		"CREATE TABLE IF NOT EXISTS TEST_WITH_COMPOSITE_PK (ID INTEGER NOT NULL, NAME TEXT NOT NULL, CONSTRAINT pk_test_with_composite_pk PRIMARY KEY(ID, NAME));\n" +
+			"CREATE TABLE IF NOT EXISTS TEST_WITH_COMPOSITE_FK (ID INTEGER NOT NULL, COMPOSITE_FK_ID INTEGER NOT NULL, COMPOSITE_FK_NAME TEXT NOT NULL, CONSTRAINT pk_test_with_composite_fk PRIMARY KEY(ID), CONSTRAINT fk_test_with_composite_pk FOREIGN KEY(COMPOSITE_FK_ID, COMPOSITE_FK_NAME) REFERENCES TEST_WITH_COMPOSITE_PK(ID, NAME));"
+	if ddl != expectedDdl {
+		log.LogError("Incorrect DDL.\nExpected\n%s\nActual\n%s", expectedDdl, ddl)
+		log.LogError("\nExpected size: %s\nActual size: %s", strconv.Itoa(len(expectedDdl)), strconv.Itoa(len(ddl)))
+		t.Fail()
+	}
+}