cesium-native/CesiumAsync/test/MockAssetResponse.h

#pragma once

#include <CesiumAsync/IAssetResponse.h>

#include <cstddef>
#include <vector>

class MockAssetResponse : public CesiumAsync::IAssetResponse {
public:
  MockAssetResponse() = default;
  MockAssetResponse(
      uint16_t statusCode,
      const std::string& contentType,
      const CesiumAsync::HttpHeaders& headers,
      const std::vector<std::byte>& data)
      : _statusCode{statusCode},
        _contentType{contentType},
        _headers{headers},
        _data{data} {}
  MockAssetResponse(MockAssetResponse&&) = default;
  MockAssetResponse(const MockAssetResponse& rhs) = default;
  MockAssetResponse& operator=(MockAssetResponse&&) = default;
  MockAssetResponse& operator=(const MockAssetResponse&) = default;

  virtual uint16_t statusCode() const override { return this->_statusCode; }

  virtual std::string contentType() const override {
    return this->_contentType;
  }

  virtual const CesiumAsync::HttpHeaders& headers() const override {
    return this->_headers;
  }

  virtual std::span<const std::byte> data() const override {
    return std::span<const std::byte>(_data.data(), _data.size());
  }

private:
  uint16_t _statusCode;
  std::string _contentType;
  CesiumAsync::HttpHeaders _headers;
  std::vector<std::byte> _data;
};