custom-kube-scheduler/kubernetes-v1.15.4/hack/e2e_test.go

/*
Copyright 2017 The Kubernetes Authors.

Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at

    http://www.apache.org/licenses/LICENSE-2.0

Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
*/

// Unit tests for hack/e2e.go shim
package main

import (
	"errors"
	"flag"
	"fmt"
	"os"
	"path/filepath"
	"reflect"
	"testing"
	"time"
)

type FileInfo struct {
	when time.Time
}

func (f FileInfo) Name() string {
	return "fake-file"
}

func (f FileInfo) Size() int64 {
	return 0
}

func (f FileInfo) Mode() os.FileMode {
	return 0
}

func (f FileInfo) ModTime() time.Time {
	return f.when
}

func (f FileInfo) IsDir() bool {
	return false
}

func (f FileInfo) Sys() interface{} {
	return f
}

func TestParse(t *testing.T) {
	cases := []struct {
		args     []string
		expected flags
		err      error
	}{
		{
			[]string{"foo", "-v=false"},
			flags{getDefault, oldDefault, []string{"--verbose-commands=false"}},
			nil,
		},
		{
			[]string{"foo", "-v"},
			flags{getDefault, oldDefault, []string{"--verbose-commands=true"}},
			nil,
		},
		{
			[]string{"hello", "world"},
			flags{getDefault, oldDefault, []string{"--verbose-commands=true", "world"}},
			nil,
		},
		{
			[]string{"hello", "--", "--venus", "--karaoke"},
			flags{getDefault, oldDefault, []string{"--verbose-commands=true", "--venus", "--karaoke"}},
			nil,
		},
		{
			[]string{"hello", "--alpha", "--beta"},
			flags{getDefault, oldDefault, []string{"--alpha", "--beta"}},
			nil,
		},
		{
			[]string{"so", "--get", "--boo"},
			flags{true, oldDefault, []string{"--boo"}},
			nil,
		},
		{
			[]string{"omg", "--get=false", "--", "ugh"},
			flags{false, oldDefault, []string{"--verbose-commands=true", "ugh"}},
			nil,
		},
		{
			[]string{"wee", "--old=5m", "--get"},
			flags{true, 5 * time.Minute, []string{"--verbose-commands=true"}},
			nil,
		},
		{
			[]string{"fun", "--times", "--old=666s"},
			flags{getDefault, oldDefault, []string{"--times", "--old=666s"}},
			nil,
		},
		{
			[]string{"wut", "-h"},
			flags{},
			flag.ErrHelp,
		},
		{
			[]string{"wut", "--", "-h"},
			flags{getDefault, oldDefault, []string{"--verbose-commands=true", "-h"}},
			nil,
		},
	}

	for i, c := range cases {
		a, err := parse(c.args)
		if err != c.err {
			t.Errorf("%d: a=%v != e%v", i, err, c.err)
		}
		e := c.expected
		if a.get != e.get {
			t.Errorf("%d: a=%v != e=%v", i, a.get, e.get)
		}
		if a.old != e.old {
			t.Errorf("%d: a=%v != e=%v", i, a.old, e.old)
		}
		if !reflect.DeepEqual(a.args, e.args) {
			t.Errorf("%d: a=%v != e=%v", i, a.args, e.args)
		}
	}
}

func TestLook(t *testing.T) {
	lpf := errors.New("LookPath failed")
	sf := errors.New("Stat failed")
	lpnc := errors.New("LookPath should not be called")
	snc := errors.New("Stat should not be called")
	cases := []struct {
		stat     error
		lookPath error
		goPath   string
		expected error
	}{
		{ // GOPATH set, stat succeeds returns gopath
			stat:     nil,
			lookPath: lpnc,
			goPath:   "fake-gopath/",
			expected: nil,
		},
		{ // GOPATH set, stat fails, terms on lookpath
			stat:     sf,
			lookPath: lpf,
			goPath:   "fake-gopath/",
			expected: lpf,
		},
		{ // GOPATH unset, stat not called, terms on lookpath
			stat:     snc,
			lookPath: lpf,
			goPath:   "",
			expected: lpf,
		},
		{ // GOPATH unset, stat not called, lookpath matches
			stat:     snc,
			lookPath: nil,
			goPath:   "",
			expected: nil,
		},
	}

	for _, c := range cases {
		l := tester{
			func(string) (os.FileInfo, error) {
				return FileInfo{}, c.stat
			},
			func(string) (string, error) {
				if c.lookPath != nil {
					return "FAILED", c.lookPath
				}
				return "$PATH-FOUND", nil
			},
			c.goPath,
			nil, // wait
		}
		if _, err := l.lookKubetest(); err != c.expected {
			t.Errorf("err: %s != %s", err, c.expected)
		}
	}
}

func TestGetKubetest(t *testing.T) {
	gp := "fake-gopath"
	gpk := filepath.Join(gp, "bin", "kubetest")
	p := "PATH"
	pk := filepath.Join(p, "kubetest")
	eu := errors.New("upgrade failed")
	euVerbose := fmt.Errorf("go get -u k8s.io/test-infra/kubetest: %v", eu)
	et := errors.New("touch failed")
	cases := []struct {
		name string
		get  bool
		old  time.Duration

		stat     string        // stat succeeds on this file
		path     bool          // file exists on path
		age      time.Duration // age of mod time on file
		upgraded bool          // go get -u succeeds
		touched  bool          // touch succeeds
		goPath   string        // GOPATH var

		returnPath  string
		returnError error
	}{
		{name: "0: Pass when on GOPATH/bin",
			get: false,
			old: 0,

			stat:     gpk,
			path:     false,
			age:      100,
			upgraded: false,
			touched:  false,
			goPath:   gp,

			returnPath:  gpk,
			returnError: nil,
		},
		{name: "1: Pass when on PATH",
			get: false,
			old: 0,

			stat:     pk,
			path:     true,
			age:      100,
			upgraded: false,
			touched:  false,
			goPath:   gp,

			returnPath:  pk,
			returnError: nil,
		},
		{name: "2: Don't upgrade if on PATH and GOPATH is ''",
			get: true,
			old: 0,

			stat:     pk,
			path:     true,
			age:      100,
			upgraded: false,
			touched:  false,
			goPath:   "",

			returnPath:  pk,
			returnError: nil,
		},
		{name: "3: Don't upgrade on PATH when young.",
			get: true,
			old: time.Hour,

			stat:     pk,
			path:     true,
			age:      time.Second,
			upgraded: false,
			touched:  false,
			goPath:   gp,

			returnPath:  pk,
			returnError: nil,
		},
		{name: "4: Upgrade if old but GOPATH is set.",
			get: true,
			old: 0,

			stat:     pk,
			path:     true,
			age:      time.Second,
			upgraded: true,
			touched:  true,
			goPath:   gp,

			returnPath:  pk,
			returnError: nil,
		},
		{name: "5: Fail if upgrade fails",
			get: true,
			old: 0,

			stat:     pk,
			path:     true,
			age:      time.Second,
			upgraded: false,
			touched:  false,
			goPath:   gpk,

			returnPath:  "",
			returnError: euVerbose,
		},
		{name: "6: Fail if touch fails",
			get: true,
			old: 0,

			stat:     pk,
			path:     true,
			age:      time.Second,
			upgraded: true,
			touched:  false,
			goPath:   gpk,

			returnPath:  "",
			returnError: et,
		},
	}

	for i, c := range cases {
		didUp := false
		didTouch := false
		l := tester{
			stat: func(p string) (os.FileInfo, error) {
				// stat
				if p != c.stat {
					return nil, fmt.Errorf("Failed to find %s", p)
				}
				return FileInfo{time.Now().Add(c.age * -1)}, nil
			},
			lookPath: func(name string) (string, error) {
				if c.path {
					return filepath.Join(p, name), nil
				}
				return "", fmt.Errorf("Not on path: %s", name)
			},
			goPath: c.goPath,
			wait: func(cmd string, args ...string) error {
				if cmd == "go" {
					if c.upgraded {
						didUp = true
						return nil
					}
					return eu
				}
				if c.touched {
					didTouch = true
					return nil
				}
				return et
			},
		}
		p, e := l.getKubetest(c.get, c.old)
		if p != c.returnPath {
			t.Errorf("%d: test=%q returnPath %q != %q", i, c.name, p, c.returnPath)
		}
		if e == nil || c.returnError == nil {
			if e != c.returnError {
				t.Errorf("%d: test=%q returnError %q != %q", i, c.name, e, c.returnError)
			}
		} else {
			if e.Error() != c.returnError.Error() {
				t.Errorf("%d: test=%q returnError %q != %q", i, c.name, e, c.returnError)
			}
		}
		if didUp != c.upgraded {
			t.Errorf("%d: test=%q bad upgrade state of %v", i, c.name, didUp)
		}
		if didTouch != c.touched {
			t.Errorf("%d: test=%q bad touch state of %v", i, c.name, didTouch)
		}
	}
}