1+ #include " common/hardware_common.h"
12#include " gtest/gtest.h"
23#include " utils/hardware/cpu_info.h"
34#include " utils/hardware/gpu_info.h"
@@ -67,23 +68,32 @@ class GpuJsonTests : public ::testing::Test {
6768 // Set up a vector of GPUs for testing
6869 cortex::hw::NvidiaAddInfo nvidia_info{" 460.32.03" , " 6.1" };
6970
70- test_gpus.push_back ({.id = " 0" ,
71- .name = " NVIDIA GeForce GTX 1080" ,
72- .version = " 1.0" ,
73- .add_info = nvidia_info,
74- .free_vram = 4096 ,
75- .total_vram = 8192 ,
76- .uuid = " GPU-12345678" ,
77- .is_activated = true });
78-
79- test_gpus.push_back ({.id = " 1" ,
80- .name = " NVIDIA GeForce RTX 2080" ,
81- .version = " 1.1" ,
82- .add_info = nvidia_info,
83- .free_vram = 6144 ,
84- .total_vram = 8192 ,
85- .uuid = " GPU-87654321" ,
86- .is_activated = false });
71+ test_gpus.push_back (cortex::hw::GPU{
72+ /* .id = */ " 0" ,
73+ /* .device_id = */ 0 ,
74+ /* .name = */ " NVIDIA GeForce GTX 1080" ,
75+ /* .version = */ " 1.0" ,
76+ /* .add_info = */ nvidia_info,
77+ /* .free_vram = */ 4096 ,
78+ /* .total_vram = */ 8192 ,
79+ /* .uuid = */ " GPU-12345678" ,
80+ /* .is_activated = */ true ,
81+ /* .vendor = */ " " ,
82+ /* .gpu_type = */ cortex::hw::GpuType::kGpuTypeDiscrete });
83+
84+ test_gpus.push_back ({
85+ /* .id = */ " 1" ,
86+ /* .device_id = */ 0 ,
87+ /* .name = */ " NVIDIA GeForce RTX 2080" ,
88+ /* .version = */ " 1.1" ,
89+ /* .add_info = */ nvidia_info,
90+ /* .free_vram = */ 6144 ,
91+ /* .total_vram = */ 8192 ,
92+ /* .uuid = */ " GPU-87654321" ,
93+ /* .is_activated = */ false ,
94+ /* .vendor = */ " " ,
95+ /* .gpu_type = */ cortex::hw::GpuType::kGpuTypeDiscrete ,
96+ });
8797 }
8898
8999 std::vector<cortex::hw::GPU> test_gpus;
@@ -169,30 +179,30 @@ TEST_F(GpuJsonTests, FromJson_ValidJson_Success) {
169179}
170180
171181class OsJsonTests : public ::testing::Test {
172- protected:
173- cortex::hw::OS test_os;
174-
175- void SetUp () override {
176- test_os.name = " Ubuntu" ;
177- test_os.version = " 20.04" ;
178- test_os.arch = " x86_64" ;
179- }
182+ protected:
183+ cortex::hw::OS test_os;
184+
185+ void SetUp () override {
186+ test_os.name = " Ubuntu" ;
187+ test_os.version = " 20.04" ;
188+ test_os.arch = " x86_64" ;
189+ }
180190};
181191
182192TEST_F (OsJsonTests, ToJson_ValidOS_Success) {
183- Json::Value json_result = cortex::hw::ToJson (test_os);
193+ Json::Value json_result = cortex::hw::ToJson (test_os);
184194
185- EXPECT_EQ (json_result[" name" ].asString (), test_os.name );
186- EXPECT_EQ (json_result[" version" ].asString (), test_os.version );
195+ EXPECT_EQ (json_result[" name" ].asString (), test_os.name );
196+ EXPECT_EQ (json_result[" version" ].asString (), test_os.version );
187197}
188198
189199TEST_F (OsJsonTests, FromJson_ValidJson_Success) {
190- Json::Value json_input;
191- json_input[" name" ] = test_os.name ;
192- json_input[" version" ] = test_os.version ;
200+ Json::Value json_input;
201+ json_input[" name" ] = test_os.name ;
202+ json_input[" version" ] = test_os.version ;
193203
194- cortex::hw::OS os_result = cortex::hw::os::FromJson (json_input);
204+ cortex::hw::OS os_result = cortex::hw::os::FromJson (json_input);
195205
196- EXPECT_EQ (os_result.name , test_os.name );
197- EXPECT_EQ (os_result.version , test_os.version );
198- }
206+ EXPECT_EQ (os_result.name , test_os.name );
207+ EXPECT_EQ (os_result.version , test_os.version );
208+ }
0 commit comments