1
1
package mysqldump
2
2
3
3
import (
4
- "testing"
5
4
"github.com/DATA-DOG/go-sqlmock"
6
- "reflect"
7
5
"io/ioutil"
8
6
"os"
7
+ "reflect"
9
8
"strings"
9
+ "testing"
10
10
)
11
11
12
12
func TestGetTablesOk (t * testing.T ) {
@@ -18,12 +18,12 @@ func TestGetTablesOk(t *testing.T) {
18
18
defer db .Close ()
19
19
20
20
rows := sqlmock .NewRows ([]string {"Tables_in_Testdb" }).
21
- AddRow ("Test_Table_1" , ).
21
+ AddRow ("Test_Table_1" ).
22
22
AddRow ("Test_Table_2" )
23
23
24
24
mock .ExpectQuery ("^SHOW TABLES$" ).WillReturnRows (rows )
25
25
26
- result , err := getTables (db );
26
+ result , err := getTables (db )
27
27
if err != nil {
28
28
t .Errorf ("error was not expected while updating stats: %s" , err )
29
29
}
@@ -55,7 +55,7 @@ func TestGetTablesNil(t *testing.T) {
55
55
56
56
mock .ExpectQuery ("^SHOW TABLES$" ).WillReturnRows (rows )
57
57
58
- result , err := getTables (db );
58
+ result , err := getTables (db )
59
59
if err != nil {
60
60
t .Errorf ("error was not expected while updating stats: %s" , err )
61
61
}
@@ -81,11 +81,11 @@ func TestGetServerVersionOk(t *testing.T) {
81
81
defer db .Close ()
82
82
83
83
rows := sqlmock .NewRows ([]string {"Version()" }).
84
- AddRow ("test_version" , )
84
+ AddRow ("test_version" )
85
85
86
86
mock .ExpectQuery ("^SELECT version()" ).WillReturnRows (rows )
87
87
88
- result , err := getServerVersion (db );
88
+ result , err := getServerVersion (db )
89
89
if err != nil {
90
90
t .Errorf ("error was not expected while updating stats: %s" , err )
91
91
}
@@ -111,11 +111,11 @@ func TestCreateTableSQLOk(t *testing.T) {
111
111
defer db .Close ()
112
112
113
113
rows := sqlmock .NewRows ([]string {"Table" , "Create Table" }).
114
- AddRow ("Test_Table" , "CREATE TABLE 'Test_Table' (`id` int(11) NOT NULL AUTO_INCREMENT,`s` char(60) DEFAULT NULL, PRIMARY KEY (`id`))ENGINE=InnoDB DEFAULT CHARSET=latin1" , )
114
+ AddRow ("Test_Table" , "CREATE TABLE 'Test_Table' (`id` int(11) NOT NULL AUTO_INCREMENT,`s` char(60) DEFAULT NULL, PRIMARY KEY (`id`))ENGINE=InnoDB DEFAULT CHARSET=latin1" )
115
115
116
116
mock .ExpectQuery ("^SHOW CREATE TABLE Test_Table$" ).WillReturnRows (rows )
117
117
118
- result , err := createTableSQL (db , "Test_Table" );
118
+ result , err := createTableSQL (db , "Test_Table" )
119
119
120
120
if err != nil {
121
121
t .Errorf ("error was not expected while updating stats: %s" , err )
@@ -133,7 +133,6 @@ func TestCreateTableSQLOk(t *testing.T) {
133
133
}
134
134
}
135
135
136
-
137
136
func TestCreateTableValuesOk (t * testing.T ) {
138
137
db , mock , err := sqlmock .New ()
139
138
if err != nil {
@@ -148,7 +147,7 @@ func TestCreateTableValuesOk(t *testing.T) {
148
147
149
148
mock .ExpectQuery ("^SELECT (.+) FROM test$" ).WillReturnRows (rows )
150
149
151
- result , err := createTableValues (db , "test" );
150
+ result , err := createTableValues (db , "test" )
152
151
if err != nil {
153
152
t .Errorf ("error was not expected while updating stats: %s" , err )
154
153
}
@@ -179,7 +178,7 @@ func TestCreateTableValuesNil(t *testing.T) {
179
178
180
179
mock .ExpectQuery ("^SELECT (.+) FROM test$" ).WillReturnRows (rows )
181
180
182
- result , err := createTableValues (db , "test" );
181
+ result , err := createTableValues (db , "test" )
183
182
if err != nil {
184
183
t .Errorf ("error was not expected while updating stats: %s" , err )
185
184
}
@@ -205,17 +204,16 @@ func TestCreateTableOk(t *testing.T) {
205
204
defer db .Close ()
206
205
207
206
createTableRows := sqlmock .NewRows ([]string {"Table" , "Create Table" }).
208
- AddRow ("Test_Table" , "CREATE TABLE 'Test_Table' (`id` int(11) NOT NULL AUTO_INCREMENT,`s` char(60) DEFAULT NULL, PRIMARY KEY (`id`))ENGINE=InnoDB DEFAULT CHARSET=latin1" , )
207
+ AddRow ("Test_Table" , "CREATE TABLE 'Test_Table' (`id` int(11) NOT NULL AUTO_INCREMENT,`s` char(60) DEFAULT NULL, PRIMARY KEY (`id`))ENGINE=InnoDB DEFAULT CHARSET=latin1" )
209
208
210
209
createTableValueRows := sqlmock .NewRows ([]string {"id" , "email" , "name" }).
211
210
AddRow (1 , nil , "Test Name 1" ).
212
211
AddRow (
2 ,
"[email protected] " ,
"Test Name 2" )
213
212
214
-
215
213
mock .ExpectQuery ("^SHOW CREATE TABLE Test_Table$" ).WillReturnRows (createTableRows )
216
214
mock .ExpectQuery ("^SELECT (.+) FROM Test_Table$" ).WillReturnRows (createTableValueRows )
217
215
218
- result , err := createTable (db , "Test_Table" );
216
+ result , err := createTable (db , "Test_Table" )
219
217
if err != nil {
220
218
t .Errorf ("error was not expected while updating stats: %s" , err )
221
219
}
@@ -226,8 +224,8 @@ func TestCreateTableOk(t *testing.T) {
226
224
}
227
225
228
226
expectedResult := & table {
229
- Name :"Test_Table" ,
230
- SQL : "CREATE TABLE 'Test_Table' (`id` int(11) NOT NULL AUTO_INCREMENT,`s` char(60) DEFAULT NULL, PRIMARY KEY (`id`))ENGINE=InnoDB DEFAULT CHARSET=latin1" ,
227
+ Name : "Test_Table" ,
228
+ SQL : "CREATE TABLE 'Test_Table' (`id` int(11) NOT NULL AUTO_INCREMENT,`s` char(60) DEFAULT NULL, PRIMARY KEY (`id`))ENGINE=InnoDB DEFAULT CHARSET=latin1" ,
231
229
Values :
"('1','','Test Name 1'),('2','[email protected] ','Test Name 2')" ,
232
230
}
233
231
@@ -249,13 +247,13 @@ func TestDumpOk(t *testing.T) {
249
247
defer db .Close ()
250
248
251
249
showTablesRows := sqlmock .NewRows ([]string {"Tables_in_Testdb" }).
252
- AddRow ("Test_Table" , )
250
+ AddRow ("Test_Table" )
253
251
254
252
serverVersionRows := sqlmock .NewRows ([]string {"Version()" }).
255
- AddRow ("test_version" , )
253
+ AddRow ("test_version" )
256
254
257
255
createTableRows := sqlmock .NewRows ([]string {"Table" , "Create Table" }).
258
- AddRow ("Test_Table" , "CREATE TABLE 'Test_Table' (`id` int(11) NOT NULL AUTO_INCREMENT,`email` char(60) DEFAULT NULL, `name` char(60), PRIMARY KEY (`id`))ENGINE=InnoDB DEFAULT CHARSET=latin1" , )
256
+ AddRow ("Test_Table" , "CREATE TABLE 'Test_Table' (`id` int(11) NOT NULL AUTO_INCREMENT,`email` char(60) DEFAULT NULL, `name` char(60), PRIMARY KEY (`id`))ENGINE=InnoDB DEFAULT CHARSET=latin1" )
259
257
260
258
createTableValueRows := sqlmock .NewRows ([]string {"id" , "email" , "name" }).
261
259
AddRow (1 , nil , "Test Name 1" ).
@@ -266,11 +264,10 @@ func TestDumpOk(t *testing.T) {
266
264
mock .ExpectQuery ("^SHOW CREATE TABLE Test_Table$" ).WillReturnRows (createTableRows )
267
265
mock .ExpectQuery ("^SELECT (.+) FROM Test_Table$" ).WillReturnRows (createTableValueRows )
268
266
269
-
270
267
dumper := & Dumper {
271
- db :db ,
268
+ db : db ,
272
269
format : "test_format" ,
273
- dir : "/tmp/" ,
270
+ dir : "/tmp/" ,
274
271
}
275
272
276
273
path , err := dumper .Dump ()
@@ -291,8 +288,7 @@ func TestDumpOk(t *testing.T) {
291
288
292
289
result := strings .Replace (strings .Split (string (f ), "-- Dump completed" )[0 ], "`" , "\\ " , - 1 )
293
290
294
-
295
- expected := `-- Go SQL Dump 0.1.1
291
+ expected := `-- Go SQL Dump ` + version + `
296
292
--
297
293
-- ------------------------------------------------------
298
294
-- Server version test_version
@@ -320,4 +316,3 @@ UNLOCK TABLES;
320
316
t .Fatalf ("expected %#v, got %#v" , expected , result )
321
317
}
322
318
}
323
-
0 commit comments