github.com/juju/juju@v0.0.0-20240430160146-1752b71fcf00/worker/uniter/runner/mocks/context_mock.go (about)

     1  // Code generated by MockGen. DO NOT EDIT.
     2  // Source: github.com/juju/juju/worker/uniter/runner/context (interfaces: Context)
     3  //
     4  // Generated by this command:
     5  //
     6  //	mockgen -package mocks -destination mocks/context_mock.go github.com/juju/juju/worker/uniter/runner/context Context
     7  //
     8  
     9  // Package mocks is a generated GoMock package.
    10  package mocks
    11  
    12  import (
    13  	reflect "reflect"
    14  	time "time"
    15  
    16  	charm "github.com/juju/charm/v12"
    17  	application "github.com/juju/juju/core/application"
    18  	model "github.com/juju/juju/core/model"
    19  	network "github.com/juju/juju/core/network"
    20  	payloads "github.com/juju/juju/core/payloads"
    21  	secrets "github.com/juju/juju/core/secrets"
    22  	params "github.com/juju/juju/rpc/params"
    23  	context "github.com/juju/juju/worker/uniter/runner/context"
    24  	jujuc "github.com/juju/juju/worker/uniter/runner/jujuc"
    25  	loggo "github.com/juju/loggo"
    26  	names "github.com/juju/names/v5"
    27  	gomock "go.uber.org/mock/gomock"
    28  )
    29  
    30  // MockContext is a mock of Context interface.
    31  type MockContext struct {
    32  	ctrl     *gomock.Controller
    33  	recorder *MockContextMockRecorder
    34  }
    35  
    36  // MockContextMockRecorder is the mock recorder for MockContext.
    37  type MockContextMockRecorder struct {
    38  	mock *MockContext
    39  }
    40  
    41  // NewMockContext creates a new mock instance.
    42  func NewMockContext(ctrl *gomock.Controller) *MockContext {
    43  	mock := &MockContext{ctrl: ctrl}
    44  	mock.recorder = &MockContextMockRecorder{mock}
    45  	return mock
    46  }
    47  
    48  // EXPECT returns an object that allows the caller to indicate expected use.
    49  func (m *MockContext) EXPECT() *MockContextMockRecorder {
    50  	return m.recorder
    51  }
    52  
    53  // ActionData mocks base method.
    54  func (m *MockContext) ActionData() (*context.ActionData, error) {
    55  	m.ctrl.T.Helper()
    56  	ret := m.ctrl.Call(m, "ActionData")
    57  	ret0, _ := ret[0].(*context.ActionData)
    58  	ret1, _ := ret[1].(error)
    59  	return ret0, ret1
    60  }
    61  
    62  // ActionData indicates an expected call of ActionData.
    63  func (mr *MockContextMockRecorder) ActionData() *gomock.Call {
    64  	mr.mock.ctrl.T.Helper()
    65  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ActionData", reflect.TypeOf((*MockContext)(nil).ActionData))
    66  }
    67  
    68  // ActionParams mocks base method.
    69  func (m *MockContext) ActionParams() (map[string]any, error) {
    70  	m.ctrl.T.Helper()
    71  	ret := m.ctrl.Call(m, "ActionParams")
    72  	ret0, _ := ret[0].(map[string]any)
    73  	ret1, _ := ret[1].(error)
    74  	return ret0, ret1
    75  }
    76  
    77  // ActionParams indicates an expected call of ActionParams.
    78  func (mr *MockContextMockRecorder) ActionParams() *gomock.Call {
    79  	mr.mock.ctrl.T.Helper()
    80  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ActionParams", reflect.TypeOf((*MockContext)(nil).ActionParams))
    81  }
    82  
    83  // AddMetric mocks base method.
    84  func (m *MockContext) AddMetric(arg0, arg1 string, arg2 time.Time) error {
    85  	m.ctrl.T.Helper()
    86  	ret := m.ctrl.Call(m, "AddMetric", arg0, arg1, arg2)
    87  	ret0, _ := ret[0].(error)
    88  	return ret0
    89  }
    90  
    91  // AddMetric indicates an expected call of AddMetric.
    92  func (mr *MockContextMockRecorder) AddMetric(arg0, arg1, arg2 any) *gomock.Call {
    93  	mr.mock.ctrl.T.Helper()
    94  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "AddMetric", reflect.TypeOf((*MockContext)(nil).AddMetric), arg0, arg1, arg2)
    95  }
    96  
    97  // AddMetricLabels mocks base method.
    98  func (m *MockContext) AddMetricLabels(arg0, arg1 string, arg2 time.Time, arg3 map[string]string) error {
    99  	m.ctrl.T.Helper()
   100  	ret := m.ctrl.Call(m, "AddMetricLabels", arg0, arg1, arg2, arg3)
   101  	ret0, _ := ret[0].(error)
   102  	return ret0
   103  }
   104  
   105  // AddMetricLabels indicates an expected call of AddMetricLabels.
   106  func (mr *MockContextMockRecorder) AddMetricLabels(arg0, arg1, arg2, arg3 any) *gomock.Call {
   107  	mr.mock.ctrl.T.Helper()
   108  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "AddMetricLabels", reflect.TypeOf((*MockContext)(nil).AddMetricLabels), arg0, arg1, arg2, arg3)
   109  }
   110  
   111  // AddUnitStorage mocks base method.
   112  func (m *MockContext) AddUnitStorage(arg0 map[string]params.StorageConstraints) error {
   113  	m.ctrl.T.Helper()
   114  	ret := m.ctrl.Call(m, "AddUnitStorage", arg0)
   115  	ret0, _ := ret[0].(error)
   116  	return ret0
   117  }
   118  
   119  // AddUnitStorage indicates an expected call of AddUnitStorage.
   120  func (mr *MockContextMockRecorder) AddUnitStorage(arg0 any) *gomock.Call {
   121  	mr.mock.ctrl.T.Helper()
   122  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "AddUnitStorage", reflect.TypeOf((*MockContext)(nil).AddUnitStorage), arg0)
   123  }
   124  
   125  // ApplicationStatus mocks base method.
   126  func (m *MockContext) ApplicationStatus() (jujuc.ApplicationStatusInfo, error) {
   127  	m.ctrl.T.Helper()
   128  	ret := m.ctrl.Call(m, "ApplicationStatus")
   129  	ret0, _ := ret[0].(jujuc.ApplicationStatusInfo)
   130  	ret1, _ := ret[1].(error)
   131  	return ret0, ret1
   132  }
   133  
   134  // ApplicationStatus indicates an expected call of ApplicationStatus.
   135  func (mr *MockContextMockRecorder) ApplicationStatus() *gomock.Call {
   136  	mr.mock.ctrl.T.Helper()
   137  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ApplicationStatus", reflect.TypeOf((*MockContext)(nil).ApplicationStatus))
   138  }
   139  
   140  // AvailabilityZone mocks base method.
   141  func (m *MockContext) AvailabilityZone() (string, error) {
   142  	m.ctrl.T.Helper()
   143  	ret := m.ctrl.Call(m, "AvailabilityZone")
   144  	ret0, _ := ret[0].(string)
   145  	ret1, _ := ret[1].(error)
   146  	return ret0, ret1
   147  }
   148  
   149  // AvailabilityZone indicates an expected call of AvailabilityZone.
   150  func (mr *MockContextMockRecorder) AvailabilityZone() *gomock.Call {
   151  	mr.mock.ctrl.T.Helper()
   152  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "AvailabilityZone", reflect.TypeOf((*MockContext)(nil).AvailabilityZone))
   153  }
   154  
   155  // ClosePortRange mocks base method.
   156  func (m *MockContext) ClosePortRange(arg0 string, arg1 network.PortRange) error {
   157  	m.ctrl.T.Helper()
   158  	ret := m.ctrl.Call(m, "ClosePortRange", arg0, arg1)
   159  	ret0, _ := ret[0].(error)
   160  	return ret0
   161  }
   162  
   163  // ClosePortRange indicates an expected call of ClosePortRange.
   164  func (mr *MockContextMockRecorder) ClosePortRange(arg0, arg1 any) *gomock.Call {
   165  	mr.mock.ctrl.T.Helper()
   166  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ClosePortRange", reflect.TypeOf((*MockContext)(nil).ClosePortRange), arg0, arg1)
   167  }
   168  
   169  // CloudSpec mocks base method.
   170  func (m *MockContext) CloudSpec() (*params.CloudSpec, error) {
   171  	m.ctrl.T.Helper()
   172  	ret := m.ctrl.Call(m, "CloudSpec")
   173  	ret0, _ := ret[0].(*params.CloudSpec)
   174  	ret1, _ := ret[1].(error)
   175  	return ret0, ret1
   176  }
   177  
   178  // CloudSpec indicates an expected call of CloudSpec.
   179  func (mr *MockContextMockRecorder) CloudSpec() *gomock.Call {
   180  	mr.mock.ctrl.T.Helper()
   181  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CloudSpec", reflect.TypeOf((*MockContext)(nil).CloudSpec))
   182  }
   183  
   184  // ConfigSettings mocks base method.
   185  func (m *MockContext) ConfigSettings() (charm.Settings, error) {
   186  	m.ctrl.T.Helper()
   187  	ret := m.ctrl.Call(m, "ConfigSettings")
   188  	ret0, _ := ret[0].(charm.Settings)
   189  	ret1, _ := ret[1].(error)
   190  	return ret0, ret1
   191  }
   192  
   193  // ConfigSettings indicates an expected call of ConfigSettings.
   194  func (mr *MockContextMockRecorder) ConfigSettings() *gomock.Call {
   195  	mr.mock.ctrl.T.Helper()
   196  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ConfigSettings", reflect.TypeOf((*MockContext)(nil).ConfigSettings))
   197  }
   198  
   199  // CreateSecret mocks base method.
   200  func (m *MockContext) CreateSecret(arg0 *jujuc.SecretCreateArgs) (*secrets.URI, error) {
   201  	m.ctrl.T.Helper()
   202  	ret := m.ctrl.Call(m, "CreateSecret", arg0)
   203  	ret0, _ := ret[0].(*secrets.URI)
   204  	ret1, _ := ret[1].(error)
   205  	return ret0, ret1
   206  }
   207  
   208  // CreateSecret indicates an expected call of CreateSecret.
   209  func (mr *MockContextMockRecorder) CreateSecret(arg0 any) *gomock.Call {
   210  	mr.mock.ctrl.T.Helper()
   211  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateSecret", reflect.TypeOf((*MockContext)(nil).CreateSecret), arg0)
   212  }
   213  
   214  // DeleteCharmStateValue mocks base method.
   215  func (m *MockContext) DeleteCharmStateValue(arg0 string) error {
   216  	m.ctrl.T.Helper()
   217  	ret := m.ctrl.Call(m, "DeleteCharmStateValue", arg0)
   218  	ret0, _ := ret[0].(error)
   219  	return ret0
   220  }
   221  
   222  // DeleteCharmStateValue indicates an expected call of DeleteCharmStateValue.
   223  func (mr *MockContextMockRecorder) DeleteCharmStateValue(arg0 any) *gomock.Call {
   224  	mr.mock.ctrl.T.Helper()
   225  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteCharmStateValue", reflect.TypeOf((*MockContext)(nil).DeleteCharmStateValue), arg0)
   226  }
   227  
   228  // DownloadResource mocks base method.
   229  func (m *MockContext) DownloadResource(arg0 string) (string, error) {
   230  	m.ctrl.T.Helper()
   231  	ret := m.ctrl.Call(m, "DownloadResource", arg0)
   232  	ret0, _ := ret[0].(string)
   233  	ret1, _ := ret[1].(error)
   234  	return ret0, ret1
   235  }
   236  
   237  // DownloadResource indicates an expected call of DownloadResource.
   238  func (mr *MockContextMockRecorder) DownloadResource(arg0 any) *gomock.Call {
   239  	mr.mock.ctrl.T.Helper()
   240  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DownloadResource", reflect.TypeOf((*MockContext)(nil).DownloadResource), arg0)
   241  }
   242  
   243  // Flush mocks base method.
   244  func (m *MockContext) Flush(arg0 string, arg1 error) error {
   245  	m.ctrl.T.Helper()
   246  	ret := m.ctrl.Call(m, "Flush", arg0, arg1)
   247  	ret0, _ := ret[0].(error)
   248  	return ret0
   249  }
   250  
   251  // Flush indicates an expected call of Flush.
   252  func (mr *MockContextMockRecorder) Flush(arg0, arg1 any) *gomock.Call {
   253  	mr.mock.ctrl.T.Helper()
   254  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Flush", reflect.TypeOf((*MockContext)(nil).Flush), arg0, arg1)
   255  }
   256  
   257  // FlushPayloads mocks base method.
   258  func (m *MockContext) FlushPayloads() error {
   259  	m.ctrl.T.Helper()
   260  	ret := m.ctrl.Call(m, "FlushPayloads")
   261  	ret0, _ := ret[0].(error)
   262  	return ret0
   263  }
   264  
   265  // FlushPayloads indicates an expected call of FlushPayloads.
   266  func (mr *MockContextMockRecorder) FlushPayloads() *gomock.Call {
   267  	mr.mock.ctrl.T.Helper()
   268  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "FlushPayloads", reflect.TypeOf((*MockContext)(nil).FlushPayloads))
   269  }
   270  
   271  // GetCharmState mocks base method.
   272  func (m *MockContext) GetCharmState() (map[string]string, error) {
   273  	m.ctrl.T.Helper()
   274  	ret := m.ctrl.Call(m, "GetCharmState")
   275  	ret0, _ := ret[0].(map[string]string)
   276  	ret1, _ := ret[1].(error)
   277  	return ret0, ret1
   278  }
   279  
   280  // GetCharmState indicates an expected call of GetCharmState.
   281  func (mr *MockContextMockRecorder) GetCharmState() *gomock.Call {
   282  	mr.mock.ctrl.T.Helper()
   283  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetCharmState", reflect.TypeOf((*MockContext)(nil).GetCharmState))
   284  }
   285  
   286  // GetCharmStateValue mocks base method.
   287  func (m *MockContext) GetCharmStateValue(arg0 string) (string, error) {
   288  	m.ctrl.T.Helper()
   289  	ret := m.ctrl.Call(m, "GetCharmStateValue", arg0)
   290  	ret0, _ := ret[0].(string)
   291  	ret1, _ := ret[1].(error)
   292  	return ret0, ret1
   293  }
   294  
   295  // GetCharmStateValue indicates an expected call of GetCharmStateValue.
   296  func (mr *MockContextMockRecorder) GetCharmStateValue(arg0 any) *gomock.Call {
   297  	mr.mock.ctrl.T.Helper()
   298  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetCharmStateValue", reflect.TypeOf((*MockContext)(nil).GetCharmStateValue), arg0)
   299  }
   300  
   301  // GetLogger mocks base method.
   302  func (m *MockContext) GetLogger(arg0 string) loggo.Logger {
   303  	m.ctrl.T.Helper()
   304  	ret := m.ctrl.Call(m, "GetLogger", arg0)
   305  	ret0, _ := ret[0].(loggo.Logger)
   306  	return ret0
   307  }
   308  
   309  // GetLogger indicates an expected call of GetLogger.
   310  func (mr *MockContextMockRecorder) GetLogger(arg0 any) *gomock.Call {
   311  	mr.mock.ctrl.T.Helper()
   312  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetLogger", reflect.TypeOf((*MockContext)(nil).GetLogger), arg0)
   313  }
   314  
   315  // GetPayload mocks base method.
   316  func (m *MockContext) GetPayload(arg0, arg1 string) (*payloads.Payload, error) {
   317  	m.ctrl.T.Helper()
   318  	ret := m.ctrl.Call(m, "GetPayload", arg0, arg1)
   319  	ret0, _ := ret[0].(*payloads.Payload)
   320  	ret1, _ := ret[1].(error)
   321  	return ret0, ret1
   322  }
   323  
   324  // GetPayload indicates an expected call of GetPayload.
   325  func (mr *MockContextMockRecorder) GetPayload(arg0, arg1 any) *gomock.Call {
   326  	mr.mock.ctrl.T.Helper()
   327  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetPayload", reflect.TypeOf((*MockContext)(nil).GetPayload), arg0, arg1)
   328  }
   329  
   330  // GetPodSpec mocks base method.
   331  func (m *MockContext) GetPodSpec() (string, error) {
   332  	m.ctrl.T.Helper()
   333  	ret := m.ctrl.Call(m, "GetPodSpec")
   334  	ret0, _ := ret[0].(string)
   335  	ret1, _ := ret[1].(error)
   336  	return ret0, ret1
   337  }
   338  
   339  // GetPodSpec indicates an expected call of GetPodSpec.
   340  func (mr *MockContextMockRecorder) GetPodSpec() *gomock.Call {
   341  	mr.mock.ctrl.T.Helper()
   342  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetPodSpec", reflect.TypeOf((*MockContext)(nil).GetPodSpec))
   343  }
   344  
   345  // GetRawK8sSpec mocks base method.
   346  func (m *MockContext) GetRawK8sSpec() (string, error) {
   347  	m.ctrl.T.Helper()
   348  	ret := m.ctrl.Call(m, "GetRawK8sSpec")
   349  	ret0, _ := ret[0].(string)
   350  	ret1, _ := ret[1].(error)
   351  	return ret0, ret1
   352  }
   353  
   354  // GetRawK8sSpec indicates an expected call of GetRawK8sSpec.
   355  func (mr *MockContextMockRecorder) GetRawK8sSpec() *gomock.Call {
   356  	mr.mock.ctrl.T.Helper()
   357  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetRawK8sSpec", reflect.TypeOf((*MockContext)(nil).GetRawK8sSpec))
   358  }
   359  
   360  // GetSecret mocks base method.
   361  func (m *MockContext) GetSecret(arg0 *secrets.URI, arg1 string, arg2, arg3 bool) (secrets.SecretValue, error) {
   362  	m.ctrl.T.Helper()
   363  	ret := m.ctrl.Call(m, "GetSecret", arg0, arg1, arg2, arg3)
   364  	ret0, _ := ret[0].(secrets.SecretValue)
   365  	ret1, _ := ret[1].(error)
   366  	return ret0, ret1
   367  }
   368  
   369  // GetSecret indicates an expected call of GetSecret.
   370  func (mr *MockContextMockRecorder) GetSecret(arg0, arg1, arg2, arg3 any) *gomock.Call {
   371  	mr.mock.ctrl.T.Helper()
   372  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetSecret", reflect.TypeOf((*MockContext)(nil).GetSecret), arg0, arg1, arg2, arg3)
   373  }
   374  
   375  // GoalState mocks base method.
   376  func (m *MockContext) GoalState() (*application.GoalState, error) {
   377  	m.ctrl.T.Helper()
   378  	ret := m.ctrl.Call(m, "GoalState")
   379  	ret0, _ := ret[0].(*application.GoalState)
   380  	ret1, _ := ret[1].(error)
   381  	return ret0, ret1
   382  }
   383  
   384  // GoalState indicates an expected call of GoalState.
   385  func (mr *MockContextMockRecorder) GoalState() *gomock.Call {
   386  	mr.mock.ctrl.T.Helper()
   387  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GoalState", reflect.TypeOf((*MockContext)(nil).GoalState))
   388  }
   389  
   390  // GrantSecret mocks base method.
   391  func (m *MockContext) GrantSecret(arg0 *secrets.URI, arg1 *jujuc.SecretGrantRevokeArgs) error {
   392  	m.ctrl.T.Helper()
   393  	ret := m.ctrl.Call(m, "GrantSecret", arg0, arg1)
   394  	ret0, _ := ret[0].(error)
   395  	return ret0
   396  }
   397  
   398  // GrantSecret indicates an expected call of GrantSecret.
   399  func (mr *MockContextMockRecorder) GrantSecret(arg0, arg1 any) *gomock.Call {
   400  	mr.mock.ctrl.T.Helper()
   401  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GrantSecret", reflect.TypeOf((*MockContext)(nil).GrantSecret), arg0, arg1)
   402  }
   403  
   404  // HasExecutionSetUnitStatus mocks base method.
   405  func (m *MockContext) HasExecutionSetUnitStatus() bool {
   406  	m.ctrl.T.Helper()
   407  	ret := m.ctrl.Call(m, "HasExecutionSetUnitStatus")
   408  	ret0, _ := ret[0].(bool)
   409  	return ret0
   410  }
   411  
   412  // HasExecutionSetUnitStatus indicates an expected call of HasExecutionSetUnitStatus.
   413  func (mr *MockContextMockRecorder) HasExecutionSetUnitStatus() *gomock.Call {
   414  	mr.mock.ctrl.T.Helper()
   415  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "HasExecutionSetUnitStatus", reflect.TypeOf((*MockContext)(nil).HasExecutionSetUnitStatus))
   416  }
   417  
   418  // HookRelation mocks base method.
   419  func (m *MockContext) HookRelation() (jujuc.ContextRelation, error) {
   420  	m.ctrl.T.Helper()
   421  	ret := m.ctrl.Call(m, "HookRelation")
   422  	ret0, _ := ret[0].(jujuc.ContextRelation)
   423  	ret1, _ := ret[1].(error)
   424  	return ret0, ret1
   425  }
   426  
   427  // HookRelation indicates an expected call of HookRelation.
   428  func (mr *MockContextMockRecorder) HookRelation() *gomock.Call {
   429  	mr.mock.ctrl.T.Helper()
   430  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "HookRelation", reflect.TypeOf((*MockContext)(nil).HookRelation))
   431  }
   432  
   433  // HookStorage mocks base method.
   434  func (m *MockContext) HookStorage() (jujuc.ContextStorageAttachment, error) {
   435  	m.ctrl.T.Helper()
   436  	ret := m.ctrl.Call(m, "HookStorage")
   437  	ret0, _ := ret[0].(jujuc.ContextStorageAttachment)
   438  	ret1, _ := ret[1].(error)
   439  	return ret0, ret1
   440  }
   441  
   442  // HookStorage indicates an expected call of HookStorage.
   443  func (mr *MockContextMockRecorder) HookStorage() *gomock.Call {
   444  	mr.mock.ctrl.T.Helper()
   445  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "HookStorage", reflect.TypeOf((*MockContext)(nil).HookStorage))
   446  }
   447  
   448  // HookVars mocks base method.
   449  func (m *MockContext) HookVars(arg0 context.Paths, arg1 bool, arg2 context.Environmenter) ([]string, error) {
   450  	m.ctrl.T.Helper()
   451  	ret := m.ctrl.Call(m, "HookVars", arg0, arg1, arg2)
   452  	ret0, _ := ret[0].([]string)
   453  	ret1, _ := ret[1].(error)
   454  	return ret0, ret1
   455  }
   456  
   457  // HookVars indicates an expected call of HookVars.
   458  func (mr *MockContextMockRecorder) HookVars(arg0, arg1, arg2 any) *gomock.Call {
   459  	mr.mock.ctrl.T.Helper()
   460  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "HookVars", reflect.TypeOf((*MockContext)(nil).HookVars), arg0, arg1, arg2)
   461  }
   462  
   463  // Id mocks base method.
   464  func (m *MockContext) Id() string {
   465  	m.ctrl.T.Helper()
   466  	ret := m.ctrl.Call(m, "Id")
   467  	ret0, _ := ret[0].(string)
   468  	return ret0
   469  }
   470  
   471  // Id indicates an expected call of Id.
   472  func (mr *MockContextMockRecorder) Id() *gomock.Call {
   473  	mr.mock.ctrl.T.Helper()
   474  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Id", reflect.TypeOf((*MockContext)(nil).Id))
   475  }
   476  
   477  // IsLeader mocks base method.
   478  func (m *MockContext) IsLeader() (bool, error) {
   479  	m.ctrl.T.Helper()
   480  	ret := m.ctrl.Call(m, "IsLeader")
   481  	ret0, _ := ret[0].(bool)
   482  	ret1, _ := ret[1].(error)
   483  	return ret0, ret1
   484  }
   485  
   486  // IsLeader indicates an expected call of IsLeader.
   487  func (mr *MockContextMockRecorder) IsLeader() *gomock.Call {
   488  	mr.mock.ctrl.T.Helper()
   489  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "IsLeader", reflect.TypeOf((*MockContext)(nil).IsLeader))
   490  }
   491  
   492  // LeaderSettings mocks base method.
   493  func (m *MockContext) LeaderSettings() (map[string]string, error) {
   494  	m.ctrl.T.Helper()
   495  	ret := m.ctrl.Call(m, "LeaderSettings")
   496  	ret0, _ := ret[0].(map[string]string)
   497  	ret1, _ := ret[1].(error)
   498  	return ret0, ret1
   499  }
   500  
   501  // LeaderSettings indicates an expected call of LeaderSettings.
   502  func (mr *MockContextMockRecorder) LeaderSettings() *gomock.Call {
   503  	mr.mock.ctrl.T.Helper()
   504  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "LeaderSettings", reflect.TypeOf((*MockContext)(nil).LeaderSettings))
   505  }
   506  
   507  // ListPayloads mocks base method.
   508  func (m *MockContext) ListPayloads() ([]string, error) {
   509  	m.ctrl.T.Helper()
   510  	ret := m.ctrl.Call(m, "ListPayloads")
   511  	ret0, _ := ret[0].([]string)
   512  	ret1, _ := ret[1].(error)
   513  	return ret0, ret1
   514  }
   515  
   516  // ListPayloads indicates an expected call of ListPayloads.
   517  func (mr *MockContextMockRecorder) ListPayloads() *gomock.Call {
   518  	mr.mock.ctrl.T.Helper()
   519  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListPayloads", reflect.TypeOf((*MockContext)(nil).ListPayloads))
   520  }
   521  
   522  // LogActionMessage mocks base method.
   523  func (m *MockContext) LogActionMessage(arg0 string) error {
   524  	m.ctrl.T.Helper()
   525  	ret := m.ctrl.Call(m, "LogActionMessage", arg0)
   526  	ret0, _ := ret[0].(error)
   527  	return ret0
   528  }
   529  
   530  // LogActionMessage indicates an expected call of LogActionMessage.
   531  func (mr *MockContextMockRecorder) LogActionMessage(arg0 any) *gomock.Call {
   532  	mr.mock.ctrl.T.Helper()
   533  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "LogActionMessage", reflect.TypeOf((*MockContext)(nil).LogActionMessage), arg0)
   534  }
   535  
   536  // ModelType mocks base method.
   537  func (m *MockContext) ModelType() model.ModelType {
   538  	m.ctrl.T.Helper()
   539  	ret := m.ctrl.Call(m, "ModelType")
   540  	ret0, _ := ret[0].(model.ModelType)
   541  	return ret0
   542  }
   543  
   544  // ModelType indicates an expected call of ModelType.
   545  func (mr *MockContextMockRecorder) ModelType() *gomock.Call {
   546  	mr.mock.ctrl.T.Helper()
   547  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ModelType", reflect.TypeOf((*MockContext)(nil).ModelType))
   548  }
   549  
   550  // NetworkInfo mocks base method.
   551  func (m *MockContext) NetworkInfo(arg0 []string, arg1 int) (map[string]params.NetworkInfoResult, error) {
   552  	m.ctrl.T.Helper()
   553  	ret := m.ctrl.Call(m, "NetworkInfo", arg0, arg1)
   554  	ret0, _ := ret[0].(map[string]params.NetworkInfoResult)
   555  	ret1, _ := ret[1].(error)
   556  	return ret0, ret1
   557  }
   558  
   559  // NetworkInfo indicates an expected call of NetworkInfo.
   560  func (mr *MockContextMockRecorder) NetworkInfo(arg0, arg1 any) *gomock.Call {
   561  	mr.mock.ctrl.T.Helper()
   562  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "NetworkInfo", reflect.TypeOf((*MockContext)(nil).NetworkInfo), arg0, arg1)
   563  }
   564  
   565  // OpenPortRange mocks base method.
   566  func (m *MockContext) OpenPortRange(arg0 string, arg1 network.PortRange) error {
   567  	m.ctrl.T.Helper()
   568  	ret := m.ctrl.Call(m, "OpenPortRange", arg0, arg1)
   569  	ret0, _ := ret[0].(error)
   570  	return ret0
   571  }
   572  
   573  // OpenPortRange indicates an expected call of OpenPortRange.
   574  func (mr *MockContextMockRecorder) OpenPortRange(arg0, arg1 any) *gomock.Call {
   575  	mr.mock.ctrl.T.Helper()
   576  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "OpenPortRange", reflect.TypeOf((*MockContext)(nil).OpenPortRange), arg0, arg1)
   577  }
   578  
   579  // OpenedPortRanges mocks base method.
   580  func (m *MockContext) OpenedPortRanges() network.GroupedPortRanges {
   581  	m.ctrl.T.Helper()
   582  	ret := m.ctrl.Call(m, "OpenedPortRanges")
   583  	ret0, _ := ret[0].(network.GroupedPortRanges)
   584  	return ret0
   585  }
   586  
   587  // OpenedPortRanges indicates an expected call of OpenedPortRanges.
   588  func (mr *MockContextMockRecorder) OpenedPortRanges() *gomock.Call {
   589  	mr.mock.ctrl.T.Helper()
   590  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "OpenedPortRanges", reflect.TypeOf((*MockContext)(nil).OpenedPortRanges))
   591  }
   592  
   593  // Prepare mocks base method.
   594  func (m *MockContext) Prepare() error {
   595  	m.ctrl.T.Helper()
   596  	ret := m.ctrl.Call(m, "Prepare")
   597  	ret0, _ := ret[0].(error)
   598  	return ret0
   599  }
   600  
   601  // Prepare indicates an expected call of Prepare.
   602  func (mr *MockContextMockRecorder) Prepare() *gomock.Call {
   603  	mr.mock.ctrl.T.Helper()
   604  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Prepare", reflect.TypeOf((*MockContext)(nil).Prepare))
   605  }
   606  
   607  // PrivateAddress mocks base method.
   608  func (m *MockContext) PrivateAddress() (string, error) {
   609  	m.ctrl.T.Helper()
   610  	ret := m.ctrl.Call(m, "PrivateAddress")
   611  	ret0, _ := ret[0].(string)
   612  	ret1, _ := ret[1].(error)
   613  	return ret0, ret1
   614  }
   615  
   616  // PrivateAddress indicates an expected call of PrivateAddress.
   617  func (mr *MockContextMockRecorder) PrivateAddress() *gomock.Call {
   618  	mr.mock.ctrl.T.Helper()
   619  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "PrivateAddress", reflect.TypeOf((*MockContext)(nil).PrivateAddress))
   620  }
   621  
   622  // PublicAddress mocks base method.
   623  func (m *MockContext) PublicAddress() (string, error) {
   624  	m.ctrl.T.Helper()
   625  	ret := m.ctrl.Call(m, "PublicAddress")
   626  	ret0, _ := ret[0].(string)
   627  	ret1, _ := ret[1].(error)
   628  	return ret0, ret1
   629  }
   630  
   631  // PublicAddress indicates an expected call of PublicAddress.
   632  func (mr *MockContextMockRecorder) PublicAddress() *gomock.Call {
   633  	mr.mock.ctrl.T.Helper()
   634  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "PublicAddress", reflect.TypeOf((*MockContext)(nil).PublicAddress))
   635  }
   636  
   637  // Relation mocks base method.
   638  func (m *MockContext) Relation(arg0 int) (jujuc.ContextRelation, error) {
   639  	m.ctrl.T.Helper()
   640  	ret := m.ctrl.Call(m, "Relation", arg0)
   641  	ret0, _ := ret[0].(jujuc.ContextRelation)
   642  	ret1, _ := ret[1].(error)
   643  	return ret0, ret1
   644  }
   645  
   646  // Relation indicates an expected call of Relation.
   647  func (mr *MockContextMockRecorder) Relation(arg0 any) *gomock.Call {
   648  	mr.mock.ctrl.T.Helper()
   649  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Relation", reflect.TypeOf((*MockContext)(nil).Relation), arg0)
   650  }
   651  
   652  // RelationIds mocks base method.
   653  func (m *MockContext) RelationIds() ([]int, error) {
   654  	m.ctrl.T.Helper()
   655  	ret := m.ctrl.Call(m, "RelationIds")
   656  	ret0, _ := ret[0].([]int)
   657  	ret1, _ := ret[1].(error)
   658  	return ret0, ret1
   659  }
   660  
   661  // RelationIds indicates an expected call of RelationIds.
   662  func (mr *MockContextMockRecorder) RelationIds() *gomock.Call {
   663  	mr.mock.ctrl.T.Helper()
   664  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RelationIds", reflect.TypeOf((*MockContext)(nil).RelationIds))
   665  }
   666  
   667  // RemoteApplicationName mocks base method.
   668  func (m *MockContext) RemoteApplicationName() (string, error) {
   669  	m.ctrl.T.Helper()
   670  	ret := m.ctrl.Call(m, "RemoteApplicationName")
   671  	ret0, _ := ret[0].(string)
   672  	ret1, _ := ret[1].(error)
   673  	return ret0, ret1
   674  }
   675  
   676  // RemoteApplicationName indicates an expected call of RemoteApplicationName.
   677  func (mr *MockContextMockRecorder) RemoteApplicationName() *gomock.Call {
   678  	mr.mock.ctrl.T.Helper()
   679  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RemoteApplicationName", reflect.TypeOf((*MockContext)(nil).RemoteApplicationName))
   680  }
   681  
   682  // RemoteUnitName mocks base method.
   683  func (m *MockContext) RemoteUnitName() (string, error) {
   684  	m.ctrl.T.Helper()
   685  	ret := m.ctrl.Call(m, "RemoteUnitName")
   686  	ret0, _ := ret[0].(string)
   687  	ret1, _ := ret[1].(error)
   688  	return ret0, ret1
   689  }
   690  
   691  // RemoteUnitName indicates an expected call of RemoteUnitName.
   692  func (mr *MockContextMockRecorder) RemoteUnitName() *gomock.Call {
   693  	mr.mock.ctrl.T.Helper()
   694  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RemoteUnitName", reflect.TypeOf((*MockContext)(nil).RemoteUnitName))
   695  }
   696  
   697  // RemoveSecret mocks base method.
   698  func (m *MockContext) RemoveSecret(arg0 *secrets.URI, arg1 *int) error {
   699  	m.ctrl.T.Helper()
   700  	ret := m.ctrl.Call(m, "RemoveSecret", arg0, arg1)
   701  	ret0, _ := ret[0].(error)
   702  	return ret0
   703  }
   704  
   705  // RemoveSecret indicates an expected call of RemoveSecret.
   706  func (mr *MockContextMockRecorder) RemoveSecret(arg0, arg1 any) *gomock.Call {
   707  	mr.mock.ctrl.T.Helper()
   708  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RemoveSecret", reflect.TypeOf((*MockContext)(nil).RemoveSecret), arg0, arg1)
   709  }
   710  
   711  // RequestReboot mocks base method.
   712  func (m *MockContext) RequestReboot(arg0 jujuc.RebootPriority) error {
   713  	m.ctrl.T.Helper()
   714  	ret := m.ctrl.Call(m, "RequestReboot", arg0)
   715  	ret0, _ := ret[0].(error)
   716  	return ret0
   717  }
   718  
   719  // RequestReboot indicates an expected call of RequestReboot.
   720  func (mr *MockContextMockRecorder) RequestReboot(arg0 any) *gomock.Call {
   721  	mr.mock.ctrl.T.Helper()
   722  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RequestReboot", reflect.TypeOf((*MockContext)(nil).RequestReboot), arg0)
   723  }
   724  
   725  // ResetExecutionSetUnitStatus mocks base method.
   726  func (m *MockContext) ResetExecutionSetUnitStatus() {
   727  	m.ctrl.T.Helper()
   728  	m.ctrl.Call(m, "ResetExecutionSetUnitStatus")
   729  }
   730  
   731  // ResetExecutionSetUnitStatus indicates an expected call of ResetExecutionSetUnitStatus.
   732  func (mr *MockContextMockRecorder) ResetExecutionSetUnitStatus() *gomock.Call {
   733  	mr.mock.ctrl.T.Helper()
   734  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ResetExecutionSetUnitStatus", reflect.TypeOf((*MockContext)(nil).ResetExecutionSetUnitStatus))
   735  }
   736  
   737  // RevokeSecret mocks base method.
   738  func (m *MockContext) RevokeSecret(arg0 *secrets.URI, arg1 *jujuc.SecretGrantRevokeArgs) error {
   739  	m.ctrl.T.Helper()
   740  	ret := m.ctrl.Call(m, "RevokeSecret", arg0, arg1)
   741  	ret0, _ := ret[0].(error)
   742  	return ret0
   743  }
   744  
   745  // RevokeSecret indicates an expected call of RevokeSecret.
   746  func (mr *MockContextMockRecorder) RevokeSecret(arg0, arg1 any) *gomock.Call {
   747  	mr.mock.ctrl.T.Helper()
   748  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RevokeSecret", reflect.TypeOf((*MockContext)(nil).RevokeSecret), arg0, arg1)
   749  }
   750  
   751  // SecretMetadata mocks base method.
   752  func (m *MockContext) SecretMetadata() (map[string]jujuc.SecretMetadata, error) {
   753  	m.ctrl.T.Helper()
   754  	ret := m.ctrl.Call(m, "SecretMetadata")
   755  	ret0, _ := ret[0].(map[string]jujuc.SecretMetadata)
   756  	ret1, _ := ret[1].(error)
   757  	return ret0, ret1
   758  }
   759  
   760  // SecretMetadata indicates an expected call of SecretMetadata.
   761  func (mr *MockContextMockRecorder) SecretMetadata() *gomock.Call {
   762  	mr.mock.ctrl.T.Helper()
   763  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SecretMetadata", reflect.TypeOf((*MockContext)(nil).SecretMetadata))
   764  }
   765  
   766  // SetActionFailed mocks base method.
   767  func (m *MockContext) SetActionFailed() error {
   768  	m.ctrl.T.Helper()
   769  	ret := m.ctrl.Call(m, "SetActionFailed")
   770  	ret0, _ := ret[0].(error)
   771  	return ret0
   772  }
   773  
   774  // SetActionFailed indicates an expected call of SetActionFailed.
   775  func (mr *MockContextMockRecorder) SetActionFailed() *gomock.Call {
   776  	mr.mock.ctrl.T.Helper()
   777  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetActionFailed", reflect.TypeOf((*MockContext)(nil).SetActionFailed))
   778  }
   779  
   780  // SetActionMessage mocks base method.
   781  func (m *MockContext) SetActionMessage(arg0 string) error {
   782  	m.ctrl.T.Helper()
   783  	ret := m.ctrl.Call(m, "SetActionMessage", arg0)
   784  	ret0, _ := ret[0].(error)
   785  	return ret0
   786  }
   787  
   788  // SetActionMessage indicates an expected call of SetActionMessage.
   789  func (mr *MockContextMockRecorder) SetActionMessage(arg0 any) *gomock.Call {
   790  	mr.mock.ctrl.T.Helper()
   791  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetActionMessage", reflect.TypeOf((*MockContext)(nil).SetActionMessage), arg0)
   792  }
   793  
   794  // SetApplicationStatus mocks base method.
   795  func (m *MockContext) SetApplicationStatus(arg0 jujuc.StatusInfo) error {
   796  	m.ctrl.T.Helper()
   797  	ret := m.ctrl.Call(m, "SetApplicationStatus", arg0)
   798  	ret0, _ := ret[0].(error)
   799  	return ret0
   800  }
   801  
   802  // SetApplicationStatus indicates an expected call of SetApplicationStatus.
   803  func (mr *MockContextMockRecorder) SetApplicationStatus(arg0 any) *gomock.Call {
   804  	mr.mock.ctrl.T.Helper()
   805  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetApplicationStatus", reflect.TypeOf((*MockContext)(nil).SetApplicationStatus), arg0)
   806  }
   807  
   808  // SetCharmStateValue mocks base method.
   809  func (m *MockContext) SetCharmStateValue(arg0, arg1 string) error {
   810  	m.ctrl.T.Helper()
   811  	ret := m.ctrl.Call(m, "SetCharmStateValue", arg0, arg1)
   812  	ret0, _ := ret[0].(error)
   813  	return ret0
   814  }
   815  
   816  // SetCharmStateValue indicates an expected call of SetCharmStateValue.
   817  func (mr *MockContextMockRecorder) SetCharmStateValue(arg0, arg1 any) *gomock.Call {
   818  	mr.mock.ctrl.T.Helper()
   819  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetCharmStateValue", reflect.TypeOf((*MockContext)(nil).SetCharmStateValue), arg0, arg1)
   820  }
   821  
   822  // SetPayloadStatus mocks base method.
   823  func (m *MockContext) SetPayloadStatus(arg0, arg1, arg2 string) error {
   824  	m.ctrl.T.Helper()
   825  	ret := m.ctrl.Call(m, "SetPayloadStatus", arg0, arg1, arg2)
   826  	ret0, _ := ret[0].(error)
   827  	return ret0
   828  }
   829  
   830  // SetPayloadStatus indicates an expected call of SetPayloadStatus.
   831  func (mr *MockContextMockRecorder) SetPayloadStatus(arg0, arg1, arg2 any) *gomock.Call {
   832  	mr.mock.ctrl.T.Helper()
   833  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetPayloadStatus", reflect.TypeOf((*MockContext)(nil).SetPayloadStatus), arg0, arg1, arg2)
   834  }
   835  
   836  // SetPodSpec mocks base method.
   837  func (m *MockContext) SetPodSpec(arg0 string) error {
   838  	m.ctrl.T.Helper()
   839  	ret := m.ctrl.Call(m, "SetPodSpec", arg0)
   840  	ret0, _ := ret[0].(error)
   841  	return ret0
   842  }
   843  
   844  // SetPodSpec indicates an expected call of SetPodSpec.
   845  func (mr *MockContextMockRecorder) SetPodSpec(arg0 any) *gomock.Call {
   846  	mr.mock.ctrl.T.Helper()
   847  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetPodSpec", reflect.TypeOf((*MockContext)(nil).SetPodSpec), arg0)
   848  }
   849  
   850  // SetProcess mocks base method.
   851  func (m *MockContext) SetProcess(arg0 context.HookProcess) {
   852  	m.ctrl.T.Helper()
   853  	m.ctrl.Call(m, "SetProcess", arg0)
   854  }
   855  
   856  // SetProcess indicates an expected call of SetProcess.
   857  func (mr *MockContextMockRecorder) SetProcess(arg0 any) *gomock.Call {
   858  	mr.mock.ctrl.T.Helper()
   859  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetProcess", reflect.TypeOf((*MockContext)(nil).SetProcess), arg0)
   860  }
   861  
   862  // SetRawK8sSpec mocks base method.
   863  func (m *MockContext) SetRawK8sSpec(arg0 string) error {
   864  	m.ctrl.T.Helper()
   865  	ret := m.ctrl.Call(m, "SetRawK8sSpec", arg0)
   866  	ret0, _ := ret[0].(error)
   867  	return ret0
   868  }
   869  
   870  // SetRawK8sSpec indicates an expected call of SetRawK8sSpec.
   871  func (mr *MockContextMockRecorder) SetRawK8sSpec(arg0 any) *gomock.Call {
   872  	mr.mock.ctrl.T.Helper()
   873  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetRawK8sSpec", reflect.TypeOf((*MockContext)(nil).SetRawK8sSpec), arg0)
   874  }
   875  
   876  // SetUnitStatus mocks base method.
   877  func (m *MockContext) SetUnitStatus(arg0 jujuc.StatusInfo) error {
   878  	m.ctrl.T.Helper()
   879  	ret := m.ctrl.Call(m, "SetUnitStatus", arg0)
   880  	ret0, _ := ret[0].(error)
   881  	return ret0
   882  }
   883  
   884  // SetUnitStatus indicates an expected call of SetUnitStatus.
   885  func (mr *MockContextMockRecorder) SetUnitStatus(arg0 any) *gomock.Call {
   886  	mr.mock.ctrl.T.Helper()
   887  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetUnitStatus", reflect.TypeOf((*MockContext)(nil).SetUnitStatus), arg0)
   888  }
   889  
   890  // SetUnitWorkloadVersion mocks base method.
   891  func (m *MockContext) SetUnitWorkloadVersion(arg0 string) error {
   892  	m.ctrl.T.Helper()
   893  	ret := m.ctrl.Call(m, "SetUnitWorkloadVersion", arg0)
   894  	ret0, _ := ret[0].(error)
   895  	return ret0
   896  }
   897  
   898  // SetUnitWorkloadVersion indicates an expected call of SetUnitWorkloadVersion.
   899  func (mr *MockContextMockRecorder) SetUnitWorkloadVersion(arg0 any) *gomock.Call {
   900  	mr.mock.ctrl.T.Helper()
   901  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetUnitWorkloadVersion", reflect.TypeOf((*MockContext)(nil).SetUnitWorkloadVersion), arg0)
   902  }
   903  
   904  // Storage mocks base method.
   905  func (m *MockContext) Storage(arg0 names.StorageTag) (jujuc.ContextStorageAttachment, error) {
   906  	m.ctrl.T.Helper()
   907  	ret := m.ctrl.Call(m, "Storage", arg0)
   908  	ret0, _ := ret[0].(jujuc.ContextStorageAttachment)
   909  	ret1, _ := ret[1].(error)
   910  	return ret0, ret1
   911  }
   912  
   913  // Storage indicates an expected call of Storage.
   914  func (mr *MockContextMockRecorder) Storage(arg0 any) *gomock.Call {
   915  	mr.mock.ctrl.T.Helper()
   916  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Storage", reflect.TypeOf((*MockContext)(nil).Storage), arg0)
   917  }
   918  
   919  // StorageTags mocks base method.
   920  func (m *MockContext) StorageTags() ([]names.StorageTag, error) {
   921  	m.ctrl.T.Helper()
   922  	ret := m.ctrl.Call(m, "StorageTags")
   923  	ret0, _ := ret[0].([]names.StorageTag)
   924  	ret1, _ := ret[1].(error)
   925  	return ret0, ret1
   926  }
   927  
   928  // StorageTags indicates an expected call of StorageTags.
   929  func (mr *MockContextMockRecorder) StorageTags() *gomock.Call {
   930  	mr.mock.ctrl.T.Helper()
   931  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "StorageTags", reflect.TypeOf((*MockContext)(nil).StorageTags))
   932  }
   933  
   934  // TrackPayload mocks base method.
   935  func (m *MockContext) TrackPayload(arg0 payloads.Payload) error {
   936  	m.ctrl.T.Helper()
   937  	ret := m.ctrl.Call(m, "TrackPayload", arg0)
   938  	ret0, _ := ret[0].(error)
   939  	return ret0
   940  }
   941  
   942  // TrackPayload indicates an expected call of TrackPayload.
   943  func (mr *MockContextMockRecorder) TrackPayload(arg0 any) *gomock.Call {
   944  	mr.mock.ctrl.T.Helper()
   945  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "TrackPayload", reflect.TypeOf((*MockContext)(nil).TrackPayload), arg0)
   946  }
   947  
   948  // UnitName mocks base method.
   949  func (m *MockContext) UnitName() string {
   950  	m.ctrl.T.Helper()
   951  	ret := m.ctrl.Call(m, "UnitName")
   952  	ret0, _ := ret[0].(string)
   953  	return ret0
   954  }
   955  
   956  // UnitName indicates an expected call of UnitName.
   957  func (mr *MockContextMockRecorder) UnitName() *gomock.Call {
   958  	mr.mock.ctrl.T.Helper()
   959  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UnitName", reflect.TypeOf((*MockContext)(nil).UnitName))
   960  }
   961  
   962  // UnitStatus mocks base method.
   963  func (m *MockContext) UnitStatus() (*jujuc.StatusInfo, error) {
   964  	m.ctrl.T.Helper()
   965  	ret := m.ctrl.Call(m, "UnitStatus")
   966  	ret0, _ := ret[0].(*jujuc.StatusInfo)
   967  	ret1, _ := ret[1].(error)
   968  	return ret0, ret1
   969  }
   970  
   971  // UnitStatus indicates an expected call of UnitStatus.
   972  func (mr *MockContextMockRecorder) UnitStatus() *gomock.Call {
   973  	mr.mock.ctrl.T.Helper()
   974  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UnitStatus", reflect.TypeOf((*MockContext)(nil).UnitStatus))
   975  }
   976  
   977  // UnitWorkloadVersion mocks base method.
   978  func (m *MockContext) UnitWorkloadVersion() (string, error) {
   979  	m.ctrl.T.Helper()
   980  	ret := m.ctrl.Call(m, "UnitWorkloadVersion")
   981  	ret0, _ := ret[0].(string)
   982  	ret1, _ := ret[1].(error)
   983  	return ret0, ret1
   984  }
   985  
   986  // UnitWorkloadVersion indicates an expected call of UnitWorkloadVersion.
   987  func (mr *MockContextMockRecorder) UnitWorkloadVersion() *gomock.Call {
   988  	mr.mock.ctrl.T.Helper()
   989  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UnitWorkloadVersion", reflect.TypeOf((*MockContext)(nil).UnitWorkloadVersion))
   990  }
   991  
   992  // UntrackPayload mocks base method.
   993  func (m *MockContext) UntrackPayload(arg0, arg1 string) error {
   994  	m.ctrl.T.Helper()
   995  	ret := m.ctrl.Call(m, "UntrackPayload", arg0, arg1)
   996  	ret0, _ := ret[0].(error)
   997  	return ret0
   998  }
   999  
  1000  // UntrackPayload indicates an expected call of UntrackPayload.
  1001  func (mr *MockContextMockRecorder) UntrackPayload(arg0, arg1 any) *gomock.Call {
  1002  	mr.mock.ctrl.T.Helper()
  1003  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UntrackPayload", reflect.TypeOf((*MockContext)(nil).UntrackPayload), arg0, arg1)
  1004  }
  1005  
  1006  // UpdateActionResults mocks base method.
  1007  func (m *MockContext) UpdateActionResults(arg0 []string, arg1 any) error {
  1008  	m.ctrl.T.Helper()
  1009  	ret := m.ctrl.Call(m, "UpdateActionResults", arg0, arg1)
  1010  	ret0, _ := ret[0].(error)
  1011  	return ret0
  1012  }
  1013  
  1014  // UpdateActionResults indicates an expected call of UpdateActionResults.
  1015  func (mr *MockContextMockRecorder) UpdateActionResults(arg0, arg1 any) *gomock.Call {
  1016  	mr.mock.ctrl.T.Helper()
  1017  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateActionResults", reflect.TypeOf((*MockContext)(nil).UpdateActionResults), arg0, arg1)
  1018  }
  1019  
  1020  // UpdateSecret mocks base method.
  1021  func (m *MockContext) UpdateSecret(arg0 *secrets.URI, arg1 *jujuc.SecretUpdateArgs) error {
  1022  	m.ctrl.T.Helper()
  1023  	ret := m.ctrl.Call(m, "UpdateSecret", arg0, arg1)
  1024  	ret0, _ := ret[0].(error)
  1025  	return ret0
  1026  }
  1027  
  1028  // UpdateSecret indicates an expected call of UpdateSecret.
  1029  func (mr *MockContextMockRecorder) UpdateSecret(arg0, arg1 any) *gomock.Call {
  1030  	mr.mock.ctrl.T.Helper()
  1031  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateSecret", reflect.TypeOf((*MockContext)(nil).UpdateSecret), arg0, arg1)
  1032  }
  1033  
  1034  // WorkloadName mocks base method.
  1035  func (m *MockContext) WorkloadName() (string, error) {
  1036  	m.ctrl.T.Helper()
  1037  	ret := m.ctrl.Call(m, "WorkloadName")
  1038  	ret0, _ := ret[0].(string)
  1039  	ret1, _ := ret[1].(error)
  1040  	return ret0, ret1
  1041  }
  1042  
  1043  // WorkloadName indicates an expected call of WorkloadName.
  1044  func (mr *MockContextMockRecorder) WorkloadName() *gomock.Call {
  1045  	mr.mock.ctrl.T.Helper()
  1046  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "WorkloadName", reflect.TypeOf((*MockContext)(nil).WorkloadName))
  1047  }
  1048  
  1049  // WriteLeaderSettings mocks base method.
  1050  func (m *MockContext) WriteLeaderSettings(arg0 map[string]string) error {
  1051  	m.ctrl.T.Helper()
  1052  	ret := m.ctrl.Call(m, "WriteLeaderSettings", arg0)
  1053  	ret0, _ := ret[0].(error)
  1054  	return ret0
  1055  }
  1056  
  1057  // WriteLeaderSettings indicates an expected call of WriteLeaderSettings.
  1058  func (mr *MockContextMockRecorder) WriteLeaderSettings(arg0 any) *gomock.Call {
  1059  	mr.mock.ctrl.T.Helper()
  1060  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "WriteLeaderSettings", reflect.TypeOf((*MockContext)(nil).WriteLeaderSettings), arg0)
  1061  }