main_test.go

func TestIsVersionOK(t *testing.T) {
        expectedMajor := "1"
        expectedMinor := "9"
        k8s := newTestSimpleK8s()
        k8s.clientset.Discovery().(*discoveryfake.FakeDiscovery).FakedServerVersion = &version.Info{
                Major: expectedMajor,
                Minor: expectedMinor,
        }
        v, err := k8s.isVersion(expectedMajor, expectedMinor)
        if err != nil {
                t.Fatal("isVersion should not raise an error")
        }
        if v != true {
                t.Fatal("isVersion should return true")
        }
}

func TestIsVersionErrorMajor(t *testing.T) {
        expectedMajor := "1"
        expectedMinor := "9"
        k8s := newTestSimpleK8s()
        k8s.clientset.Discovery().(*discoveryfake.FakeDiscovery).FakedServerVersion = &version.Info{
                Major: "wrong",
                Minor: "wrong",
        }
        _, err := k8s.isVersion(expectedMajor, expectedMinor)
        if err == nil {
                t.Fatal("isVersion should raise an error")
        }
        expected := "Major version does not match"
        if err.Error() != expected {
                t.Fatal("Raised error should be: " + expected)
        }
}

func TestIsVersionErrorMinor(t *testing.T) {
        expectedMajor := "1"
        expectedMinor := "9"
        k8s := newTestSimpleK8s()
        k8s.clientset.Discovery().(*discoveryfake.FakeDiscovery).FakedServerVersion = &version.Info{
                Major: expectedMajor,
                Minor: "wrong",
        }
        _, err := k8s.isVersion(expectedMajor, expectedMinor)
        if err == nil {
                t.Fatal("isVersion should raise an error")
        }
        expected := "Minor version does not match"
        if err.Error() != expected {
                t.Fatal("Raised error should be: " + expected)
        }
}