1 files changed, 45 insertions, 0 deletions
diff --git a/model/post_test.go b/model/post_test.go
index 1096ccbd8..64c8a2f3d 100644
--- a/model/post_test.go
+++ b/model/post_test.go
@@ -113,3 +113,48 @@ func TestPostIsSystemMessage(t *testing.T) {
 		t.Fatalf("TestPostIsSystemMessage failed, expected post2.IsSystemMessage() to be true")
 	}
 }
+
+func TestPostIsUserActivitySystemMessage(t *testing.T) {
+	post1 := Post{Message: "test_1"}
+	post1.PreSave()
+
+	if post1.IsUserActivitySystemMessage() {
+		t.Fatalf("TestPostIsUserActivitySystemMessage failed, expected post1.IsSystemMessage() to be false")
+	}
+
+	post2 := Post{Message: "test_2", Type: POST_JOIN_LEAVE}
+	post2.PreSave()
+	if !post2.IsUserActivitySystemMessage() {
+		t.Fatalf("TestPostIsUserActivitySystemMessage failed, expected post2.IsSystemMessage() to be true")
+	}
+
+	post3 := Post{Message: "test_3", Type: POST_JOIN_CHANNEL}
+	post3.PreSave()
+	if !post3.IsUserActivitySystemMessage() {
+		t.Fatalf("TestPostIsUserActivitySystemMessage failed, expected post3.IsSystemMessage() to be true")
+	}
+
+	post4 := Post{Message: "test_4", Type: POST_LEAVE_CHANNEL}
+	post4.PreSave()
+	if !post4.IsUserActivitySystemMessage() {
+		t.Fatalf("TestPostIsUserActivitySystemMessage failed, expected post4.IsSystemMessage() to be true")
+	}
+
+	post5 := Post{Message: "test_5", Type: POST_ADD_REMOVE}
+	post5.PreSave()
+	if !post5.IsUserActivitySystemMessage() {
+		t.Fatalf("TestPostIsUserActivitySystemMessage failed, expected post5.IsSystemMessage() to be true")
+	}
+
+	post6 := Post{Message: "test_6", Type: POST_ADD_TO_CHANNEL}
+	post6.PreSave()
+	if !post6.IsUserActivitySystemMessage() {
+		t.Fatalf("TestPostIsUserActivitySystemMessage failed, expected post6.IsSystemMessage() to be true")
+	}
+
+	post7 := Post{Message: "test_7", Type: POST_REMOVE_FROM_CHANNEL}
+	post7.PreSave()
+	if !post7.IsUserActivitySystemMessage() {
+		t.Fatalf("TestPostIsUserActivitySystemMessage failed, expected post7.IsSystemMessage() to be true")
+	}
+}