1 files changed, 70 insertions, 0 deletions
diff --git a/tests/test_cmd_roster.c b/tests/test_cmd_roster.c
new file mode 100644
index 00000000..9cfb711c
--- /dev/null
+++ b/tests/test_cmd_roster.c
@@ -0,0 +1,70 @@
+#include <stdarg.h>
+#include <stddef.h>
+#include <setjmp.h>
+#include <cmocka.h>
+#include <stdlib.h>
+#include <string.h>
+#include <glib.h>
+
+#include "ui/ui.h"
+#include "ui/mock_ui.h"
+
+#include "xmpp/xmpp.h"
+#include "xmpp/mock_xmpp.h"
+
+#include "roster_list.h"
+#include "command/commands.h"
+
+static void test_with_connection_status(jabber_conn_status_t status)
+{
+    mock_cons_show();
+    CommandHelp *help = malloc(sizeof(CommandHelp));
+
+    mock_connection_status(status);
+
+    expect_cons_show("You are not currently connected.");
+
+    gboolean result = cmd_roster(NULL, *help);
+    assert_true(result);
+
+    free(help);
+}
+
+void cmd_roster_shows_message_when_disconnecting(void **state)
+{
+    test_with_connection_status(JABBER_DISCONNECTING);
+}
+
+void cmd_roster_shows_message_when_connecting(void **state)
+{
+    test_with_connection_status(JABBER_CONNECTING);
+}
+
+void cmd_roster_shows_message_when_disconnected(void **state)
+{
+    test_with_connection_status(JABBER_DISCONNECTED);
+}
+
+void cmd_roster_shows_message_when_undefined(void **state)
+{
+    test_with_connection_status(JABBER_UNDEFINED);
+}
+
+void cmd_roster_shows_roster_when_no_args(void **state)
+{
+    mock_cons_show_roster();
+    CommandHelp *help = malloc(sizeof(CommandHelp));
+    gchar *args[] = { NULL };
+
+    mock_connection_status(JABBER_CONNECTED);
+    roster_init();
+    roster_add("bob@server.org", "bob", NULL, "both", FALSE);
+    GSList *roster = roster_get_contacts();
+    cons_show_roster_expect(roster);
+
+    gboolean result = cmd_roster(args, *help);
+    assert_true(result);
+
+    free(help);
+    roster_free();
+}