refactor(mockfs): automatically clean up with t.Cleanup()
This commit is contained in:
@@ -35,6 +35,12 @@ func new(t testing.TB, dirs []string) *MockFS {
|
||||
if err != nil {
|
||||
t.Fatalf("create db: %v", err)
|
||||
}
|
||||
t.Cleanup(func() {
|
||||
if err := dbc.Close(); err != nil {
|
||||
t.Fatalf("close db: %v", err)
|
||||
}
|
||||
})
|
||||
|
||||
if err := dbc.Migrate(db.MigrationContext{}); err != nil {
|
||||
t.Fatalf("migrate db db: %v", err)
|
||||
}
|
||||
@@ -89,12 +95,6 @@ func (m *MockFS) ResetDates() {
|
||||
}
|
||||
}
|
||||
|
||||
func (m *MockFS) CleanUp() {
|
||||
if err := m.db.Close(); err != nil {
|
||||
m.t.Fatalf("close db: %v", err)
|
||||
}
|
||||
}
|
||||
|
||||
func (m *MockFS) AddItems() { m.addItems("", false) }
|
||||
func (m *MockFS) AddItemsPrefix(prefix string) { m.addItems(prefix, false) }
|
||||
func (m *MockFS) AddItemsWithCovers() { m.addItems("", true) }
|
||||
|
||||
Reference in New Issue
Block a user