Feature/Add MessagesAndPurge to server (#174)

Author: mitar

message.go

@@ -167,3 +167,14 @@ func (messages *messages) copyInternal() []Message {
 
 	return copiedMessages
 }
+
+// Returns all messages and removes them at the same time
+func (messages *messages) purge() []Message {
+	messages.Lock()
+	defer messages.Unlock()
+
+	copiedMessages := messages.copyInternal()
+	messages.items = nil
+
+	return copiedMessages
+}

message_test.go

@@ -256,3 +256,14 @@ func TestMessagesCopy(t *testing.T) {
 		assert.Equal(t, *message, copyMessages[0])
 	})
 }
+
+func TestMessagesPurge(t *testing.T) {
+	t.Run("purges messages from items slice", func(t *testing.T) {
+		message, messages := new(Message), new(messages)
+		messages.append(message)
+
+		assert.Len(t, messages.copy(), 1)
+		assert.Len(t, messages.purge(), 1)
+		assert.Len(t, messages.copy(), 0)
+	})
+}

server.go

@@ -124,6 +124,12 @@ func (server *Server) Messages() []Message {
 	return server.messages.copy()
 }
 
+// Public interface to get access to server messages
+// and at the same time removes them
+func (server *Server) MessagesAndPurge() []Message {
+	return server.messages.purge()
+}
+
 // Thread-safe getter of server port.
 // Returns server.portNumber
 func (server *Server) PortNumber() int {

server_test.go

@@ -163,6 +163,16 @@ func TestServerMessages(t *testing.T) {
 		assert.NotSame(t, server.messages.items, server.Messages())
 		server.messages.RUnlock()
 	})
+
+	t.Run("no messages after purge", func(t *testing.T) {
+		server := newServer(configuration)
+		message := new(Message)
+		server.messages.append(message)
+
+		assert.NotEmpty(t, server.Messages())
+		assert.NotEmpty(t, server.MessagesAndPurge())
+		assert.Empty(t, server.Messages())
+	})
 }
 
 func TestServerPortNumber(t *testing.T) {