response_writer_test.go

package porthos

import (
	"encoding/json"
	"os"
	"testing"
	"time"

	"github.com/streadway/amqp"
)

type ResponseExample struct {
	Sum float64 `json:"sum"`
}

func TestResponseWriterJSON(t *testing.T) {
	b, _ := NewBroker(os.Getenv("AMQP_URL"))
	ch, _ := b.openChannel()
	q, _ := ch.QueueDeclare("", false, false, true, false, nil)
	dc, _ := ch.Consume(q.Name, "", true, false, false, false, nil)

	response := newResponse()
	response.JSON(200, ResponseExample{Sum: 10})

	rw := &responseWriter{
		channel: ch,
		autoAck: true,
		delivery: amqp.Delivery{
			ReplyTo:       q.Name,
			CorrelationId: "correlationId",
		},
	}

	rw.Write(response)

	select {
	case response := <-dc:
		if response.Headers["statusCode"].(int32) != 200 {
			t.Errorf("Expected status code was 200, got: %d", response.Headers["statusCode"])
		}

		if response.ContentType != "application/json" {
			t.Errorf("Expected content type application/json, got: %s", response.ContentType)
		}

		var responseTest ResponseExample
		json.Unmarshal(response.Body, &responseTest)

		if responseTest.Sum != 10 {
			t.Errorf("Response failed, expected: 10, got: %s", responseTest.Sum)
		}

		return
	case <-time.After(3 * time.Second):
		t.Fatal("No response receive. Timedout.")
	}
}

func TestResponseWriterRaw(t *testing.T) {
	b, _ := NewBroker(os.Getenv("AMQP_URL"))
	ch, _ := b.openChannel()
	q, _ := ch.QueueDeclare("", false, false, true, false, nil)
	dc, _ := ch.Consume(q.Name, "", true, false, false, false, nil)

	response := newResponse()
	response.Raw(201, "text/plain", []byte("Some Response Text"))

	rw := &responseWriter{
		channel: ch,
		autoAck: true,
		delivery: amqp.Delivery{
			ReplyTo:       q.Name,
			CorrelationId: "correlationId",
		},
	}

	rw.Write(response)

	select {
	case response := <-dc:
		if response.Headers["statusCode"].(int32) != 201 {
			t.Errorf("Expected status code was 201, got: %d", response.Headers["statusCode"])
		}

		if response.ContentType != "text/plain" {
			t.Errorf("Expected content type text/plain, got: %s", response.ContentType)
		}

		if string(response.Body) != "Some Response Text" {
			t.Errorf("Response failed, expected: Some Response Text, got: %s", string(response.Body))
		}

		return
	case <-time.After(3 * time.Second):
		t.Fatal("No response receive. Timedout.")
	}
}

func TestResponseWriterEmpty(t *testing.T) {
	b, _ := NewBroker(os.Getenv("AMQP_URL"))
	ch, _ := b.openChannel()
	q, _ := ch.QueueDeclare("", false, false, true, false, nil)
	dc, _ := ch.Consume(q.Name, "", true, false, false, false, nil)

	response := newResponse()
	response.Empty(202)

	rw := &responseWriter{
		channel: ch,
		autoAck: true,
		delivery: amqp.Delivery{
			ReplyTo:       q.Name,
			CorrelationId: "correlationId",
		},
	}

	rw.Write(response)

	select {
	case response := <-dc:
		if response.Headers["statusCode"].(int32) != 202 {
			t.Errorf("Expected status code was 202, got: %d", response.Headers["statusCode"])
		}

		if response.ContentType != "" {
			t.Errorf("Expected content type nil, got: %s", response.ContentType)
		}

		if string(response.Body) != "" {
			t.Errorf("Response failed, expected: nil, got: %s", string(response.Body))
		}

		return
	case <-time.After(3 * time.Second):
		t.Fatal("No response receive. Timedout.")
	}
}