// testExecution sets up boiler plate mocked objects then delegates to a more
// specific tester, then asserts general expectations on the mocked objects.
// It is the responsibility of the inner tester to set up expectations
// and assert specific result conditions.
func testExecution(t *testing.T, commandtester CommandTester) {
// create mocked objects
mockCancelFlag := new(task.MockCancelFlag)
mockExecuter := new(executers.MockCommandExecuter)
mockS3Uploader := new(pluginutil.MockDefaultPlugin)
// create plugin
p := new(Plugin)
p.StdoutFileName = "stdout"
p.StderrFileName = "stderr"
p.MaxStdoutLength = 1000
p.MaxStderrLength = 1000
p.OutputTruncatedSuffix = "-more-"
p.UploadToS3Sync = true
p.ExecuteCommand = pluginutil.CommandExecuter(mockExecuter.Execute)
p.ExecuteUploadOutputToS3Bucket = pluginutil.UploadOutputToS3BucketExecuter(mockS3Uploader.UploadOutputToS3Bucket)
// run inner command tester
commandtester(p, mockCancelFlag, mockExecuter, mockS3Uploader)
// assert that the expectations were met
mockExecuter.AssertExpectations(t)
mockCancelFlag.AssertExpectations(t)
mockS3Uploader.AssertExpectations(t)
}
func setS3UploaderExpectations(mockS3Uploader *pluginutil.MockDefaultPlugin, t TestCase, p *Plugin) {
var emptyArray []string
orchestrationDir := filepath.Join(orchestrationDirectory, fileutil.RemoveInvalidChars(t.Input.ID))
mockS3Uploader.On("UploadOutputToS3Bucket", mock.Anything, t.Input.ID, orchestrationDir, s3BucketName, s3KeyPrefix, false, "", t.Output.Stdout, t.Output.Stderr).Return(emptyArray)
}