internal

type FetchDataTest struct {
	// Name of the test.
	Name  string
	Check config.Check
	// Initialise the check before testing.
	Init bool
	// Func to run before running the check
	PreFetch func(t *testing.T)
	// Expected values after running the check.
	ExpectPasses   []string
	ExpectBreaches []result.Breach
	ExpectDataMap  map[string][]byte
}

type MockInsightsRemoteTestState struct {
	LastCallBody     string
	LastCallHeaders  map[string]string
	LastCallEndpoint string
	LastCallStatus   string
}

type RemediateTest struct {
	// Name of the test.
	Name     string
	Check    config.Check
	Breaches []result.Breach
	// Func to run before running Remediate
	PreRun func(t *testing.T)
	// Expected values after running Remediate.
	ExpectGeneratedCommand  string
	ExpectStatusFail        bool
	ExpectNoBreach          bool
	ExpectBreaches          []result.Breach
	ExpectRemediationStatus result.RemediationStatus
	ExpectNoRemediations    bool
	ExpectRemediations      []string
}

type RunCheckTest struct {
	// Name of the test.
	Name  string
	Check config.Check
	// Initialise the check before testing.
	Init bool
	// Sort the results after running the check.
	Sort bool
	// Func to run before running the check
	PreRun func(t *testing.T)
	// Expected values after running the check.
	ExpectStatus result.Status
	ExpectNoPass bool
	ExpectPasses []string
	ExpectNoFail bool
	ExpectFails  []result.Breach
}

type TestShellCommand struct {
	OutputterFunc func() ([]byte, error)
}