opt.ReadTimeout = time.Millisecond
db = pg.Connect(opt)
})
AfterEach(func() {
Expect(db.Close()).NotTo(HaveOccurred())
})
It("slow query timeouts", func() {
_, err := db.Exec(`SELECT pg_sleep(1)`)
Expect(err.(net.Error).Timeout()).To(BeTrue())
})
Describe("WithTimeout", func() {
It("slow query passes", func() {
_, err := db.WithTimeout(time.Minute).Exec(`SELECT pg_sleep(1)`)
Expect(err).NotTo(HaveOccurred())
})
})
})
var _ = Describe("Listener.ReceiveTimeout", func() {
var db *pg.DB
var ln *pg.Listener
BeforeEach(func() {
opt := pgOptions()
opt.PoolSize = 1
db = pg.Connect(opt)
var err error
