github.com/juju/juju@v0.0.0-20240327075706-a90865de2538/worker/caasupgrader/mock_test.go (about)

     1  // Copyright 2019 Canonical Ltd.
     2  // Licensed under the AGPLv3, see LICENCE file for details.
     3  
     4  package caasupgrader_test
     5  
     6  import (
     7  	"github.com/juju/testing"
     8  	"github.com/juju/version/v2"
     9  
    10  	"github.com/juju/juju/core/watcher"
    11  )
    12  
    13  type mockUpgraderClient struct {
    14  	testing.Stub
    15  
    16  	desired version.Number
    17  	actual  version.Binary
    18  	watcher watcher.NotifyWatcher
    19  }
    20  
    21  func (m *mockUpgraderClient) DesiredVersion(tag string) (version.Number, error) {
    22  	m.Stub.AddCall("DesiredVersion", tag)
    23  	return m.desired, nil
    24  }
    25  
    26  func (m *mockUpgraderClient) SetVersion(tag string, v version.Binary) error {
    27  	m.Stub.AddCall("SetVersion", tag, v)
    28  	m.actual = v
    29  	return nil
    30  }
    31  
    32  func (m *mockUpgraderClient) WatchAPIVersion(agentTag string) (watcher.NotifyWatcher, error) {
    33  	return m.watcher, nil
    34  }
    35  
    36  type mockOperatorUpgrader struct {
    37  	testing.Stub
    38  }
    39  
    40  func (m *mockOperatorUpgrader) Upgrade(appName string, vers version.Number) error {
    41  	m.AddCall("Upgrade", appName, vers)
    42  	return nil
    43  }