mirror of
https://github.com/aljazceru/kata-containers.git
synced 2025-12-17 22:34:25 +01:00
fe99e7e165
Add host memory size(Total/Free/Avaiable) and CPU cores in host info for `kata-runtime kata-env`. Fixes: #405 Signed-off-by: bin liu <bin@hyper.sh>
57 lines
1.4 KiB
Go
57 lines
1.4 KiB
Go
// Copyright (c) 2018 Intel Corporation
|
|
//
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
//
|
|
|
|
// +build arm64 ppc64le
|
|
|
|
package main
|
|
|
|
import (
|
|
"io/ioutil"
|
|
"os"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func testEnvGetEnvInfoSetsCPUTypeGeneric(t *testing.T) {
|
|
assert := assert.New(t)
|
|
|
|
tmpdir, err := ioutil.TempDir("", "")
|
|
assert.NoError(err)
|
|
defer os.RemoveAll(tmpdir)
|
|
|
|
savedArchRequiredCPUFlags := archRequiredCPUFlags
|
|
savedArchRequiredCPUAttribs := archRequiredCPUAttribs
|
|
savedArchRequiredKernelModules := archRequiredKernelModules
|
|
|
|
defer func() {
|
|
archRequiredCPUFlags = savedArchRequiredCPUFlags
|
|
archRequiredCPUAttribs = savedArchRequiredCPUAttribs
|
|
archRequiredKernelModules = savedArchRequiredKernelModules
|
|
}()
|
|
|
|
assert.Empty(archRequiredCPUFlags)
|
|
assert.Empty(archRequiredCPUAttribs)
|
|
assert.NotEmpty(archRequiredKernelModules)
|
|
|
|
configFile, config, err := makeRuntimeConfig(tmpdir)
|
|
assert.NoError(err)
|
|
|
|
expectedEnv, err := getExpectedSettings(config, tmpdir, configFile)
|
|
assert.NoError(err)
|
|
|
|
env, err := getEnvInfo(configFile, config)
|
|
assert.NoError(err)
|
|
|
|
// Free/Available are changing
|
|
expectedEnv.Host.Memory = env.Host.Memory
|
|
|
|
assert.Equal(expectedEnv, env)
|
|
|
|
assert.Equal(archRequiredCPUFlags, savedArchRequiredCPUFlags)
|
|
assert.Equal(archRequiredCPUAttribs, savedArchRequiredCPUAttribs)
|
|
assert.Equal(archRequiredKernelModules, savedArchRequiredKernelModules)
|
|
}
|