github.com/artisanhe/tools@v1.0.1-0.20210607022958-19a8fef2eb04/gen_method/template_test.go (about)

     1  package gen_method
     2  
     3  import (
     4  	"os"
     5  	"testing"
     6  	"text/template"
     7  
     8  	"github.com/artisanhe/tools/codegen"
     9  )
    10  
    11  var gFile = os.Stdout
    12  
    13  func TestTableNameTemplate(t *testing.T) {
    14  	t.Log(gFile)
    15  	data := TableNameTemplateParam{
    16  		StructName:   "UserTest",
    17  		ReceiverName: "ut",
    18  		PackageName:  "modules",
    19  	}
    20  
    21  	tmpl := template.Must(template.New("create").Parse(TableNameTemplate))
    22  	if err := tmpl.Execute(gFile, data); err != nil {
    23  		t.Error(err)
    24  	}
    25  }
    26  
    27  func TestCreateTemplate(t *testing.T) {
    28  	t.Log(gFile)
    29  	data := TableNameTemplateParam{
    30  		StructName:   "UserTest",
    31  		ReceiverName: "ut",
    32  	}
    33  
    34  	tmpl := template.Must(template.New("create").Parse(CreateTemplate))
    35  	if err := tmpl.Execute(gFile, data); err != nil {
    36  		t.Error(err)
    37  	}
    38  }
    39  
    40  func TestFetchTemplate(t *testing.T) {
    41  	data := new(FetchTemplateParam)
    42  	data.StructName = "UserTest"
    43  	data.ReceiverName = "ut"
    44  	data.Field = "Id"
    45  	data.DbField = "F_id"
    46  
    47  	tmpl := template.Must(template.New("create").Parse(FetchTemplate))
    48  	if err := tmpl.Execute(gFile, data); err != nil {
    49  		t.Error(err)
    50  	}
    51  }
    52  
    53  func TestBatchFetchTemplate(t *testing.T) {
    54  	data := new(BatchFetchTemplateParam)
    55  	data.StructName = "UserTest"
    56  	data.ReceiverName = "ut"
    57  	data.Field = "Id"
    58  	data.DbField = "F_id"
    59  	data.FieldType = "uint64"
    60  
    61  	tmpl := template.Must(template.New("create").Parse(BatchFetchTemplate))
    62  	if err := tmpl.Execute(gFile, data); err != nil {
    63  		t.Error(err)
    64  	}
    65  }
    66  
    67  func TestFetchList(t *testing.T) {
    68  	data := FetchListTemplateParam{}
    69  	data.StructName = "UserTest"
    70  	data.ReceiverListName = "utl"
    71  	data.StructListName = "UserTestList"
    72  	tmpl := template.Must(template.New("create").Parse(FetchListTemplate))
    73  	if err := tmpl.Execute(gFile, data); err != nil {
    74  		t.Error(err)
    75  	}
    76  
    77  }
    78  
    79  func TestSplitStringByWord(t *testing.T) {
    80  	tmpStr := []string{
    81  		"adcIDTest",
    82  		"AbcDref",
    83  		"bbIDArtisan",
    84  		"GenerateURITest",
    85  		"GolangURIIDIPTest",
    86  		"test123",
    87  		"niceGirl",
    88  		"GolangURIIDIPTestAPIIDasebb",
    89  	}
    90  
    91  	t.Log(tmpStr[0][len(tmpStr[0]):])
    92  	for _, str := range tmpStr {
    93  		t.Log(str, codegen.ToUpperSnakeCase(str))
    94  	}
    95  }
    96  
    97  func TestMain(m *testing.M) {
    98  	//file, err := os.Create("./generate_user.go")
    99  	//if err != nil {
   100  	//panic(err)
   101  	//}
   102  	//gFile = file
   103  	os.Exit(m.Run())
   104  }