diff options
Diffstat (limited to 'srv/src/post/store_test.go')
| -rw-r--r-- | srv/src/post/store_test.go | 22 |
1 files changed, 7 insertions, 15 deletions
diff --git a/srv/src/post/store_test.go b/srv/src/post/store_test.go index 9b0ee45..1b4fb30 100644 --- a/srv/src/post/store_test.go +++ b/srv/src/post/store_test.go @@ -1,13 +1,12 @@ package post import ( - "io/ioutil" - "os" "sort" "strconv" "testing" "time" + "github.com/mediocregopher/blog.mediocregopher.com/srv/cfg" "github.com/stretchr/testify/assert" "github.com/tilinna/clock" ) @@ -29,25 +28,18 @@ type storeTestHarness struct { func newStoreTestHarness(t *testing.T) storeTestHarness { - clock := clock.NewMock(time.Now().UTC().Truncate(1 * time.Hour)) + var dataDir cfg.DataDir - tmpFile, err := ioutil.TempFile(os.TempDir(), "mediocre-blog-post-store-test-") - if err != nil { - t.Fatal("Cannot create temporary file", err) + if err := dataDir.Init(); err != nil { + t.Fatal(err) } - tmpFilePath := tmpFile.Name() - tmpFile.Close() - t.Logf("using temporary sqlite file at %q", tmpFilePath) + t.Cleanup(func() { dataDir.Close() }) - t.Cleanup(func() { - if err := os.Remove(tmpFilePath); err != nil { - panic(err) - } - }) + clock := clock.NewMock(time.Now().UTC().Truncate(1 * time.Hour)) store, err := NewStore(StoreParams{ - DBFilePath: tmpFilePath, + DataDir: dataDir, }) assert.NoError(t, err) |