@@ -256,89 +256,3 @@ func TestQueryAndExit(t *testing.T) {
256
256
assert .Equal (t , "100 val2" + sqlcmd .SqlcmdEol + sqlcmd .SqlcmdEol + oneRowAffected + sqlcmd .SqlcmdEol , string (bytes ), "Incorrect output from run" )
257
257
}
258
258
}
259
-
260
- << << << < HEAD
261
- == == == =
262
- // Test to verify fix for issue: https://github.com/microsoft/go-sqlcmd/issues/98
263
- // 1. Verify when -b is passed in (ExitOnError), we don't always get an error (even when input is good)
264
- // 2, Verify when the input is actually bad, we do get an error
265
- func TestExitOnError (t * testing.T ) {
266
- args = newArguments ()
267
- args .InputFile = []string {"testdata/select100.sql" }
268
- args .ErrorsToStderr = 0
269
- args .ExitOnError = true
270
- if canTestAzureAuth () {
271
- args .UseAad = true
272
- }
273
-
274
- vars := sqlcmd .InitializeVariables (! args .DisableCmdAndWarn )
275
- setVars (vars , & args )
276
-
277
- exitCode , err := run (vars , & args )
278
- assert .NoError (t , err , "run" )
279
- assert .Equal (t , 0 , exitCode , "exitCode" )
280
-
281
- args .InputFile = []string {"testdata/bad.sql" }
282
-
283
- vars = sqlcmd .InitializeVariables (! args .DisableCmdAndWarn )
284
- setVars (vars , & args )
285
-
286
- exitCode , err = run (vars , & args )
287
- assert .NoError (t , err , "run" )
288
- assert .Equal (t , 1 , exitCode , "exitCode" )
289
-
290
- t .Logf ("Test Completed" ) // Needs some output to stdout to count as a test
291
- }
292
-
293
- >> >> >> > main
294
- func TestAzureAuth (t * testing.T ) {
295
-
296
- if ! canTestAzureAuth () {
297
- t .Skip ("Server name is not an Azure DB name" )
298
- }
299
- o , err := os .CreateTemp ("" , "sqlcmdmain" )
300
- assert .NoError (t , err , "os.CreateTemp" )
301
- defer os .Remove (o .Name ())
302
- defer o .Close ()
303
- args = newArguments ()
304
- args .Query = "SELECT 'AZURE'"
305
- args .OutputFile = o .Name ()
306
- args .UseAad = true
307
- vars := sqlcmd .InitializeVariables (! args .DisableCmdAndWarn )
308
- vars .Set (sqlcmd .SQLCMDMAXVARTYPEWIDTH , "0" )
309
- setVars (vars , & args )
310
- exitCode , err := run (vars , & args )
311
- assert .NoError (t , err , "run" )
312
- assert .Equal (t , 0 , exitCode , "exitCode" )
313
- bytes , err := os .ReadFile (o .Name ())
314
- if assert .NoError (t , err , "os.ReadFile" ) {
315
- assert .Equal (t , "AZURE" + sqlcmd .SqlcmdEol + sqlcmd .SqlcmdEol + oneRowAffected + sqlcmd .SqlcmdEol , string (bytes ), "Incorrect output from run" )
316
- }
317
- }
318
-
319
- << << << < HEAD
320
- == == == =
321
- func TestMissingInputFile (t * testing.T ) {
322
- args = newArguments ()
323
- args .InputFile = []string {"testdata/missingFile.sql" }
324
-
325
- if canTestAzureAuth () {
326
- args .UseAad = true
327
- }
328
-
329
- vars := sqlcmd .InitializeVariables (! args .DisableCmdAndWarn )
330
- setVars (vars , & args )
331
-
332
- exitCode , err := run (vars , & args )
333
- assert .Error (t , err , "run" )
334
- assert .Contains (t , err .Error (), "Error occurred while opening or operating on file" , "Unexpected error: " + err .Error ())
335
- assert .Equal (t , 1 , exitCode , "exitCode" )
336
- }
337
-
338
- >> >> >> > main
339
- // Assuming public Azure, use AAD when SQLCMDUSER environment variable is not set
340
- func canTestAzureAuth () bool {
341
- server := os .Getenv (sqlcmd .SQLCMDSERVER )
342
- userName := os .Getenv (sqlcmd .SQLCMDUSER )
343
- return strings .Contains (server , ".database.windows.net" ) && userName == ""
344
- }
0 commit comments