Example #1
0
func TestExecutorOnlyFailure(t *testing.T) {
	var statusCodesToTest = getIntsInRange(100, 599)
	testServer := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		w.WriteHeader(http.StatusOK)
	}))
	defer testServer.Close()

	for _, statusCodeTest := range statusCodesToTest {
		writer := new(bytes.Buffer)
		formatter := &testResultFormatter{}
		exec := NewExecutor(false, true, false, formatter, writer, 1)
		target, _ := config.NewTarget("abc", testServer.URL, config.MethodGet, statusCodeTest)
		targets := []*config.Target{target}

		actualExitCode := exec.Execute(targets)
		actualResult := formatter.lastResult
		actualOutput := writer.String()

		if statusCodeTest == 200 {
			assert.Empty(t, actualOutput)
			assert.Nil(t, actualResult)
			assert.Equal(t, 0, actualExitCode)
		} else {
			assert.NotEmpty(t, actualOutput)
			assert.NotNil(t, actualResult)
			assert.Equal(t, 1, actualExitCode)
			assert.False(t, actualResult.Success)
		}
	}
}
Example #2
0
func TestExecutorOnlyStatusChangeAndOnlyFailure(t *testing.T) {
	statusToReturn := http.StatusOK
	testServer := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		w.WriteHeader(statusToReturn)
	}))
	defer testServer.Close()

	writer := new(bytes.Buffer)
	formatter := &testResultFormatter{}
	exec := NewExecutor(false, true, true, formatter, writer, 1)
	target, _ := config.NewTarget("abc", testServer.URL, config.MethodGet, 200)
	targets := []*config.Target{target}

	// first time - success
	actualExitCode := exec.Execute(targets)
	actualResult := formatter.lastResult
	actualOutput := writer.String()

	assert.Equal(t, 0, actualExitCode)
	assert.Empty(t, actualOutput)
	assert.Nil(t, actualResult)

	// -- reset --
	formatter.lastResult = nil
	writer.Reset()

	// second time - success - no change
	actualExitCode = exec.Execute(targets)
	actualResult = formatter.lastResult
	actualOutput = writer.String()

	assert.Equal(t, 0, actualExitCode)
	assert.Empty(t, actualOutput)
	assert.Nil(t, actualResult)

	// -- reset --
	formatter.lastResult = nil
	writer.Reset()

	// third time - failure - change
	statusToReturn = http.StatusBadRequest
	actualExitCode = exec.Execute(targets)
	actualResult = formatter.lastResult
	actualOutput = writer.String()

	assert.Equal(t, 1, actualExitCode)
	assert.NotEmpty(t, actualOutput)
	assert.NotNil(t, actualResult)
	assert.False(t, actualResult.Success)

	// -- reset --
	formatter.lastResult = nil
	writer.Reset()

	// fourth time - failure - no change
	actualExitCode = exec.Execute(targets)
	actualResult = formatter.lastResult
	actualOutput = writer.String()

	assert.Equal(t, 0, actualExitCode)
	assert.Empty(t, actualOutput)
	assert.Nil(t, actualResult)

	// -- reset --
	formatter.lastResult = nil
	writer.Reset()

	// fifth time - success - change
	statusToReturn = http.StatusOK
	actualExitCode = exec.Execute(targets)
	actualResult = formatter.lastResult
	actualOutput = writer.String()

	assert.Equal(t, 0, actualExitCode)
	assert.Empty(t, actualOutput)
	assert.Nil(t, actualResult)
}