From 8bbef2865455754425a84b86680a89bff8aa7691 Mon Sep 17 00:00:00 2001 From: gcontini <1121667+gcontini@users.noreply.github.com> Date: 周六, 19 10月 2019 08:50:45 +0800 Subject: [PATCH] issue #64 , issue #56 (part) --- test/library/Os_Linux_test.cpp | 79 +++++++++++++++++++++++++++++---------- 1 files changed, 59 insertions(+), 20 deletions(-) diff --git a/test/library/Os_Linux_test.cpp b/test/library/Os_Linux_test.cpp index 1d5c9c6..d916375 100644 --- a/test/library/Os_Linux_test.cpp +++ b/test/library/Os_Linux_test.cpp @@ -1,6 +1,6 @@ #define BOOST_TEST_MODULE os_linux_test //#define BOOST_TEST_MAIN -#define BOOST_TEST_DYN_LINK +//#define BOOST_TEST_DYN_LINK #include <boost/test/unit_test.hpp> #include "../../src/library/os/os.h" #include <build_properties.h> @@ -10,29 +10,39 @@ using namespace std; BOOST_AUTO_TEST_CASE( read_disk_id ) { - DiskInfo * diskInfos = NULL; - size_t disk_info_size = 0; - FUNCTION_RETURN result = getDiskInfos(NULL, &disk_info_size); - BOOST_CHECK_EQUAL(result, OK); - BOOST_CHECK_GT(disk_info_size, 0); - diskInfos = (DiskInfo*) malloc(sizeof(DiskInfo) * disk_info_size); - result = getDiskInfos(diskInfos, &disk_info_size); - BOOST_CHECK_EQUAL(result, OK); - BOOST_CHECK_GT(strlen(diskInfos[0].device), 0); - BOOST_CHECK_GT(strlen(diskInfos[0].label), 0); - BOOST_CHECK_GT(diskInfos[0].disk_sn[0], 0); + VIRTUALIZATION virt = getVirtualization(); + if(virt == NONE || virt == VM) { + DiskInfo * diskInfos = NULL; + size_t disk_info_size = 0; + FUNCTION_RETURN result = getDiskInfos(NULL, &disk_info_size); + BOOST_CHECK_EQUAL(result, FUNC_RET_OK); + BOOST_CHECK_GT(disk_info_size, 0); + diskInfos = (DiskInfo*) malloc(sizeof(DiskInfo) * disk_info_size); + result = getDiskInfos(diskInfos, &disk_info_size); + BOOST_CHECK_EQUAL(result, FUNC_RET_OK); + BOOST_CHECK_GT(strlen(diskInfos[0].device), 0); + BOOST_CHECK_GT(strlen(diskInfos[0].label), 0); + BOOST_CHECK_GT(diskInfos[0].disk_sn[0], 0); + free(diskInfos); + } else if(virt == CONTAINER){ + //docker or lxc diskInfo is not meaningful + DiskInfo * diskInfos = NULL; + size_t disk_info_size = 0; + FUNCTION_RETURN result = getDiskInfos(NULL, &disk_info_size); + BOOST_CHECK_EQUAL(result, FUNC_RET_NOT_AVAIL); + } } -/* + BOOST_AUTO_TEST_CASE( read_network_adapters ) { - AdapterInfo * adapter_info = NULL; + OsAdapterInfo * adapter_info = NULL; size_t adapter_info_size = 0; FUNCTION_RETURN result = getAdapterInfos(NULL, &adapter_info_size); - BOOST_CHECK_EQUAL(result, OK); + BOOST_CHECK_EQUAL(result, FUNC_RET_OK); BOOST_CHECK_GT(adapter_info_size, 0); - adapter_info = (AdapterInfo*) malloc( - sizeof(AdapterInfo) * adapter_info_size); + adapter_info = (OsAdapterInfo*) malloc( + sizeof(OsAdapterInfo) * adapter_info_size); result = getAdapterInfos(adapter_info, &adapter_info_size); - BOOST_CHECK_EQUAL(result, OK); + BOOST_CHECK_EQUAL(result, FUNC_RET_OK); for (size_t i = 0; i < adapter_info_size; i++) { cout << "Interface found: " << string(adapter_info[i].description) << endl; @@ -42,8 +52,37 @@ if (string(adapter_info[i].description) == "lo") { BOOST_CHECK_NE(adapter_info[i].ipv4_address[0], 0); } else { - BOOST_CHECK_NE(adapter_info[i].mac_address[0], 0); + bool mac_is_0 = true; + for(int j=0;j<6;j++){ + mac_is_0 = mac_is_0 && (adapter_info[i].mac_address[j]==0); + } + BOOST_CHECK_MESSAGE( !mac_is_0, + "Mac address for interface " << adapter_info[i].description << " is 0"); } } + free(adapter_info); +} -}*/ +BOOST_AUTO_TEST_CASE( get_cpuid ) { + BOOST_CHECK_EQUAL(1, 1); +} + +//To test if virtualization is detected correctly define an env variable VIRT_ENV +//otherwise the test is skipped +BOOST_AUTO_TEST_CASE( test_virtualization ) { + const char *env = getenv("VIRT_ENV"); + if (env != NULL) { + if (strcmp(env, "CONTAINER") == 0) { + VIRTUALIZATION virt = getVirtualization(); + BOOST_CHECK_EQUAL(virt, CONTAINER); + } else if (strcmp(env, "VM") == 0) { + BOOST_FAIL("check for vm not implemented"); + } else if (strcmp(env, "NONE") == 0) { + VIRTUALIZATION virt = getVirtualization(); + BOOST_CHECK_EQUAL(virt, NONE); + } else { + BOOST_FAIL(string("value ") + env + " not supported: VM,CONTAINER,NONE"); + } + } +} + -- Gitblit v1.9.1