tlbmc/collector/fake_fru_collector.cc

#include "tlbmc/collector/fake_fru_collector.h"

#include "absl/status/statusor.h"
#include "fru.pb.h"
#include "memory.pb.h"
#include "processor.pb.h"
#include "status.pb.h"

namespace milotic_tlbmc {

absl::StatusOr<FruTable>
FakeFruCollectorForTest::ScanAllFrusDeterministically() {
  FruTable frus;
  auto map = frus.mutable_key_to_fru();
  Fru fru1;
  fru1.mutable_attributes()->set_presence_status(
      milotic_tlbmc::PresenceStatus::PRESENCE_STATUS_PRESENT);
  fru1.mutable_attributes()
      ->mutable_expected_hardware_info()
      ->set_expected_barepath("/phys/PE0");
  map->insert({"fru1", fru1});

  Fru fru2;
  fru2.mutable_attributes()->set_presence_status(
      milotic_tlbmc::PresenceStatus::PRESENCE_STATUS_NOT_FOUND);
  fru2.mutable_attributes()
      ->mutable_expected_hardware_info()
      ->set_expected_barepath("/phys/PE1");
  map->insert({"fru2", fru2});
  return frus;
}

FruTable FakeFruCollectorForTest::GetCopyOfSmbiosFruTable() const {
  FruTable frus;
  auto* map = frus.mutable_key_to_fru();
  Fru fru_proc;
  fru_proc.mutable_attributes()
      ->mutable_expected_hardware_info()
      ->set_expected_barepath("/phys/CPU0");
  fru_proc.mutable_attributes()->set_presence_status(
      milotic_tlbmc::PresenceStatus::PRESENCE_STATUS_PRESENT);
  fru_proc.mutable_data()->mutable_processor_info()->set_stepping("2");
  map->insert({"/phys/CPU0", fru_proc});

  Fru fru_mem;
  fru_mem.mutable_attributes()
      ->mutable_expected_hardware_info()
      ->set_expected_barepath("/phys/DIMM0");
  fru_mem.mutable_attributes()->set_presence_status(
      milotic_tlbmc::PresenceStatus::PRESENCE_STATUS_PRESENT);
  fru_mem.mutable_data()->mutable_memory_info()->set_capacity_mib(65536);
  fru_mem.mutable_data()->mutable_memory_info()->set_serial_number(
      "80AD032427E1A75876");
  fru_mem.mutable_data()->mutable_memory_info()->set_part_number(
      "HMCG94AGBRA181N");
  fru_mem.mutable_data()->mutable_memory_info()->set_manufacturer("Hynix");
  map->insert({"/phys/DIMM0", fru_mem});

  return frus;
}

}  // namespace milotic_tlbmc