1 files changed, 45 insertions, 0 deletions
diff --git a/test/test_user.py b/test/test_user.py
new file mode 100644
index 00000000..a462f45b
--- /dev/null
+++ b/test/test_user.py
@@ -0,0 +1,45 @@
+"""
+Tests for the User class
+"""
+
+import pytest
+from datetime import datetime
+from slixmpp import JID
+from poezio.user import User
+
+
+@pytest.fixture
+def user1():
+    return User(
+        'nick1',
+        'member',
+        'xa',
+        'My Status!',
+        'moderator',
+        JID('foo@muc/nick1'),
+        False,
+        'red',
+    )
+
+
+def test_new_user(user1):
+    assert user1.last_talked == datetime(1, 1, 1)
+    assert user1.jid == JID('foo@muc/nick1')
+    assert user1.chatstate is None
+    assert user1.affiliation == 'member'
+    assert user1.show == 'xa'
+    assert user1.status == 'My Status!'
+    assert user1.role == 'moderator'
+    assert user1.nick == 'nick1'
+    assert user1.color == (196, -1)
+    assert str(user1) == '>nick1<'
+
+
+def test_change_nick(user1):
+    user1.change_nick('nick2')
+    assert user1.nick == 'nick2'
+
+
+def test_change_color(user1):
+    user1.change_color('blue', deterministic=False)
+    assert user1.color == (21, -1)