Add automated tests and Travis-CI configuration

Author: dubek

.gitignore

@@ -1,2 +1,3 @@
 /bin
 /release
+/rabbitmq-dump-queue

.travis.yml

@@ -0,0 +1,13 @@
+sudo: required
+services:
+  - rabbitmq
+
+language: go
+go:
+  - 1.6.2
+  - tip
+
+install: true
+script:
+  - go build .
+  - go test -v .

CHANGELOG.md

@@ -1,5 +1,10 @@
 # Change log
 
+## (in progress)
+
+* Add system tests against a local RabbitMQ server.
+
+
 ## v0.2 (2016-07-01)
 
 * Add `-full` option to dump the message header and properties into

README.md

@@ -81,6 +81,15 @@ Note that the same approach is used by RabbitMQ's management HTTP API (the
 `/api/queues/{vhost}/{queue}/get` endpoint with `requeue=true`).
 
 
+## Testing
+
+To run the automated tests, have a RabbitMQ server listen on 127.0.0.1:5672
+with user `guest` and password `guest`.  Then run:
+
+    go build .
+    go test -v .
+
+
 ## Contributing
 
 Github pull requests and issues are welcome.

main_test.go

@@ -0,0 +1,215 @@
+package main
+
+import (
+	"encoding/json"
+	"fmt"
+	"io/ioutil"
+	"os"
+	"os/exec"
+	"strings"
+	"testing"
+
+	"github.com/streadway/amqp"
+)
+
+const (
+	TEST_AMQP_URI   = "amqp://guest:[email protected]:5672/"
+	TEST_QUEUE_NAME = "test-rabbitmq-dump-queue"
+)
+
+func makeAmqpMessage(i int) amqp.Publishing {
+	headers := make(amqp.Table)
+	headers["my-header"] = fmt.Sprintf("my-value-%d", i)
+	return amqp.Publishing{
+		Headers:     headers,
+		ContentType: "text/plain",
+		Priority:    4,
+		MessageId:   fmt.Sprintf("msgid-%d", i),
+		Body:        []byte(fmt.Sprintf("message-%d-body", i)),
+	}
+}
+
+// Publish 10 messages to the queue
+func populateTestQueue(t *testing.T, messagesToPublish int) {
+	conn, err := amqp.Dial(TEST_AMQP_URI)
+	if err != nil {
+		t.Fatalf("Dial: %s", err)
+	}
+	defer conn.Close()
+
+	channel, err := conn.Channel()
+	if err != nil {
+		t.Fatalf("Channel: %s", err)
+	}
+
+	_, err = channel.QueueDeclare(TEST_QUEUE_NAME, true, false, false, false, nil)
+	if err != nil {
+		t.Fatalf("QueueDeclare: %s", err)
+	}
+
+	_, err = channel.QueuePurge(TEST_QUEUE_NAME, false)
+	if err != nil {
+		t.Fatalf("QueuePurge: %s", err)
+	}
+
+	for i := 0; i < messagesToPublish; i++ {
+		err = channel.Publish("", TEST_QUEUE_NAME, false, false, makeAmqpMessage(i))
+		if err != nil {
+			t.Fatalf("Publish: %s", err)
+		}
+	}
+}
+
+func deleteTestQueue(t *testing.T) {
+	conn, err := amqp.Dial(TEST_AMQP_URI)
+	if err != nil {
+		t.Fatalf("Dial: %s", err)
+	}
+	defer conn.Close()
+
+	channel, err := conn.Channel()
+	if err != nil {
+		t.Fatalf("Channel: %s", err)
+	}
+
+	_, err = channel.QueueDelete(TEST_QUEUE_NAME, false, false, false)
+	if err != nil {
+		t.Fatalf("QueueDelete: %s", err)
+	}
+}
+
+func getTestQueueLength(t *testing.T) int {
+	conn, err := amqp.Dial(TEST_AMQP_URI)
+	if err != nil {
+		t.Fatalf("Dial: %s", err)
+	}
+	defer conn.Close()
+
+	channel, err := conn.Channel()
+	if err != nil {
+		t.Fatalf("Channel: %s", err)
+	}
+
+	queue, err := channel.QueueInspect(TEST_QUEUE_NAME)
+	if err != nil {
+		t.Fatalf("QueueInspect: %s", err)
+	}
+
+	return queue.Messages
+}
+
+func run(t *testing.T, commandLine string) string {
+	queueLengthBeforeDump := getTestQueueLength(t)
+	args := strings.Split(commandLine, " ")
+	output, err := exec.Command("./rabbitmq-dump-queue", args...).CombinedOutput()
+	if err != nil {
+		t.Fatalf("run: %s: %s", err, string(output))
+	}
+	queueLengthAfterDump := getTestQueueLength(t)
+	if queueLengthAfterDump != queueLengthBeforeDump {
+		t.Errorf("Queue length changed after rabbitmq-dump-queue: expected %d but got %d", queueLengthBeforeDump, queueLengthAfterDump)
+	}
+	return string(output)
+}
+
+func verifyFileContent(t *testing.T, filename, expectedContent string) {
+	content, err := ioutil.ReadFile(filename)
+	if err != nil {
+		t.Fatalf("Error reading %s: %s", filename, err)
+	}
+	if expectedContent != string(content) {
+		t.Errorf("Wrong content for %s: expected '%s', got '%s'", filename, expectedContent, string(content))
+	}
+}
+
+func TestNormal(t *testing.T) {
+	os.MkdirAll("tmp-test", 0775)
+	defer os.RemoveAll("tmp-test")
+	populateTestQueue(t, 10)
+	defer deleteTestQueue(t)
+	output := run(t, "-uri="+TEST_AMQP_URI+" -queue="+TEST_QUEUE_NAME+" -max-messages=3 -output-dir=tmp-test")
+	expectedOutput := "tmp-test/msg-0000\n" +
+		"tmp-test/msg-0001\n" +
+		"tmp-test/msg-0002\n"
+	if output != expectedOutput {
+		t.Errorf("Wrong output: expected '%s' but got '%s'", expectedOutput, output)
+	}
+	verifyFileContent(t, "tmp-test/msg-0000", "message-0-body")
+	verifyFileContent(t, "tmp-test/msg-0001", "message-1-body")
+	verifyFileContent(t, "tmp-test/msg-0002", "message-2-body")
+	_, err := os.Stat("tmp-test/msg-0003")
+	if !os.IsNotExist(err) {
+		t.Errorf("Expected msg-0003 to not exist: %v", err)
+	}
+}
+
+func TestEmptyQueue(t *testing.T) {
+	os.MkdirAll("tmp-test", 0775)
+	defer os.RemoveAll("tmp-test")
+	populateTestQueue(t, 0)
+	defer deleteTestQueue(t)
+	output := run(t, "-uri="+TEST_AMQP_URI+" -queue="+TEST_QUEUE_NAME+" -max-messages=3 -output-dir=tmp-test")
+	expectedOutput := ""
+	if output != expectedOutput {
+		t.Errorf("Wrong output: expected '%s' but got '%s'", expectedOutput, output)
+	}
+}
+
+func TestMaxMessagesLargerThanQueueLength(t *testing.T) {
+	os.MkdirAll("tmp-test", 0775)
+	defer os.RemoveAll("tmp-test")
+	populateTestQueue(t, 3)
+	defer deleteTestQueue(t)
+	output := run(t, "-uri="+TEST_AMQP_URI+" -queue="+TEST_QUEUE_NAME+" -max-messages=9 -output-dir=tmp-test")
+	expectedOutput := "tmp-test/msg-0000\n" +
+		"tmp-test/msg-0001\n" +
+		"tmp-test/msg-0002\n"
+	if output != expectedOutput {
+		t.Errorf("Wrong output: expected '%s' but got '%s'", expectedOutput, output)
+	}
+}
+
+func TestFull(t *testing.T) {
+	os.MkdirAll("tmp-test", 0775)
+	defer os.RemoveAll("tmp-test")
+	populateTestQueue(t, 10)
+	defer deleteTestQueue(t)
+	output := run(t, "-uri="+TEST_AMQP_URI+" -queue="+TEST_QUEUE_NAME+" -max-messages=3 -output-dir=tmp-test -full")
+	expectedOutput := "tmp-test/msg-0000\n" +
+		"tmp-test/msg-0000-headers+properties.json\n" +
+		"tmp-test/msg-0001\n" +
+		"tmp-test/msg-0001-headers+properties.json\n" +
+		"tmp-test/msg-0002\n" +
+		"tmp-test/msg-0002-headers+properties.json\n"
+	if output != expectedOutput {
+		t.Errorf("Wrong output: expected '%s' but got '%s'", expectedOutput, output)
+	}
+	verifyFileContent(t, "tmp-test/msg-0000", "message-0-body")
+	jsonContent, err := ioutil.ReadFile("tmp-test/msg-0000-headers+properties.json")
+	if err != nil {
+		t.Fatalf("Error reading tmp-test/msg-0000-headers+properties.json: %s", err)
+	}
+	var v map[string]interface{}
+	err = json.Unmarshal(jsonContent, &v)
+	if err != nil {
+		t.Fatalf("Error unmarshaling JSON: %s", err)
+	}
+
+	headers, ok := v["headers"].(map[string]interface{})
+	if !ok {
+		t.Fatalf("Wrong data type for 'headers' in JSON")
+	}
+	if headers["my-header"] != "my-value-0" {
+		t.Errorf("Wrong value for my-header, got: %v", headers["my-header"])
+	}
+
+	properties, ok := v["properties"].(map[string]interface{})
+	if !ok {
+		t.Fatalf("Wrong data type for 'properties' in JSON")
+	}
+	if properties["priority"] != 4.0 || // JSON numbers are floats
+		properties["content_type"] != "text/plain" ||
+		properties["message_id"] != "msgid-0" {
+		t.Errorf("Wrong property value: properties = %#v", properties)
+	}
+}