github.com/cryptomisa/mattermost-server@v5.11.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/model"
    10  )
    11  
    12  func TestAssignRole(t *testing.T) {
    13  	th := Setup().InitBasic()
    14  	defer th.TearDown()
    15  
    16  	th.CheckCommand(t, "roles", "system_admin", th.BasicUser.Email)
    17  
    18  	if result := <-th.App.Srv.Store.User().GetByEmail(th.BasicUser.Email); result.Err != nil {
    19  		t.Fatal(result.Err)
    20  	} else {
    21  		user := result.Data.(*model.User)
    22  		if user.Roles != "system_user system_admin" {
    23  			t.Fatal("Got wrong roles:", user.Roles)
    24  		}
    25  	}
    26  
    27  	th.CheckCommand(t, "roles", "member", th.BasicUser.Email)
    28  
    29  	if result := <-th.App.Srv.Store.User().GetByEmail(th.BasicUser.Email); result.Err != nil {
    30  		t.Fatal(result.Err)
    31  	} else {
    32  		user := result.Data.(*model.User)
    33  		if user.Roles != "system_user" {
    34  			t.Fatal("Got wrong roles:", user.Roles, user.Id)
    35  		}
    36  	}
    37  }