nagios_test.go

package main

import (
	"path/filepath"
	"reflect"
	"strings"
	"testing"
)

func TestGetNagiosTasks(t *testing.T) {
	tasks := make(chan Task, 1)
	runner := &FakeRunner{}
	GetNagiosTasks(tasks, runner, nil)
	task := <-tasks

	want := "Nagios pod could not be found"
	if err := task(); err == nil || !strings.Contains(err.Error(), want) {
		t.Errorf("task() = %q, want substring of %q", err, want)
	}
}

func TestGetNagiosStatusData(t *testing.T) {
	tests := []struct {
		project string
		pod     string
		calls   []RunCall
	}{
		{
			project: "test-project",
			pod:     "pod",
			calls: []RunCall{
				{
					[]string{"oc", "-n", "test-project", "exec", "pod", "--", "cat", "/var/log/nagios/status.dat"},
					filepath.Join("projects", "test-project", "nagios", "pod_status.dat"),
				},
			},
		},
	}
	for i, tt := range tests {
		runner := &FakeRunner{}
		task := GetNagiosStatusData(runner, tt.project, tt.pod)
		if err := task(); err != nil {
			t.Errorf("test %d: task() = %v, want %v", i, err, nil)
		}
		if !reflect.DeepEqual(runner.Calls, tt.calls) {
			t.Errorf("test %d: runner.Calls = %q, want %q", i, runner.Calls, tt.calls)
		}
	}
}
func TestGetNagiosHistoricalData(t *testing.T) {
	tests := []struct {
		project string
		pod     string
		calls   []RunCall
	}{
		{
			project: "test-project",
			pod:     "pod",
			calls: []RunCall{
				{
					[]string{"oc", "-n", "test-project", "exec", "pod", "--", "tar", "-c", "-C", "/var/log/nagios", "archives"},
					filepath.Join("projects", "test-project", "nagios", "pod_history.tar"),
				},
			},
		},
	}
	for i, tt := range tests {
		runner := &FakeRunner{}
		task := GetNagiosHistoricalData(runner, tt.project, tt.pod)
		if err := task(); err != nil {
			t.Errorf("test %d: task() = %v, want %v", i, err, nil)
		}
		if !reflect.DeepEqual(runner.Calls, tt.calls) {
			t.Errorf("test %d: runner.Calls = %q, want %q", i, runner.Calls, tt.calls)
		}
	}
}