func setExecuterExpectations(mockExecuter *executers.MockCommandExecuter, t TestCase, cancelFlag task.CancelFlag, p *Plugin) { orchestrationDir := filepath.Join(orchestrationDirectory, fileutil.RemoveInvalidChars(t.Input.ID)) stdoutFilePath := filepath.Join(orchestrationDir, p.StdoutFileName) stderrFilePath := filepath.Join(orchestrationDir, p.StderrFileName) mockExecuter.On("Execute", mock.Anything, t.Input.WorkingDirectory, stdoutFilePath, stderrFilePath, cancelFlag, mock.Anything, mock.Anything, mock.Anything).Return( readerFromString(t.Output.Stdout), readerFromString(t.Output.Stderr), t.Output.ExitCode, t.ExecuterErrors) }
// 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) }