Add REST call for purging an entire mailbox

smtpd/datastore.go

@@ -15,6 +15,7 @@ type DataStore interface {
 type Mailbox interface {
 	GetMessages() ([]Message, error)
 	GetMessage(id string) (Message, error)
+	Purge() error
 	NewMessage() Message
 	String() string
 }

smtpd/filestore.go

@@ -178,6 +178,12 @@ func (mb *FileMailbox) GetMessage(id string) (Message, error) {
 	return nil, fmt.Errorf("Message %s not in index", id)
 }
 
+// Delete all messages in this mailbox
+func (mb *FileMailbox) Purge() error {
+	mb.messages = mb.messages[:0]
+	return mb.writeIndex()
+}
+
 // readIndex loads the mailbox index data from disk
 func (mb *FileMailbox) readIndex() error {
 	// Clear message slice, open index

smtpd/filestore_test.go

@@ -218,6 +218,47 @@ func TestFSDelete(t *testing.T) {
 
 }
 
+// Test purging a mailbox
+func TestFSPurge(t *testing.T) {
+	ds := setupDataStore()
+	defer teardownDataStore(ds)
+
+	mbName := "fred"
+	subjects := []string{"alpha", "bravo", "charlie", "delta", "echo"}
+
+	for _, subj := range subjects {
+		// Add a message
+		deliverMessage(ds, mbName, subj, time.Now())
+	}
+
+	mb, err := ds.MailboxFor(mbName)
+	if err != nil {
+		panic(err)
+	}
+	msgs, err := mb.GetMessages()
+	if err != nil {
+		panic(err)
+	}
+	assert.Equal(t, len(subjects), len(msgs), "Expected %v message(s), but got %v",
+		len(subjects), len(msgs))
+
+	// Purge mailbox
+	err = mb.Purge()
+	assert.Nil(t, err)
+
+	// Confirm deletion
+	mb, err = ds.MailboxFor(mbName)
+	if err != nil {
+		panic(err)
+	}
+	msgs, err = mb.GetMessages()
+	if err != nil {
+		panic(err)
+	}
+
+	assert.Equal(t, len(msgs), 0, "Expected mailbox to have zero messages, got %v", len(msgs))
+}
+
 // Test message size calculation
 func TestFSSize(t *testing.T) {
 	ds := setupDataStore()

smtpd/retention_test.go

@@ -92,6 +92,11 @@ func (m *MockMailbox) GetMessage(id string) (Message, error) {
 	return args.Get(0).(Message), args.Error(1)
 }
 
+func (m *MockMailbox) Purge() error {
+	args := m.Called()
+	return args.Error(0)
+}
+
 func (m *MockMailbox) NewMessage() Message {
 	args := m.Called()
 	return args.Get(0).(Message)