From 560c7accbc92c81d9e1ef59a5c3ca788f4a21bad Mon Sep 17 00:00:00 2001 From: Mark Bates Date: Fri, 8 Dec 2017 21:26:21 -0500 Subject: [PATCH] added a DBDelta function --- model.go | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/model.go b/model.go index 51b9703..eb8f5b0 100644 --- a/model.go +++ b/model.go @@ -23,6 +23,15 @@ func (m *Model) SetupTest() { func (m *Model) TearDownTest() {} +func (m *Model) DBDelta(delta int, name string, fn func()) { + sc, err := m.DB.Count(name) + m.NoError(err) + fn() + ec, err := m.DB.Count(name) + m.NoError(err) + m.Equal(ec, sc+delta) +} + func NewModel() *Model { m := &Model{} c, err := pop.Connect(envy.Get("GO_ENV", "test"))