mock_rom.h

Go to the documentation of this file.

#ifndef YAZE_TEST_MOCKS_MOCK_ROM_H
#define YAZE_TEST_MOCKS_MOCK_ROM_H
 
#include <gmock/gmock.h>
#include <gtest/gtest.h>
 
#include "testing.h"
 
#include "app/rom.h"
 
namespace yaze {
namespace test {
 
class MockRom : public Rom {
 public:
  MockRom() = default;
 
  // Override the only virtual method in Rom
  MOCK_METHOD(absl::Status, WriteHelper, (const WriteAction&), (override));
 
  absl::Status SetTestData(const std::vector<uint8_t>& data) {
    auto status = LoadFromData(data, false);  // Don't load Zelda3 specific data
    if (status.ok()) {
      test_data_loaded_ = true;
    }
    return status;
  }
  
  void SetObjectData(int object_id, const std::vector<uint8_t>& data) {
    object_data_[object_id] = data;
  }
  
  void SetRoomData(int room_id, const std::vector<uint8_t>& data) {
    room_data_[room_id] = data;
  }
  
  bool HasObjectData(int object_id) const {
    return object_data_.find(object_id) != object_data_.end();
  }
  
  bool HasRoomData(int room_id) const {
    return room_data_.find(room_id) != room_data_.end();
  }
 
  bool IsValid() const {
    return test_data_loaded_ && is_loaded() && size() > 0;
  }
 
  const std::vector<uint8_t>& GetObjectData(int object_id) const {
    static const std::vector<uint8_t> empty;
    auto it = object_data_.find(object_id);
    return (it != object_data_.end()) ? it->second : empty;
  }
 
  const std::vector<uint8_t>& GetRoomData(int room_id) const {
    static const std::vector<uint8_t> empty;
    auto it = room_data_.find(room_id);
    return (it != room_data_.end()) ? it->second : empty;
  }
 
 private:
  bool test_data_loaded_ = false;
  std::map<int, std::vector<uint8_t>> object_data_;
  std::map<int, std::vector<uint8_t>> room_data_;
};
 
}  // namespace test
}  // namespace yaze
 
#endif