From 54b593ed476bae09a2926ef0d3eaf42ed36afe7c Mon Sep 17 00:00:00 2001
From: Andrey Petrov <andrey.petrov@shazow.net>
Date: Sun, 21 Dec 2014 14:30:42 -0800
Subject: [PATCH] Message tests.

---
 chat/message_test.go | 34 ++++++++++++++++++++++++++++++++++
 1 file changed, 34 insertions(+)
 create mode 100644 chat/message_test.go

diff --git a/chat/message_test.go b/chat/message_test.go
new file mode 100644
index 0000000..120a7ca
--- /dev/null
+++ b/chat/message_test.go
@@ -0,0 +1,34 @@
+package chat
+
+import "testing"
+
+func TestMessage(t *testing.T) {
+	var expected, actual string
+
+	expected = " * foo"
+	actual = NewMessage("foo").String()
+	if actual != expected {
+		t.Errorf("Got: `%s`; Expected: `%s`", actual, expected)
+	}
+
+	u := NewUser("foo")
+	expected = "foo: hello"
+	actual = NewMessage("hello").From(u).String()
+	if actual != expected {
+		t.Errorf("Got: `%s`; Expected: `%s`", actual, expected)
+	}
+
+	expected = "-> hello"
+	actual = NewMessage("hello").To(u).String()
+	if actual != expected {
+		t.Errorf("Got: `%s`; Expected: `%s`", actual, expected)
+	}
+
+	expected = "[PM from foo] hello"
+	actual = NewMessage("hello").From(u).To(u).String()
+	if actual != expected {
+		t.Errorf("Got: `%s`; Expected: `%s`", actual, expected)
+	}
+}
+
+// TODO: Add theme rendering tests