@@ -62,11 +62,11 @@ func Test_getTableName(t *testing.T) {
6262 }, {
6363 name : "Test 3" ,
6464 model : & testTable3 {},
65- want : "test_table_3 " ,
65+ want : "test_table3 " ,
6666 }, {
6767 name : "Test 4" ,
6868 model : & testTable4 {},
69- want : "test_table_4 " ,
69+ want : "test_table4 " ,
7070 },
7171 }
7272 for _ , tt := range tests {
@@ -93,19 +93,19 @@ func Test_getColumns(t *testing.T) {
9393 {
9494 name : "Test" ,
9595 model : & testTable {},
96- want : []string {"field_1 " , "field_2 " },
96+ want : []string {"field1 " , "field2 " },
9797 }, {
9898 name : "Test 2" ,
9999 model : & testTable2 {},
100- want : []string {"field_1 " , "field_2 " },
100+ want : []string {"field1 " , "field2 " },
101101 }, {
102102 name : "Test 3" ,
103103 model : & testTable3 {},
104- want : []string {"field_2 " },
104+ want : []string {"field2 " },
105105 }, {
106106 name : "Test 4" ,
107107 model : & testTable4 {},
108- want : []string {"field_2 " , "field_3 " },
108+ want : []string {"field2 " , "field3 " },
109109 },
110110 }
111111 for _ , tt := range tests {
@@ -152,7 +152,7 @@ func TestGenerate(t *testing.T) {
152152 & testTable {}, & testTable2 {}, & testTable3 {}, & testTable4 {},
153153 },
154154 },
155- want : `{"version":3,"sources":[{"name":"mysql","kind":"","tables":[{"object_relationships":[],"array_relationships":[],"select_permissions":[{"role":"user","permission":{"columns":["field_1","field_2"],"limit":5,"allow_aggregations":true,"filter":{}}}],"configuration":{"comment":null,"custom_root_fields":null,"custom_column_names":null},"table":{"schema":"public","name":"test_table"}},{"object_relationships":[],"array_relationships":[],"select_permissions":[{"role":"user","permission":{"columns":["field_1","field_2"],"limit":5,"allow_aggregations":true,"filter":{}}}],"configuration":{"comment":null,"custom_root_fields":null,"custom_column_names":null},"table":{"schema":"public","name":"fake_name"}},{"object_relationships":[],"array_relationships":[],"select_permissions":[{"role":"user","permission":{"columns":["field_2"],"limit":5,"allow_aggregations":true,"filter":{}}}],"configuration":{"comment":null,"custom_root_fields":null,"custom_column_names":null},"table":{"schema":"public","name":"test_table_3"}},{"object_relationships":[],"array_relationships":[],"select_permissions":[{"role":"user","permission":{"columns":["field_2","field_3"],"limit":5,"allow_aggregations":true,"filter":{}}}],"configuration":{"comment":null,"custom_root_fields":null,"custom_column_names":null},"table":{"schema":"public","name":"test_table_4"}}],"configuration":{"connection_info":{"use_prepared_statements":false,"isolation_level":"","database_url":""}}}]}` ,
155+ want : `{"version":3,"sources":[{"name":"mysql","kind":"","tables":[{"object_relationships":[],"array_relationships":[],"select_permissions":[{"role":"user","permission":{"columns":["field1","field2"],"limit":5,"allow_aggregations":true,"filter":{}}}],"configuration":{"comment":null,"custom_root_fields":null,"custom_column_names":null},"table":{"schema":"public","name":"test_table"}},{"object_relationships":[],"array_relationships":[],"select_permissions":[{"role":"user","permission":{"columns":["field1","field2"],"limit":5,"allow_aggregations":true,"filter":{}}}],"configuration":{"comment":null,"custom_root_fields":null,"custom_column_names":null},"table":{"schema":"public","name":"fake_name"}},{"object_relationships":[],"array_relationships":[],"select_permissions":[{"role":"user","permission":{"columns":["field2"],"limit":5,"allow_aggregations":true,"filter":{}}}],"configuration":{"comment":null,"custom_root_fields":null,"custom_column_names":null},"table":{"schema":"public","name":"test_table3"}},{"object_relationships":[],"array_relationships":[],"select_permissions":[{"role":"user","permission":{"columns":["field2","field3"],"limit":5,"allow_aggregations":true,"filter":{}}}],"configuration":{"comment":null,"custom_root_fields":null,"custom_column_names":null},"table":{"schema":"public","name":"test_table4"}}],"configuration":{"connection_info":{"use_prepared_statements":false,"isolation_level":"","database_url":""}}}]}` ,
156156 },
157157 }
158158 for _ , tt := range tests {
0 commit comments