| 197 | } |
| 198 | |
| 199 | func (s *SQLTestSuite) TestQueryLogger() { |
| 200 | logLevel := db.LC().Level() |
| 201 | |
| 202 | db.LC().SetLogger(logrus.New()) |
| 203 | db.LC().SetLevel(db.LogLevelDebug) |
| 204 | |
| 205 | defer func() { |
| 206 | db.LC().SetLogger(nil) |
| 207 | db.LC().SetLevel(logLevel) |
| 208 | }() |
| 209 | |
| 210 | sess := s.Session() |
| 211 | |
| 212 | _, err := sess.Collection("artist").Find().Count() |
| 213 | s.Equal(nil, err) |
| 214 | |
| 215 | _, err = sess.Collection("artist_x").Find().Count() |
| 216 | s.NotEqual(nil, err) |
| 217 | } |
| 218 | |
| 219 | func (s *SQLTestSuite) TestExpectCursorError() { |
| 220 | sess := s.Session() |