github.com/caos/orbos@v1.5.14-0.20221103111702-e6cd0cea7ad4/internal/operator/boom/application/mock/applications.go (about)

     1  package application
     2  
     3  import (
     4  	"testing"
     5  
     6  	"github.com/caos/orbos/internal/operator/boom/api/v1beta1"
     7  	"github.com/caos/orbos/internal/operator/boom/name"
     8  	"github.com/golang/mock/gomock"
     9  )
    10  
    11  const (
    12  	Name      name.Application = "mockapplication"
    13  	Namespace                  = "mocknamespace"
    14  )
    15  
    16  type TestApplication struct {
    17  	app *MockApplication
    18  }
    19  
    20  func NewTestApplication(t *testing.T) *TestApplication {
    21  
    22  	mockApp := NewMockApplication(gomock.NewController(t))
    23  	mockApp.EXPECT().GetName().AnyTimes().DoAndReturn(Name)
    24  	app := &TestApplication{
    25  		app: mockApp,
    26  	}
    27  
    28  	return app
    29  }
    30  
    31  func (a *TestApplication) Application() *MockApplication {
    32  	return a.app
    33  }
    34  
    35  func (a *TestApplication) SetDeploy(spec *v1beta1.ToolsetSpec, b bool) *TestApplication {
    36  	a.app.EXPECT().Deploy(spec).AnyTimes().Return(b)
    37  	return a
    38  }
    39  
    40  type TestYAMLApplication struct {
    41  	app *MockYAMLApplication
    42  }
    43  
    44  func NewTestYAMLApplication(t *testing.T) *TestYAMLApplication {
    45  
    46  	mockApp := NewMockYAMLApplication(gomock.NewController(t))
    47  	mockApp.EXPECT().GetName().AnyTimes().Return(Name)
    48  	app := &TestYAMLApplication{
    49  		app: mockApp,
    50  	}
    51  
    52  	return app
    53  }
    54  
    55  func (a *TestYAMLApplication) Application() *MockYAMLApplication {
    56  	return a.app
    57  }
    58  
    59  func (a *TestYAMLApplication) SetDeploy(spec *v1beta1.ToolsetSpec, b bool) *TestYAMLApplication {
    60  	a.app.EXPECT().Deploy(spec).AnyTimes().Return(b)
    61  	return a
    62  }
    63  
    64  func (a *TestYAMLApplication) SetGetYaml(spec *v1beta1.ToolsetSpec, yaml string) *TestYAMLApplication {
    65  	a.app.EXPECT().GetYaml(gomock.Any(), spec).AnyTimes().Return(yaml)
    66  	return a
    67  }