github.com/levb/mattermost-server@v5.3.1+incompatible/cmd/mattermost/commands/roles_test.go (about)

     1  // Copyright (c) 2015-present Mattermost, Inc. All Rights Reserved.
     2  // See License.txt for license information.
     3  
     4  package commands
     5  
     6  import (
     7  	"testing"
     8  
     9  	"github.com/mattermost/mattermost-server/api4"
    10  	"github.com/mattermost/mattermost-server/model"
    11  )
    12  
    13  func TestAssignRole(t *testing.T) {
    14  	th := api4.Setup().InitBasic()
    15  	defer th.TearDown()
    16  
    17  	CheckCommand(t, "roles", "system_admin", th.BasicUser.Email)
    18  
    19  	if result := <-th.App.Srv.Store.User().GetByEmail(th.BasicUser.Email); result.Err != nil {
    20  		t.Fatal(result.Err)
    21  	} else {
    22  		user := result.Data.(*model.User)
    23  		if user.Roles != "system_user system_admin" {
    24  			t.Fatal("Got wrong roles:", user.Roles)
    25  		}
    26  	}
    27  
    28  	CheckCommand(t, "roles", "member", th.BasicUser.Email)
    29  
    30  	if result := <-th.App.Srv.Store.User().GetByEmail(th.BasicUser.Email); result.Err != nil {
    31  		t.Fatal(result.Err)
    32  	} else {
    33  		user := result.Data.(*model.User)
    34  		if user.Roles != "system_user" {
    35  			t.Fatal("Got wrong roles:", user.Roles, user.Id)
    36  		}
    37  	}
    38  }