ConcretePeaksPresenterVsiTest.h

#ifndef CONCRETE_PEAKS_PRESENTER_VSI_TEST_H_
#define CONCRETE_PEAKS_PRESENTER_VSI_TEST_H_

#include "MantidVatesAPI/ConcretePeaksPresenterVsi.h"
#include "MantidAPI/IPeaksWorkspace.h"
#include "MantidVatesAPI/ViewFrustum.h"
#include "MantidDataObjects/PeaksWorkspace.h"
#include "MantidGeometry/Crystal/PeakShape.h"
#include "MantidDataObjects/PeakShapeSpherical.h"
#include "MantidDataObjects/PeakShapeEllipsoid.h"
#include "MantidDataObjects/NoShape.h"
#include "MantidKernel/V3D.h"
#include "MockObjects.h"
#include <boost/shared_ptr.hpp>
#include <boost/make_shared.hpp>

#include <cxxtest/TestSuite.h>
#include <gmock/gmock.h>
#include <gtest/gtest.h>


#include <string>

using namespace ::testing;
using namespace Mantid::VATES;

class MockPeakConcrete : public Mantid::DataObjects::Peak
{
public:
  MOCK_CONST_METHOD0(getHKL, Mantid::Kernel::V3D (void));
  MOCK_CONST_METHOD0(getQLabFrame, Mantid::Kernel::V3D (void));
  MOCK_CONST_METHOD0(getQSampleFrame, Mantid::Kernel::V3D (void));
};

class MockPeaksWorkspaceConcrete : public Mantid::DataObjects::PeaksWorkspace
{
public:
  MOCK_CONST_METHOD0(getSpecialCoordinateSystem, Mantid::Kernel::SpecialCoordinateSystem());
  MOCK_METHOD1(getPeak, Mantid::DataObjects::Peak & (int peakNum));
  MOCK_CONST_METHOD1(getPeak, const Mantid::DataObjects::Peak & (int peakNum));
};

class ConcretePeaksPresenterVsiTest : public CxxTest::TestSuite {
public:
  void testSetupPresenterCorrectly() {
    // Arrange
    std::string frame = "testFrame";
    std::string name = "name";
    
    LeftPlane left(1.0, 0.0, 0.0, 1.0);

    RightPlane right(-1.0, 0.0, 0.0, 1.0);
    BottomPlane bottom(0.0, 1.0, 0.0, 1.0);
    TopPlane top(0.0, -1.0, 0.0, 1.0);
    FarPlane farPlane(0.0, 0.0, 1.0, 1.0);
    NearPlane nearPlane(0.0, 0.0, -1.0,1.0);
    ViewFrustum_const_sptr frustum = boost::make_shared<const Mantid::VATES::ViewFrustum>(left, right, bottom, top, farPlane, nearPlane);

    boost::shared_ptr<MockPeaksWorkspaceConcrete> pw_ptr(new MockPeaksWorkspaceConcrete());
    // Act
    ConcretePeaksPresenterVsi presenter(pw_ptr, frustum, frame);

    // Assert
    TSM_ASSERT_EQUALS("Should have recorded the frame", presenter.getFrame(), frame);
  }

  void testCorrectPeaksInfoIsExtractedForValidRow() {
    // Arrange
    std::string frame = "Q_SAMPLE";
    
    LeftPlane left(1.0, 0.0, 0.0, 1.0);
    RightPlane right(-1.0, 0.0, 0.0, 1.0);
    BottomPlane bottom(0.0, 1.0, 0.0, 1.0);
    TopPlane top(0.0, -1.0, 0.0, 1.0);
    FarPlane farPlane(0.0, 0.0, 1.0, 1.0);
    NearPlane nearPlane(0.0, 0.0, -1.0,1.0);
    ViewFrustum_const_sptr frustum = boost::make_shared<const Mantid::VATES::ViewFrustum>(left, right, bottom, top, farPlane, nearPlane);

    Mantid::Kernel::V3D coordinate(1,0,0);
    double peakRadius = 10;
    Mantid::Kernel::SpecialCoordinateSystem coordinateSystem = Mantid::Kernel::SpecialCoordinateSystem::QSample;
    Mantid::Geometry::PeakShape_sptr shape(new Mantid::DataObjects::PeakShapeSpherical(peakRadius, coordinateSystem, "test", 1));
    MockPeakConcrete peak;
    peak.setPeakShape(shape);
    EXPECT_CALL(peak, getQLabFrame()).Times(0);
    EXPECT_CALL(peak, getHKL()).Times(0);
    EXPECT_CALL(peak, getQSampleFrame()).WillOnce(Return(coordinate));


    boost::shared_ptr<MockPeaksWorkspaceConcrete> pw_ptr(new MockPeaksWorkspaceConcrete());
    MockPeaksWorkspaceConcrete & pw = *pw_ptr;

    EXPECT_CALL(pw, getSpecialCoordinateSystem()).WillOnce(Return(coordinateSystem));
    EXPECT_CALL(pw, getPeak(_)).Times(2).WillRepeatedly(ReturnRef(peak));

    // Act
    ConcretePeaksPresenterVsi presenter(pw_ptr, frustum, frame);
    double radius = 0;
    Mantid::Kernel::V3D coord(0,0,0);
    presenter.getPeaksInfo(pw_ptr,0,coord, radius, coordinateSystem);

    //Assert
    TSM_ASSERT_EQUALS("Should have a radius of 10", radius, peakRadius);
    TSM_ASSERT_EQUALS("Should have the same coordinate", coord, coordinate);
  }
};

#endif