Updating unit tests

Signed-off-by: Nishant Totla <nishanttotla@gmail.com>
This commit is contained in:
Nishant Totla 2016-04-06 15:28:47 -07:00
parent f5e578188f
commit 40456d0926
No known key found for this signature in database
GPG Key ID: 7EA5781C9B3D0C19
4 changed files with 16 additions and 16 deletions

View File

@ -69,10 +69,10 @@ func TestAffinities(t *testing.T) {
func TestConsolidateResourceFields(t *testing.T) {
config := BuildContainerConfig(container.Config{}, container.HostConfig{Resources: container.Resources{Memory: 4242, MemorySwap: 4343, CPUShares: 4444, CpusetCpus: "1-2"}}, network.NetworkingConfig{})
assert.Equal(t, config.Memory, int64(4242))
assert.Equal(t, config.MemorySwap, int64(4343))
assert.Equal(t, config.CPUShares, int64(4444))
assert.Equal(t, config.CpusetCpus, "1-2")
assert.Equal(t, config.HostConfig.Memory, int64(4242))
assert.Equal(t, config.HostConfig.MemorySwap, int64(4343))
assert.Equal(t, config.HostConfig.CPUShares, int64(4444))
assert.Equal(t, config.HostConfig.CpusetCpus, "1-2")
}
func TestAddAffinity(t *testing.T) {

View File

@ -314,8 +314,8 @@ func TestCreateContainer(t *testing.T) {
assert.NoError(t, engine.ConnectWithClient(client, apiClient))
assert.True(t, engine.isConnected())
mockConfig := config
mockConfig.CPUShares = int64(math.Ceil(float64(config.CPUShares*1024) / float64(mockInfo.NCPU)))
mockConfig := *config
mockConfig.HostConfig.CPUShares = int64(math.Ceil(float64(config.HostConfig.CPUShares*1024) / float64(mockInfo.NCPU)))
// Everything is ok
name := "test1"
@ -334,7 +334,7 @@ func TestCreateContainer(t *testing.T) {
// Image not found, pullImage == false
name = "test2"
mockConfig.CPUShares = int64(math.Ceil(float64(config.CPUShares*1024) / float64(mockInfo.NCPU)))
mockConfig.HostConfig.CPUShares = int64(math.Ceil(float64(config.HostConfig.CPUShares*1024) / float64(mockInfo.NCPU)))
// FIXMEENGINEAPI : below should return an engine-api error, or something custom
apiClient.On("ContainerCreate", mock.Anything, &mockConfig.Config, &mockConfig.HostConfig, &mockConfig.NetworkingConfig, name).Return(types.ContainerCreateResponse{}, dockerclient.ErrImageNotFound).Once()
container, err = engine.Create(config, name, false, auth)
@ -345,7 +345,7 @@ func TestCreateContainer(t *testing.T) {
name = "test3"
id = "id3"
apiClient.On("ImageList", mock.Anything, mock.AnythingOfType("ImageListOptions")).Return([]types.Image{}, nil).Once()
mockConfig.CPUShares = int64(math.Ceil(float64(config.CPUShares*1024) / float64(mockInfo.NCPU)))
mockConfig.HostConfig.CPUShares = int64(math.Ceil(float64(config.HostConfig.CPUShares*1024) / float64(mockInfo.NCPU)))
apiClient.On("ImagePull", mock.Anything, types.ImagePullOptions{ImageID: config.Image, Tag: "latest"}, mock.Anything).Return(readCloser, nil).Once()
// FIXMEENGINEAPI : below should return an engine-api error, or something custom
apiClient.On("ContainerCreate", mock.Anything, &mockConfig.Config, &mockConfig.HostConfig, &mockConfig.NetworkingConfig, name).Return(types.ContainerCreateResponse{}, dockerclient.ErrImageNotFound).Once()

View File

@ -170,22 +170,22 @@ func TestPlaceContainerOvercommit(t *testing.T) {
config := createConfig(0, 0)
// Below limit should still work.
config.Memory = 90 * 1024 * 1024 * 1024
config.HostConfig.Memory = 90 * 1024 * 1024 * 1024
node := selectTopNode(t, s, config, nodes)
assert.Equal(t, node, nodes[0])
// At memory limit should still work.
config.Memory = 100 * 1024 * 1024 * 1024
config.HostConfig.Memory = 100 * 1024 * 1024 * 1024
node = selectTopNode(t, s, config, nodes)
assert.Equal(t, node, nodes[0])
// Up to 105% it should still work.
config.Memory = 105 * 1024 * 1024 * 1024
config.HostConfig.Memory = 105 * 1024 * 1024 * 1024
node = selectTopNode(t, s, config, nodes)
assert.Equal(t, node, nodes[0])
// Above it should return an error.
config.Memory = 106 * 1024 * 1024 * 1024
config.HostConfig.Memory = 106 * 1024 * 1024 * 1024
_, err = s.RankAndSort(config, nodes)
assert.Error(t, err)
}

View File

@ -164,22 +164,22 @@ func TestSpreadPlaceContainerOvercommit(t *testing.T) {
config := createConfig(0, 0)
// Below limit should still work.
config.Memory = 90 * 1024 * 1024 * 1024
config.HostConfig.Memory = 90 * 1024 * 1024 * 1024
node := selectTopNode(t, s, config, nodes)
assert.Equal(t, node, nodes[0])
// At memory limit should still work.
config.Memory = 100 * 1024 * 1024 * 1024
config.HostConfig.Memory = 100 * 1024 * 1024 * 1024
node = selectTopNode(t, s, config, nodes)
assert.Equal(t, node, nodes[0])
// Up to 105% it should still work.
config.Memory = 105 * 1024 * 1024 * 1024
config.HostConfig.Memory = 105 * 1024 * 1024 * 1024
node = selectTopNode(t, s, config, nodes)
assert.Equal(t, node, nodes[0])
// Above it should return an error.
config.Memory = 106 * 1024 * 1024 * 1024
config.HostConfig.Memory = 106 * 1024 * 1024 * 1024
_, err := s.RankAndSort(config, nodes)
assert.Error(t, err)
}