Added additional test cases and fixed existing ones for ChannelMember.MarkUnreadLevel

2 files changed, 20 insertions, 0 deletions

diff --git a/model/channel_member_test.go b/model/channel_member_test.go
index 3b64ffbf7..ae3da73cc 100644
--- a/model/channel_member_test.go
+++ b/model/channel_member_test.go
@@ -32,6 +32,7 @@ func TestChannelMemberIsValid(t *testing.T) {
 
 	o.Roles = "missing"
 	o.NotifyLevel = CHANNEL_NOTIFY_ALL
+	o.MarkUnreadLevel = CHANNEL_MARK_UNREAD_ALL
 	o.UserId = NewId()
 	if err := o.IsValid(); err == nil {
 		t.Fatal("should be invalid")
@@ -53,6 +54,16 @@ func TestChannelMemberIsValid(t *testing.T) {
 		t.Fatal(err)
 	}
 
+	o.MarkUnreadLevel = "123456789012345678901"
+	if err := o.IsValid(); err == nil {
+		t.Fatal("should be invalid")
+	}
+
+	o.MarkUnreadLevel = CHANNEL_MARK_UNREAD_ALL
+	if err := o.IsValid(); err != nil {
+		t.Fatal(err)
+	}
+
 	o.Roles = ""
 	if err := o.IsValid(); err != nil {
 		t.Fatal(err)
diff --git a/model/client.go b/model/client.go
index 26e00864d..1d1c5a52c 100644
--- a/model/client.go
+++ b/model/client.go
@@ -459,6 +459,15 @@ func (c *Client) UpdateNotifyLevel(data map[string]string) (*Result, *AppError)
 	}
 }
 
+func (c *Client) UpdateMarkUnreadLevel(data map[string]string) (*Result, *AppError) {
+	if r, err := c.DoApiPost("/channels/update_mark_unread_level", MapToJson(data)); err != nil {
+		return nil, err
+	} else {
+		return &Result{r.Header.Get(HEADER_REQUEST_ID),
+			r.Header.Get(HEADER_ETAG_SERVER), MapFromJson(r.Body)}, nil
+	}
+}
+
 func (c *Client) GetChannels(etag string) (*Result, *AppError) {
 	if r, err := c.DoApiGet("/channels/", "", etag); err != nil {
 		return nil, err