added a DBDelta function

gobuffalo · Dec 9, 2017 · 560c7ac · 560c7ac
1 parent 985d7ac
commit 560c7ac
Showing 1 changed file with 9 additions and 0 deletions.
diff --git 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"))