103 lines
2.2 KiB
Go
103 lines
2.2 KiB
Go
|
package mailinglist
|
||
|
|
|
||
|
|
import (
|
||
|
|
"io"
|
||
|
|
"io/ioutil"
|
||
|
|
"os"
|
||
|
|
"testing"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"github.com/stretchr/testify/assert"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestStore(t *testing.T) {
|
||
|
|
tmpFile, err := ioutil.TempFile(os.TempDir(), "mediocre-blog-mailinglist-store-test-")
|
||
|
|
if err != nil {
|
||
|
|
t.Fatal("Cannot create temporary file", err)
|
||
|
|
}
|
||
|
|
tmpFilePath := tmpFile.Name()
|
||
|
|
tmpFile.Close()
|
||
|
|
|
||
|
|
t.Logf("using temporary sqlite file at %q", tmpFilePath)
|
||
|
|
|
||
|
|
t.Cleanup(func() {
|
||
|
|
if err := os.Remove(tmpFilePath); err != nil {
|
||
|
|
panic(err)
|
||
|
|
}
|
||
|
|
})
|
||
|
|
|
||
|
|
store, err := NewStore(tmpFilePath)
|
||
|
|
assert.NoError(t, err)
|
||
|
|
|
||
|
|
t.Cleanup(func() {
|
||
|
|
assert.NoError(t, store.Close())
|
||
|
|
})
|
||
|
|
|
||
|
|
now := func() time.Time {
|
||
|
|
return time.Now().Truncate(time.Second)
|
||
|
|
}
|
||
|
|
|
||
|
|
assertGet := func(t *testing.T, email Email) {
|
||
|
|
t.Helper()
|
||
|
|
|
||
|
|
gotEmail, err := store.Get(email.Email)
|
||
|
|
assert.NoError(t, err)
|
||
|
|
assert.Equal(t, email, gotEmail)
|
||
|
|
|
||
|
|
gotEmail, err = store.GetBySubToken(email.SubToken)
|
||
|
|
assert.NoError(t, err)
|
||
|
|
assert.Equal(t, email, gotEmail)
|
||
|
|
|
||
|
|
if email.UnsubToken != "" {
|
||
|
|
gotEmail, err = store.GetByUnsubToken(email.UnsubToken)
|
||
|
|
assert.NoError(t, err)
|
||
|
|
assert.Equal(t, email, gotEmail)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
assertNotFound := func(t *testing.T, email string) {
|
||
|
|
t.Helper()
|
||
|
|
_, err := store.Get(email)
|
||
|
|
assert.ErrorIs(t, err, ErrNotFound)
|
||
|
|
}
|
||
|
|
|
||
|
|
// now start actual tests
|
||
|
|
|
||
|
|
// GetAll should not do anything, there's no data
|
||
|
|
_, err = store.GetAll()()
|
||
|
|
assert.ErrorIs(t, err, io.EOF)
|
||
|
|
|
||
|
|
emailFoo := Email{
|
||
|
|
Email: "foo",
|
||
|
|
SubToken: "subTokenFoo",
|
||
|
|
CreatedAt: now(),
|
||
|
|
}
|
||
|
|
|
||
|
|
// email isn't stored yet, shouldn't exist
|
||
|
|
assertNotFound(t, emailFoo.Email)
|
||
|
|
|
||
|
|
// Set an email, now it should exist
|
||
|
|
assert.NoError(t, store.Set(emailFoo))
|
||
|
|
assertGet(t, emailFoo)
|
||
|
|
|
||
|
|
// Update the email with an unsub token
|
||
|
|
emailFoo.UnsubToken = "unsubTokenFoo"
|
||
|
|
emailFoo.VerifiedAt = now()
|
||
|
|
assert.NoError(t, store.Set(emailFoo))
|
||
|
|
assertGet(t, emailFoo)
|
||
|
|
|
||
|
|
// GetAll should now only return that email
|
||
|
|
iter := store.GetAll()
|
||
|
|
gotEmail, err := iter()
|
||
|
|
assert.NoError(t, err)
|
||
|
|
assert.Equal(t, emailFoo, gotEmail)
|
||
|
|
_, err = iter()
|
||
|
|
assert.ErrorIs(t, err, io.EOF)
|
||
|
|
|
||
|
|
// Delete the email, it should be gone
|
||
|
|
assert.NoError(t, store.Delete(emailFoo.Email))
|
||
|
|
assertNotFound(t, emailFoo.Email)
|
||
|
|
_, err = store.GetAll()()
|
||
|
|
assert.ErrorIs(t, err, io.EOF)
|
||
|
|
}
|