message_test.go

package sse

import "testing"

func TestEmptyMessage(t *testing.T) {
	msg := Message{}

	if msg.String() != "\n" {
		t.Fatal("Message does not match.")
	}
}

func TestDataMessage(t *testing.T) {
	msg := Message{data: "test"}

	if msg.String() != "data: test\n\n" {
		t.Fatal("Message does not match.")
	}
}

func TestMessage(t *testing.T) {
	msg := Message{
		"123",
		"test",
		"myevent",
		10 * 1000,
	}

	if msg.String() != "id: 123\nretry: 10000\nevent: myevent\ndata: test\n\n" {
		t.Fatal("Message does not match.")
	}
}

func TestMultilineDataMessage(t *testing.T) {
	msg := Message{data: "test\ntest"}

	if msg.String() != "data: test\ndata: test\n\n" {
		t.Fail()
	}
}

func TestSpecialCharacterMessage(t *testing.T) {
	msg := Message{data: "%x%o"}

	if msg.String() != "data: %x%o\n\n" {
		t.Fatal("Message does not match.")
	}
}