connector_test.go 1.3 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667
  1. // +build go1.10
  2. package pq
  3. import (
  4. "context"
  5. "database/sql"
  6. "database/sql/driver"
  7. "testing"
  8. )
  9. func TestNewConnector_WorksWithOpenDB(t *testing.T) {
  10. name := ""
  11. c, err := NewConnector(name)
  12. if err != nil {
  13. t.Fatal(err)
  14. }
  15. db := sql.OpenDB(c)
  16. defer db.Close()
  17. // database/sql might not call our Open at all unless we do something with
  18. // the connection
  19. txn, err := db.Begin()
  20. if err != nil {
  21. t.Fatal(err)
  22. }
  23. txn.Rollback()
  24. }
  25. func TestNewConnector_Connect(t *testing.T) {
  26. name := ""
  27. c, err := NewConnector(name)
  28. if err != nil {
  29. t.Fatal(err)
  30. }
  31. db, err := c.Connect(context.Background())
  32. if err != nil {
  33. t.Fatal(err)
  34. }
  35. defer db.Close()
  36. // database/sql might not call our Open at all unless we do something with
  37. // the connection
  38. txn, err := db.(driver.ConnBeginTx).BeginTx(context.Background(), driver.TxOptions{})
  39. if err != nil {
  40. t.Fatal(err)
  41. }
  42. txn.Rollback()
  43. }
  44. func TestNewConnector_Driver(t *testing.T) {
  45. name := ""
  46. c, err := NewConnector(name)
  47. if err != nil {
  48. t.Fatal(err)
  49. }
  50. db, err := c.Driver().Open(name)
  51. if err != nil {
  52. t.Fatal(err)
  53. }
  54. defer db.Close()
  55. // database/sql might not call our Open at all unless we do something with
  56. // the connection
  57. txn, err := db.(driver.ConnBeginTx).BeginTx(context.Background(), driver.TxOptions{})
  58. if err != nil {
  59. t.Fatal(err)
  60. }
  61. txn.Rollback()
  62. }