70 lines
2.5 KiB
Go
70 lines
2.5 KiB
Go
// Copyright 2022
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
package orm
|
|
|
|
import (
|
|
"fmt"
|
|
"github.com/stretchr/testify/assert"
|
|
"testing"
|
|
)
|
|
|
|
type ModelWithComments struct {
|
|
Id int `orm:"description(user id)"`
|
|
UserName string `orm:"size(30);unique;description(user name)"`
|
|
Email string `orm:"size(100)"`
|
|
Password string `orm:"size(100)"`
|
|
}
|
|
|
|
func TestGetDbCreateSQLWithComment(t *testing.T) {
|
|
al := getDbAlias("default")
|
|
modelCache.clean()
|
|
RegisterModel(&ModelWithComments{})
|
|
queries, _, _ := modelCache.getDbCreateSQL(al)
|
|
header := `-- --------------------------------------------------
|
|
-- Table Structure for ` + fmt.Sprintf("`%s`", modelCache.allOrdered()[0].fullName) + `
|
|
-- --------------------------------------------------`
|
|
if al.Driver == DRPostgres {
|
|
assert.Equal(t, queries, []string{header + `
|
|
CREATE TABLE IF NOT EXISTS "model_with_comments" (
|
|
"id" serial NOT NULL PRIMARY KEY,
|
|
"user_name" varchar(30) NOT NULL DEFAULT '' UNIQUE,
|
|
"email" varchar(100) NOT NULL DEFAULT '' ,
|
|
"password" varchar(100) NOT NULL DEFAULT ''
|
|
);
|
|
COMMENT ON COLUMN "model_with_comments"."id" is 'user id';
|
|
COMMENT ON COLUMN "model_with_comments"."user_name" is 'user name';`,
|
|
})
|
|
} else if al.Driver == DRMySQL {
|
|
assert.Equal(t, queries, []string{header + `
|
|
CREATE TABLE IF NOT EXISTS ` + "`model_with_comments`" + ` (
|
|
` + "`id`" + ` integer AUTO_INCREMENT NOT NULL PRIMARY KEY COMMENT 'user id',
|
|
` + "`user_name`" + ` varchar(30) NOT NULL DEFAULT '' UNIQUE COMMENT 'user name',
|
|
` + "`email`" + ` varchar(100) NOT NULL DEFAULT '' ,
|
|
` + "`password`" + ` varchar(100) NOT NULL DEFAULT ''
|
|
) ENGINE=INNODB;`,
|
|
})
|
|
} else if al.Driver == DRSqlite {
|
|
assert.Equal(t, queries, []string{header + `
|
|
CREATE TABLE IF NOT EXISTS ` + "`model_with_comments`" + ` (
|
|
` + "`id`" + ` integer NOT NULL PRIMARY KEY AUTOINCREMENT,
|
|
` + "`user_name`" + ` varchar(30) NOT NULL DEFAULT '' UNIQUE,
|
|
` + "`email`" + ` varchar(100) NOT NULL DEFAULT '' ,
|
|
` + "`password`" + ` varchar(100) NOT NULL DEFAULT ''
|
|
);`,
|
|
})
|
|
}
|
|
modelCache.clean()
|
|
}
|