feat: implement sponge commands

pkg/sql2code/sql2code_test.go

@@ -0,0 +1,153 @@
+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)
+}