vendor/github.com/go-sql-driver/mysql/errors_test.go

   1 // Go MySQL Driver - A MySQL-Driver for Go's database/sql package
   2 //
   3 // Copyright 2013 The Go-MySQL-Driver Authors. All rights reserved.
   4 //
   5 // This Source Code Form is subject to the terms of the Mozilla Public
   6 // License, v. 2.0. If a copy of the MPL was not distributed with this file,
   7 // You can obtain one at http://mozilla.org/MPL/2.0/.
   8
   9 package mysql
  10
  11 import (
  12         "bytes"
  13         "log"
  14         "testing"
  15 )
  16
  17 func TestErrorsSetLogger(t *testing.T) {
  18         previous := errLog
  19         defer func() {
  20                 errLog = previous
  21         }()
  22
  23         // set up logger
  24         const expected = "prefix: test\n"
  25         buffer := bytes.NewBuffer(make([]byte, 0, 64))
  26         logger := log.New(buffer, "prefix: ", 0)
  27
  28         // print
  29         SetLogger(logger)
  30         errLog.Print("test")
  31
  32         // check result
  33         if actual := buffer.String(); actual != expected {
  34                 t.Errorf("expected %q, got %q", expected, actual)
  35         }
  36 }
  37
  38 func TestErrorsStrictIgnoreNotes(t *testing.T) {
  39         runTests(t, dsn+"&sql_notes=false", func(dbt *DBTest) {
  40                 dbt.mustExec("DROP TABLE IF EXISTS does_not_exist")
  41         })
  42 }