diff --git a/cmd/kube-proxy/app/server_test.go b/cmd/kube-proxy/app/server_test.go index b2fc19a036d..f9c70a4dce1 100644 --- a/cmd/kube-proxy/app/server_test.go +++ b/cmd/kube-proxy/app/server_test.go @@ -52,6 +52,15 @@ func (fake *fakeIPTablesVersioner) GetVersion() (string, error) { return fake.version, fake.err } +type fakeIPSetVersioner struct { + version string // what to return + err error // what to return +} + +func (fake *fakeIPSetVersioner) GetVersion() (string, error) { + return fake.version, fake.err +} + type fakeKernelCompatTester struct { ok bool } @@ -72,8 +81,10 @@ func Test_getProxyMode(t *testing.T) { annotationKey string annotationVal string iptablesVersion string + ipsetVersion string kernelCompat bool iptablesError error + ipsetError error expected string }{ { // flag says userspace @@ -128,7 +139,8 @@ func Test_getProxyMode(t *testing.T) { for i, c := range cases { versioner := &fakeIPTablesVersioner{c.iptablesVersion, c.iptablesError} kcompater := &fakeKernelCompatTester{c.kernelCompat} - r := getProxyMode(c.flag, versioner, kcompater) + ipsetver := &fakeIPSetVersioner{c.ipsetVersion, c.ipsetError} + r := getProxyMode(c.flag, versioner, ipsetver, kcompater) if r != c.expected { t.Errorf("Case[%d] Expected %q, got %q", i, c.expected, r) }