sponge/pkg/sql2code/sql2code_test.go

package sql2code

import (
	"github.com/stretchr/testify/assert"
	"testing"
)

var sqlData = `
create table user
(
    id         bigint unsigned auto_increment
        primary key,
    created_at datetime        null,
    updated_at datetime        null,
    deleted_at datetime        null,
    name       char(50)        not null comment '用户名',
    password   char(100)       not null comment '密码',
    email      char(50)        not null comment '邮件',
    phone      bigint unsigned not null comment '手机号码',
    age        tinyint         not null comment '年龄',
    gender     tinyint         not null comment '性别，1:男，2:女，3:未知',
    constraint user_email_uindex
        unique (email)
);
`

func TestGenerateOne(t *testing.T) {
	type args struct {
		args *Args
	}
	tests := []struct {
		name    string
		args    args
		wantErr bool
	}{
		{
			name: "sql form param",
			args: args{args: &Args{
				SQL: sqlData,
			}},
			wantErr: false,
		},
		{
			name: "sql from file",
			args: args{args: &Args{
				DDLFile: "test.sql",
			}},
			wantErr: false,
		},
		//{
		//	name: "sql from db",
		//	args: args{args: &Args{
		//		DBDsn:   "root:123456@(192.168.3.37:3306)/test",
		//		DBTable: "user",
		//	}},
		//	wantErr: false,
		//},
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			got, err := GenerateOne(tt.args.args)
			if (err != nil) != tt.wantErr {
				t.Errorf("GenerateOne() error = %v, wantErr %v", err, tt.wantErr)
				return
			}
			t.Log(got)
		})
	}
}

func TestGenerate(t *testing.T) {
	type args struct {
		args *Args
	}
	tests := []struct {
		name    string
		args    args
		wantErr bool
	}{
		{
			name: "sql form param",
			args: args{args: &Args{
				SQL: sqlData,
			}},
			wantErr: false,
		},
		//{
		//	name: "sql from db",
		//	args: args{args: &Args{
		//		DBDsn:   "root:123456@(127.0.0.1:3306)/test",
		//		DBTable: "user",
		//	}},
		//	wantErr: false,
		//},
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			got, err := Generate(tt.args.args)
			if (err != nil) != tt.wantErr {
				t.Errorf("Generate() error = %v, wantErr %v", err, tt.wantErr)
				return
			}
			t.Log(got)
		})
	}
}

func TestGenerateError(t *testing.T) {
	a := &Args{}
	_, err := Generate(a)
	assert.Error(t, err)
	_, err = GenerateOne(a)
	assert.Error(t, err)

	a = &Args{DDLFile: "notfound.sql"}
	_, err = Generate(a)
	assert.Error(t, err)

	a = &Args{DBDsn: "root:123456@(127.0.0.1:3306)/test"}
	_, err = Generate(a)
	assert.Error(t, err)

	a = &Args{DBDsn: "root:123456@(127.0.0.1:3306)/test", DBTable: "user"}
	_, err = Generate(a)
	assert.Error(t, err)

	a = &Args{DDLFile: "test.sql", CodeType: "unknown"}
	_, err = GenerateOne(a)
	t.Log(err)
	assert.Error(t, err)
}

func Test_getOptions(t *testing.T) {
	a := &Args{
		Package:        "Package",
		GormType:       true,
		JSONTag:        true,
		ForceTableName: true,
		Charset:        "Charset",
		Collation:      "Collation",
		TablePrefix:    "TablePrefix",
		ColumnPrefix:   "ColumnPrefix",
		NoNullType:     true,
		NullStyle:      "sql",
	}

	o := getOptions(a)
	assert.NotNil(t, o)
	a.NullStyle = "ptr"
	assert.NotNil(t, o)
	a.NullStyle = "default"
	assert.NotNil(t, o)
}