mirror of
https://github.com/jhillyerd/inbucket.git
synced 2025-12-17 09:37:02 +00:00
d9b5e40c87
- retention: Start from pkg main instead of server/smtp - file: Remove DefaultStore() constructor - storage: AllMailboxes replaced with VisitMailboxes for #69 - test: Stub VisitMailboxes for #80
56 lines
1.5 KiB
Go
56 lines
1.5 KiB
Go
package storage_test
|
|
|
|
import (
|
|
"fmt"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/jhillyerd/inbucket/pkg/config"
|
|
"github.com/jhillyerd/inbucket/pkg/storage"
|
|
"github.com/jhillyerd/inbucket/pkg/test"
|
|
)
|
|
|
|
func TestDoRetentionScan(t *testing.T) {
|
|
ds := test.NewStore()
|
|
// Mockup some different aged messages (num is in hours)
|
|
new1 := mockMessage(0)
|
|
new2 := mockMessage(1)
|
|
new3 := mockMessage(2)
|
|
old1 := mockMessage(4)
|
|
old2 := mockMessage(12)
|
|
old3 := mockMessage(24)
|
|
ds.AddMessage("mb1", new1)
|
|
ds.AddMessage("mb1", old1)
|
|
ds.AddMessage("mb1", old2)
|
|
ds.AddMessage("mb2", old3)
|
|
ds.AddMessage("mb2", new2)
|
|
ds.AddMessage("mb3", new3)
|
|
// Test 4 hour retention
|
|
cfg := config.DataStoreConfig{
|
|
RetentionMinutes: 239,
|
|
RetentionSleep: 0,
|
|
}
|
|
shutdownChan := make(chan bool)
|
|
rs := storage.NewRetentionScanner(cfg, ds, shutdownChan)
|
|
if err := rs.DoScan(); err != nil {
|
|
t.Error(err)
|
|
}
|
|
// Delete should not have been called on new messages
|
|
new1.AssertNotCalled(t, "Delete")
|
|
new2.AssertNotCalled(t, "Delete")
|
|
new3.AssertNotCalled(t, "Delete")
|
|
// Delete should have been called once on old messages
|
|
old1.AssertNumberOfCalls(t, "Delete", 1)
|
|
old2.AssertNumberOfCalls(t, "Delete", 1)
|
|
old3.AssertNumberOfCalls(t, "Delete", 1)
|
|
}
|
|
|
|
// Make a MockMessage of a specific age
|
|
func mockMessage(ageHours int) *storage.MockMessage {
|
|
msg := &storage.MockMessage{}
|
|
msg.On("ID").Return(fmt.Sprintf("MSG[age=%vh]", ageHours))
|
|
msg.On("Date").Return(time.Now().Add(time.Duration(ageHours*-1) * time.Hour))
|
|
msg.On("Delete").Return(nil)
|
|
return msg
|
|
}
|