// 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,
}