Add CPU and RAM information to metrics
Bug: b/365537477
Test: manual, TH
Change-Id: Ia76caaff2324f29b1e806915cab77bbb26df4ecb
diff --git a/ui/metrics/hostinfo_linux_test.go b/ui/metrics/hostinfo_linux_test.go
new file mode 100644
index 0000000..c44e453
--- /dev/null
+++ b/ui/metrics/hostinfo_linux_test.go
@@ -0,0 +1,118 @@
+// Copyright 2024 Google Inc. All rights reserved.
+//
+// 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 metrics
+
+// This file contain code to extract host information on linux from
+// /proc/cpuinfo and /proc/meminfo relevant to machine performance
+
+import (
+ "reflect"
+ "testing"
+
+ "android/soong/finder/fs"
+)
+
+func TestNewCpuInfo(t *testing.T) {
+ fs := fs.NewMockFs(nil)
+
+ if err := fs.MkDirs("/proc"); err != nil {
+ t.Fatalf("failed to create /proc dir: %v", err)
+ }
+ cpuFileName := "/proc/cpuinfo"
+
+ if err := fs.WriteFile(cpuFileName, cpuData, 0644); err != nil {
+ t.Fatalf("failed to write file %s: %v", cpuFileName, err)
+ }
+
+ cpuInfo, err := NewCpuInfo(fs)
+ if err != nil {
+ t.Fatalf("got %v, want nil for error", err)
+ }
+
+ if !reflect.DeepEqual(cpuInfo, expectedCpuInfo) {
+ t.Errorf("got %v, expecting %v for CpuInfo", cpuInfo, expectedCpuInfo)
+ }
+
+}
+
+func TestNewMemInfo(t *testing.T) {
+ fs := fs.NewMockFs(nil)
+
+ if err := fs.MkDirs("/proc"); err != nil {
+ t.Fatalf("failed to create /proc dir: %v", err)
+ }
+ memFileName := "/proc/meminfo"
+
+ if err := fs.WriteFile(memFileName, memData, 0644); err != nil {
+ t.Fatalf("failed to write file %s: %v", memFileName, err)
+ }
+
+ memInfo, err := NewMemInfo(fs)
+ if err != nil {
+ t.Fatalf("got %v, want nil for error", err)
+ }
+
+ if !reflect.DeepEqual(memInfo, expectedMemInfo) {
+ t.Errorf("got %v, expecting %v for MemInfo", memInfo, expectedMemInfo)
+ }
+
+}
+
+var cpuData = []byte(`processor : 0
+vendor_id : %%VENDOR%%
+cpu family : 123
+model : 456
+model name : %%CPU MODEL NAME%%
+stepping : 0
+cpu MHz : 5555.555
+cache size : 512 KB
+physical id : 0
+siblings : 128
+core id : 0
+cpu cores : 64
+apicid : 0
+initial apicid : 0
+fpu : yes
+fpu_exception : yes
+cpuid level : 789
+wp : yes
+flags : %%cpu flags go here%%
+bugs : %%bugs go here%%
+
+processor : 1
+vendor_id : %%BADVENDOR%%
+cpu family : 234
+model : 567
+model name : %%BAD MODEL NAME%%
+flags : %%BAD cpu flags go here%%
+`)
+
+var expectedCpuInfo = &CpuInfo{
+ VendorId: "%%VENDOR%%",
+ ModelName: "%%CPU MODEL NAME%%",
+ CpuCores: 64,
+ Flags: "%%cpu flags go here%%",
+}
+
+var memData = []byte(`MemTotal: 1000 mB
+MemFree: 10240000
+MemAvailable: 3000 kB
+Buffers: 7177844 kB
+`)
+
+var expectedMemInfo = &MemInfo{
+ MemTotal: 1048576000,
+ MemFree: 10240000,
+ MemAvailable: 3072000,
+}