#include "../../msproviders/rowproviders/msrowproviderbase.h"
#include "../../msproviders/rowproviders/directmsrowprovider.h"
#include "../../msproviders/rowproviders/bdamsrowprovider.h"

#include <boost/test/unit_test.hpp>

#include <schaapcommon/reordering/msselection.h>

using schaapcommon::reordering::MSSelection;

namespace wsclean {

BOOST_AUTO_TEST_SUITE(ms_row_provider_base)

BOOST_AUTO_TEST_CASE(create_ms_row_proverider_invalid_ms) {
  BOOST_CHECK_THROW(
      ((void)MakeMsRowProvider("invalid_measurement_set", MSSelection{},
                               std::map<size_t, size_t>{{0, 0}}, "DATA",
                               "MODEL_DATA", false)),
      std::runtime_error);
}

BOOST_AUTO_TEST_CASE(create_ms_row_proverider_direct_ms) {
  std::unique_ptr<MsRowProviderBase> provider = MakeMsRowProvider(
      "test_data/MWA_MOCK.ms", MSSelection{}, std::map<size_t, size_t>{{0, 0}},
      "DATA", "MODEL_DATA", false);

  BOOST_REQUIRE(provider);
  BOOST_CHECK(dynamic_cast<DirectMSRowProvider*>(provider.get()));
}

BOOST_AUTO_TEST_CASE(create_ms_row_proverider_bda_ms) {
  std::unique_ptr<MsRowProviderBase> provider = MakeMsRowProvider(
      "test_data/MWA_BDA_MOCK.ms", MSSelection{},
      std::map<size_t, size_t>{{0, 0}}, "DATA", "MODEL_DATA", false);

  BOOST_REQUIRE(provider);
  BOOST_CHECK_NO_THROW((void)dynamic_cast<BdaMsRowProvider&>(*provider));
}

BOOST_AUTO_TEST_CASE(ms_has_bda_data) {
  BOOST_CHECK(!MsHasBdaData(casacore::MeasurementSet{"test_data/MWA_MOCK.ms"}));
  BOOST_CHECK(
      MsHasBdaData(casacore::MeasurementSet{"test_data/MWA_BDA_MOCK.ms"}));
}

BOOST_AUTO_TEST_SUITE_END()

}  // namespace wsclean
