github.com/coincircle/mattermost-server@v4.8.1-0.20180321182714-9d701c704416+incompatible/plugin/plugintest/api.go (about)

     1  // Copyright (c) 2015-present Mattermost, Inc. All Rights Reserved.
     2  // See License.txt for license information.
     3  
     4  package plugintest
     5  
     6  import (
     7  	"github.com/stretchr/testify/mock"
     8  
     9  	"github.com/mattermost/mattermost-server/model"
    10  	"github.com/mattermost/mattermost-server/plugin"
    11  )
    12  
    13  type API struct {
    14  	mock.Mock
    15  	Store *KeyValueStore
    16  }
    17  
    18  type KeyValueStore struct {
    19  	mock.Mock
    20  }
    21  
    22  var _ plugin.API = (*API)(nil)
    23  var _ plugin.KeyValueStore = (*KeyValueStore)(nil)
    24  
    25  func (m *API) LoadPluginConfiguration(dest interface{}) error {
    26  	ret := m.Called(dest)
    27  	if f, ok := ret.Get(0).(func(interface{}) error); ok {
    28  		return f(dest)
    29  	}
    30  	return ret.Error(0)
    31  }
    32  
    33  func (m *API) RegisterCommand(command *model.Command) error {
    34  	ret := m.Called(command)
    35  	if f, ok := ret.Get(0).(func(*model.Command) error); ok {
    36  		return f(command)
    37  	}
    38  	return ret.Error(0)
    39  }
    40  
    41  func (m *API) UnregisterCommand(teamId, trigger string) error {
    42  	ret := m.Called(teamId, trigger)
    43  	if f, ok := ret.Get(0).(func(string, string) error); ok {
    44  		return f(teamId, trigger)
    45  	}
    46  	return ret.Error(0)
    47  }
    48  
    49  func (m *API) CreateUser(user *model.User) (*model.User, *model.AppError) {
    50  	ret := m.Called(user)
    51  	if f, ok := ret.Get(0).(func(*model.User) (*model.User, *model.AppError)); ok {
    52  		return f(user)
    53  	}
    54  	userOut, _ := ret.Get(0).(*model.User)
    55  	err, _ := ret.Get(1).(*model.AppError)
    56  	return userOut, err
    57  }
    58  
    59  func (m *API) DeleteUser(userId string) *model.AppError {
    60  	ret := m.Called(userId)
    61  	if f, ok := ret.Get(0).(func(string) *model.AppError); ok {
    62  		return f(userId)
    63  	}
    64  	err, _ := ret.Get(0).(*model.AppError)
    65  	return err
    66  }
    67  
    68  func (m *API) GetUser(userId string) (*model.User, *model.AppError) {
    69  	ret := m.Called(userId)
    70  	if f, ok := ret.Get(0).(func(string) (*model.User, *model.AppError)); ok {
    71  		return f(userId)
    72  	}
    73  	user, _ := ret.Get(0).(*model.User)
    74  	err, _ := ret.Get(1).(*model.AppError)
    75  	return user, err
    76  }
    77  
    78  func (m *API) GetUserByEmail(email string) (*model.User, *model.AppError) {
    79  	ret := m.Called(email)
    80  	if f, ok := ret.Get(0).(func(string) (*model.User, *model.AppError)); ok {
    81  		return f(email)
    82  	}
    83  	user, _ := ret.Get(0).(*model.User)
    84  	err, _ := ret.Get(1).(*model.AppError)
    85  	return user, err
    86  }
    87  
    88  func (m *API) GetUserByUsername(name string) (*model.User, *model.AppError) {
    89  	ret := m.Called(name)
    90  	if f, ok := ret.Get(0).(func(string) (*model.User, *model.AppError)); ok {
    91  		return f(name)
    92  	}
    93  	user, _ := ret.Get(0).(*model.User)
    94  	err, _ := ret.Get(1).(*model.AppError)
    95  	return user, err
    96  }
    97  
    98  func (m *API) UpdateUser(user *model.User) (*model.User, *model.AppError) {
    99  	ret := m.Called(user)
   100  	if f, ok := ret.Get(0).(func(*model.User) (*model.User, *model.AppError)); ok {
   101  		return f(user)
   102  	}
   103  	userOut, _ := ret.Get(0).(*model.User)
   104  	err, _ := ret.Get(1).(*model.AppError)
   105  	return userOut, err
   106  }
   107  
   108  func (m *API) CreateTeam(team *model.Team) (*model.Team, *model.AppError) {
   109  	ret := m.Called(team)
   110  	if f, ok := ret.Get(0).(func(*model.Team) (*model.Team, *model.AppError)); ok {
   111  		return f(team)
   112  	}
   113  	teamOut, _ := ret.Get(0).(*model.Team)
   114  	err, _ := ret.Get(1).(*model.AppError)
   115  	return teamOut, err
   116  }
   117  
   118  func (m *API) DeleteTeam(teamId string) *model.AppError {
   119  	ret := m.Called(teamId)
   120  	if f, ok := ret.Get(0).(func(string) *model.AppError); ok {
   121  		return f(teamId)
   122  	}
   123  	err, _ := ret.Get(0).(*model.AppError)
   124  	return err
   125  }
   126  
   127  func (m *API) GetTeam(teamId string) (*model.Team, *model.AppError) {
   128  	ret := m.Called(teamId)
   129  	if f, ok := ret.Get(0).(func(string) (*model.Team, *model.AppError)); ok {
   130  		return f(teamId)
   131  	}
   132  	team, _ := ret.Get(0).(*model.Team)
   133  	err, _ := ret.Get(1).(*model.AppError)
   134  	return team, err
   135  }
   136  
   137  func (m *API) GetTeamByName(name string) (*model.Team, *model.AppError) {
   138  	ret := m.Called(name)
   139  	if f, ok := ret.Get(0).(func(string) (*model.Team, *model.AppError)); ok {
   140  		return f(name)
   141  	}
   142  	team, _ := ret.Get(0).(*model.Team)
   143  	err, _ := ret.Get(1).(*model.AppError)
   144  	return team, err
   145  }
   146  
   147  func (m *API) UpdateTeam(team *model.Team) (*model.Team, *model.AppError) {
   148  	ret := m.Called(team)
   149  	if f, ok := ret.Get(0).(func(*model.Team) (*model.Team, *model.AppError)); ok {
   150  		return f(team)
   151  	}
   152  	teamOut, _ := ret.Get(0).(*model.Team)
   153  	err, _ := ret.Get(1).(*model.AppError)
   154  	return teamOut, err
   155  }
   156  
   157  func (m *API) CreateChannel(channel *model.Channel) (*model.Channel, *model.AppError) {
   158  	ret := m.Called(channel)
   159  	if f, ok := ret.Get(0).(func(*model.Channel) (*model.Channel, *model.AppError)); ok {
   160  		return f(channel)
   161  	}
   162  	channelOut, _ := ret.Get(0).(*model.Channel)
   163  	err, _ := ret.Get(1).(*model.AppError)
   164  	return channelOut, err
   165  }
   166  
   167  func (m *API) DeleteChannel(channelId string) *model.AppError {
   168  	ret := m.Called(channelId)
   169  	if f, ok := ret.Get(0).(func(string) *model.AppError); ok {
   170  		return f(channelId)
   171  	}
   172  	err, _ := ret.Get(0).(*model.AppError)
   173  	return err
   174  }
   175  
   176  func (m *API) GetChannel(channelId string) (*model.Channel, *model.AppError) {
   177  	ret := m.Called(channelId)
   178  	if f, ok := ret.Get(0).(func(string) (*model.Channel, *model.AppError)); ok {
   179  		return f(channelId)
   180  	}
   181  	channel, _ := ret.Get(0).(*model.Channel)
   182  	err, _ := ret.Get(1).(*model.AppError)
   183  	return channel, err
   184  }
   185  
   186  func (m *API) GetChannelByName(name, teamId string) (*model.Channel, *model.AppError) {
   187  	ret := m.Called(name, teamId)
   188  	if f, ok := ret.Get(0).(func(_, _ string) (*model.Channel, *model.AppError)); ok {
   189  		return f(name, teamId)
   190  	}
   191  	channel, _ := ret.Get(0).(*model.Channel)
   192  	err, _ := ret.Get(1).(*model.AppError)
   193  	return channel, err
   194  }
   195  
   196  func (m *API) GetDirectChannel(userId1, userId2 string) (*model.Channel, *model.AppError) {
   197  	ret := m.Called(userId1, userId2)
   198  	if f, ok := ret.Get(0).(func(_, _ string) (*model.Channel, *model.AppError)); ok {
   199  		return f(userId1, userId2)
   200  	}
   201  	channel, _ := ret.Get(0).(*model.Channel)
   202  	err, _ := ret.Get(1).(*model.AppError)
   203  	return channel, err
   204  }
   205  
   206  func (m *API) GetGroupChannel(userIds []string) (*model.Channel, *model.AppError) {
   207  	ret := m.Called(userIds)
   208  	if f, ok := ret.Get(0).(func([]string) (*model.Channel, *model.AppError)); ok {
   209  		return f(userIds)
   210  	}
   211  	channel, _ := ret.Get(0).(*model.Channel)
   212  	err, _ := ret.Get(1).(*model.AppError)
   213  	return channel, err
   214  }
   215  
   216  func (m *API) UpdateChannel(channel *model.Channel) (*model.Channel, *model.AppError) {
   217  	ret := m.Called(channel)
   218  	if f, ok := ret.Get(0).(func(*model.Channel) (*model.Channel, *model.AppError)); ok {
   219  		return f(channel)
   220  	}
   221  	channelOut, _ := ret.Get(0).(*model.Channel)
   222  	err, _ := ret.Get(1).(*model.AppError)
   223  	return channelOut, err
   224  }
   225  
   226  func (m *API) GetChannelMember(channelId, userId string) (*model.ChannelMember, *model.AppError) {
   227  	ret := m.Called(channelId, userId)
   228  	if f, ok := ret.Get(0).(func(_, _ string) (*model.ChannelMember, *model.AppError)); ok {
   229  		return f(channelId, userId)
   230  	}
   231  	member, _ := ret.Get(0).(*model.ChannelMember)
   232  	err, _ := ret.Get(1).(*model.AppError)
   233  	return member, err
   234  }
   235  
   236  func (m *API) CreatePost(post *model.Post) (*model.Post, *model.AppError) {
   237  	ret := m.Called(post)
   238  	if f, ok := ret.Get(0).(func(*model.Post) (*model.Post, *model.AppError)); ok {
   239  		return f(post)
   240  	}
   241  	postOut, _ := ret.Get(0).(*model.Post)
   242  	err, _ := ret.Get(1).(*model.AppError)
   243  	return postOut, err
   244  }
   245  
   246  func (m *API) DeletePost(postId string) *model.AppError {
   247  	ret := m.Called(postId)
   248  	if f, ok := ret.Get(0).(func(string) *model.AppError); ok {
   249  		return f(postId)
   250  	}
   251  	err, _ := ret.Get(0).(*model.AppError)
   252  	return err
   253  }
   254  
   255  func (m *API) GetPost(postId string) (*model.Post, *model.AppError) {
   256  	ret := m.Called(postId)
   257  	if f, ok := ret.Get(0).(func(string) (*model.Post, *model.AppError)); ok {
   258  		return f(postId)
   259  	}
   260  	post, _ := ret.Get(0).(*model.Post)
   261  	err, _ := ret.Get(1).(*model.AppError)
   262  	return post, err
   263  }
   264  
   265  func (m *API) UpdatePost(post *model.Post) (*model.Post, *model.AppError) {
   266  	ret := m.Called(post)
   267  	if f, ok := ret.Get(0).(func(*model.Post) (*model.Post, *model.AppError)); ok {
   268  		return f(post)
   269  	}
   270  	postOut, _ := ret.Get(0).(*model.Post)
   271  	err, _ := ret.Get(1).(*model.AppError)
   272  	return postOut, err
   273  }
   274  
   275  func (m *API) KeyValueStore() plugin.KeyValueStore {
   276  	return m.Store
   277  }
   278  
   279  func (m *KeyValueStore) Set(key string, value []byte) *model.AppError {
   280  	ret := m.Called(key, value)
   281  	if f, ok := ret.Get(0).(func(string, []byte) *model.AppError); ok {
   282  		return f(key, value)
   283  	}
   284  	err, _ := ret.Get(0).(*model.AppError)
   285  	return err
   286  }
   287  
   288  func (m *KeyValueStore) Get(key string) ([]byte, *model.AppError) {
   289  	ret := m.Called(key)
   290  	if f, ok := ret.Get(0).(func(string) ([]byte, *model.AppError)); ok {
   291  		return f(key)
   292  	}
   293  	psv, _ := ret.Get(0).([]byte)
   294  	err, _ := ret.Get(1).(*model.AppError)
   295  	return psv, err
   296  }
   297  
   298  func (m *KeyValueStore) Delete(key string) *model.AppError {
   299  	ret := m.Called(key)
   300  	if f, ok := ret.Get(0).(func(string) *model.AppError); ok {
   301  		return f(key)
   302  	}
   303  	err, _ := ret.Get(0).(*model.AppError)
   304  	return err
   305  }