57 lines
1.3 KiB
Go
57 lines
1.3 KiB
Go
package pgstore
|
|
|
|
import (
|
|
"net/http"
|
|
"os"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
func TestCleanup(t *testing.T) {
|
|
dsn := os.Getenv("PGSTORE_TEST_CONN")
|
|
if dsn == "" {
|
|
t.Skip("This test requires a real database.")
|
|
}
|
|
|
|
ss, err := NewPGStore(dsn, []byte(secret))
|
|
if err != nil {
|
|
t.Fatal("Failed to get store", err)
|
|
}
|
|
|
|
defer ss.Close()
|
|
// Start the cleanup goroutine.
|
|
defer ss.StopCleanup(ss.Cleanup(time.Millisecond * 500))
|
|
|
|
req, err := http.NewRequest("GET", "http://www.example.com", nil)
|
|
if err != nil {
|
|
t.Fatal("Failed to create request", err)
|
|
}
|
|
|
|
session, err := ss.Get(req, "newsess")
|
|
if err != nil {
|
|
t.Fatal("Failed to create session", err)
|
|
}
|
|
|
|
// Expire the session.
|
|
session.Options.MaxAge = 1
|
|
|
|
m := make(http.Header)
|
|
if err = ss.Save(req, headerOnlyResponseWriter(m), session); err != nil {
|
|
t.Fatal("failed to save session:", err.Error())
|
|
}
|
|
|
|
// Give the ticker a moment to run.
|
|
time.Sleep(time.Millisecond * 1500)
|
|
|
|
// SELECT expired sessions. We should get a count of zero back.
|
|
var count int
|
|
err = ss.DbPool.QueryRow("SELECT count(*) FROM http_sessions WHERE expires_on < now()").Scan(&count)
|
|
if err != nil {
|
|
t.Fatalf("failed to select expired sessions from DB: %v", err)
|
|
}
|
|
|
|
if count > 0 {
|
|
t.Fatalf("ticker did not delete expired sessions: want 0 got %v", count)
|
|
}
|
|
}
|