| /* |
| 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. |
| */ |
| |
| package remote |
| |
| import ( |
| "testing" |
| "time" |
| |
| "github.com/stretchr/testify/assert" |
| "github.com/stretchr/testify/require" |
| internalapi "k8s.io/kubernetes/pkg/kubelet/apis/cri" |
| apitest "k8s.io/kubernetes/pkg/kubelet/apis/cri/testing" |
| fakeremote "k8s.io/kubernetes/pkg/kubelet/remote/fake" |
| ) |
| |
| const ( |
| defaultConnectionTimeout = 15 * time.Second |
| ) |
| |
| // createAndStartFakeRemoteRuntime creates and starts fakeremote.RemoteRuntime. |
| // It returns the RemoteRuntime, endpoint on success. |
| // Users should call fakeRuntime.Stop() to cleanup the server. |
| func createAndStartFakeRemoteRuntime(t *testing.T) (*fakeremote.RemoteRuntime, string) { |
| endpoint, err := fakeremote.GenerateEndpoint() |
| require.NoError(t, err) |
| |
| fakeRuntime := fakeremote.NewFakeRemoteRuntime() |
| fakeRuntime.Start(endpoint) |
| |
| return fakeRuntime, endpoint |
| } |
| |
| func createRemoteRuntimeService(endpoint string, t *testing.T) internalapi.RuntimeService { |
| runtimeService, err := NewRemoteRuntimeService(endpoint, defaultConnectionTimeout) |
| require.NoError(t, err) |
| |
| return runtimeService |
| } |
| |
| func TestVersion(t *testing.T) { |
| fakeRuntime, endpoint := createAndStartFakeRemoteRuntime(t) |
| defer fakeRuntime.Stop() |
| |
| r := createRemoteRuntimeService(endpoint, t) |
| version, err := r.Version(apitest.FakeVersion) |
| assert.NoError(t, err) |
| assert.Equal(t, apitest.FakeVersion, version.Version) |
| assert.Equal(t, apitest.FakeRuntimeName, version.RuntimeName) |
| } |