#ifndef MANTID_DATAHANDLING_EventWorkspaceCollectionTEST_H_
#define MANTID_DATAHANDLING_EventWorkspaceCollectionTEST_H_

#include <cxxtest/TestSuite.h>
#include "MantidKernel/TimeSeriesProperty.h"

#include <memory>
#include <boost/make_shared.hpp>

#include "MantidDataHandling/EventWorkspaceCollection.h"
#include "MantidDataObjects/EventWorkspace.h"
#include "MantidAPI/WorkspaceGroup.h"

using namespace Mantid::DataHandling;
using namespace Mantid::DataObjects;
using namespace Mantid::API;
using namespace Mantid::Kernel;

namespace {

EventWorkspaceCollection_uptr
makeEventWorkspaceCollection(unsigned int decoratorSize) {

  EventWorkspaceCollection_uptr decorator(new EventWorkspaceCollection);

  std::unique_ptr<const TimeSeriesProperty<int>> periodLog(
      new const TimeSeriesProperty<int>("period_log"));

  decorator->setNPeriods(decoratorSize, periodLog);

  return decorator;
}
}

class EventWorkspaceCollectionTest : public CxxTest::TestSuite {
public:
  // This pair of boilerplate methods prevent the suite being created statically
  // This means the constructor isn't called when running other tests
  static EventWorkspaceCollectionTest *createSuite() {
    return new EventWorkspaceCollectionTest();
  }
  static void destroySuite(EventWorkspaceCollectionTest *suite) {
    delete suite;
  }

  void test_constructor() {
    EventWorkspaceCollection decorator;
    TSM_ASSERT_EQUALS("Always one period by default", 1, decorator.nPeriods());
  }

  void test_output_single_workspace() {
    EventWorkspaceCollection decorator;
    TSM_ASSERT_EQUALS("Always one period by default", 1, decorator.nPeriods());
    TS_ASSERT_EQUALS(decorator.combinedWorkspace(),
                     decorator.getSingleHeldWorkspace());
  }

  void test_output_multiple_workspaces() {
    EventWorkspaceCollection decorator;

    std::unique_ptr<const TimeSeriesProperty<int>> periodLog(
        new const TimeSeriesProperty<int>("period_log"));
    decorator.setNPeriods(3, periodLog);

    WorkspaceGroup_sptr outWS = boost::dynamic_pointer_cast<WorkspaceGroup>(
        decorator.combinedWorkspace());
    TSM_ASSERT("Should be a WorkspaceGroup", outWS);
    TS_ASSERT_EQUALS(3, outWS->size());
  }

  void test_set_geometryFlag() {

    EventWorkspaceCollection_uptr decorator = makeEventWorkspaceCollection(3);

    const int geometryFlag = 3;

    decorator->setGeometryFlag(geometryFlag);

    WorkspaceGroup_sptr outWS = boost::dynamic_pointer_cast<WorkspaceGroup>(
        decorator->combinedWorkspace());

    for (size_t i = 0; i < outWS->size(); ++i) {
      auto memberWS =
          boost::dynamic_pointer_cast<EventWorkspace>(outWS->getItem(i));
      TSM_ASSERT_EQUALS(
          "Child workspaces should all have the geometry flag set",
          geometryFlag, memberWS->sample().getGeometryFlag());
    }
  }

  void test_set_thickness() {

    EventWorkspaceCollection_uptr decorator = makeEventWorkspaceCollection(3);

    const float thickness = 3;

    decorator->setThickness(thickness);

    WorkspaceGroup_sptr outWS = boost::dynamic_pointer_cast<WorkspaceGroup>(
        decorator->combinedWorkspace());

    for (size_t i = 0; i < outWS->size(); ++i) {
      auto memberWS =
          boost::dynamic_pointer_cast<EventWorkspace>(outWS->getItem(i));
      TSM_ASSERT_EQUALS("Child workspaces should all have the thickness set",
                        thickness, memberWS->sample().getThickness());
    }
  }

  void test_set_height() {
    EventWorkspaceCollection_uptr decorator = makeEventWorkspaceCollection(3);

    const float height = 3;

    decorator->setHeight(height);

    WorkspaceGroup_sptr outWS = boost::dynamic_pointer_cast<WorkspaceGroup>(
        decorator->combinedWorkspace());

    for (size_t i = 0; i < outWS->size(); ++i) {
      auto memberWS =
          boost::dynamic_pointer_cast<EventWorkspace>(outWS->getItem(i));
      TSM_ASSERT_EQUALS("Child workspaces should all have the height set",
                        height, memberWS->sample().getHeight());
    }
  }

  void test_set_width() {

    EventWorkspaceCollection_uptr decorator = makeEventWorkspaceCollection(3);

    const float width = 3;

    decorator->setWidth(width);

    WorkspaceGroup_sptr outWS = boost::dynamic_pointer_cast<WorkspaceGroup>(
        decorator->combinedWorkspace());

    for (size_t i = 0; i < outWS->size(); ++i) {
      auto memberWS =
          boost::dynamic_pointer_cast<EventWorkspace>(outWS->getItem(i));
      TSM_ASSERT_EQUALS("Child workspaces should all have the width set", width,
                        memberWS->sample().getWidth());
    }
  }
};

#endif /* MANTID_DATAHANDLING_EventWorkspaceCollectionTEST_H_ */