mock_store.go

package pirsch

import (
	"database/sql"
	"github.com/jmoiron/sqlx"
	"sync"
	"time"
)

// storeMock is a Store implementation for testing purposes.
// It overwrites all functions required for testing using PostgresStore as a base implementation.
type storeMock struct {
	PostgresStore

	hits []Hit
	m    sync.Mutex
}

func newTestStore() *storeMock {
	return &storeMock{hits: make([]Hit, 0)}
}

func (store *storeMock) NewTx() *sqlx.Tx {
	return nil
}

func (store *storeMock) Commit(tx *sqlx.Tx) {
	if err := tx.Commit(); err != nil {
		panic(err)
	}
}

func (store *storeMock) Rollback(tx *sqlx.Tx) {
	if err := tx.Rollback(); err != nil {
		panic(err)
	}
}

func (store *storeMock) SaveHits(hits []Hit) error {
	store.m.Lock()
	defer store.m.Unlock()
	store.hits = append(store.hits, hits...)
	return nil
}

func (store *storeMock) Session(tenantID sql.NullInt64, fingerprint string, maxAge time.Time) time.Time {
	return time.Now()
}