/*
 *  This program is free software; you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License as published by
 *  the Free Software Foundation; either version 2 of the License, or
 *  (at your option) any later version.
 *
 *  This program is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU Library General Public License for more details.
 *
 *  You should have received a copy of the GNU General Public License
 *  along with this program; if not, write to the Free Software
 *  Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
 *
 * StreamRpcChannelTest.cpp
 * Test fixture for the StreamRpcChannel class
 * Copyright (C) 2005-2008 Simon Newton
 */

#include <cppunit/extensions/HelperMacros.h>
#include <google/protobuf/stubs/common.h>
#include <string>
#include "ola/io/SelectServer.h"
#include "ola/network/Socket.h"
#include "common/rpc/StreamRpcChannel.h"
#include "common/rpc/SimpleRpcController.h"
#include "common/rpc/TestService.pb.h"

using google::protobuf::NewCallback;
using ola::io::LoopbackDescriptor;
using ola::io::SelectServer;
using ola::rpc::EchoReply;
using ola::rpc::EchoRequest;
using ola::rpc::STREAMING_NO_RESPONSE;
using ola::rpc::SimpleRpcController;
using ola::rpc::StreamRpcChannel;
using ola::rpc::TestService;
using ola::rpc::TestService_Stub;
using std::string;

/*
 * Our test implementation
 */
class TestServiceImpl: public TestService {
  public:
    explicit TestServiceImpl(SelectServer *ss): m_ss(ss) {}
    ~TestServiceImpl() {}

    void Echo(::google::protobuf::RpcController* controller,
              const EchoRequest* request,
              EchoReply* response,
              ::google::protobuf::Closure* done);

    void FailedEcho(::google::protobuf::RpcController* controller,
                    const EchoRequest* request,
                    EchoReply* response,
                    ::google::protobuf::Closure* done);

    void Stream(::google::protobuf::RpcController* controller,
                const ::ola::rpc::EchoRequest* request,
                STREAMING_NO_RESPONSE* response,
                ::google::protobuf::Closure* done);

  private:
    SelectServer *m_ss;
};


class StreamRpcChannelTest: public CppUnit::TestFixture {
  CPPUNIT_TEST_SUITE(StreamRpcChannelTest);
  CPPUNIT_TEST(testEcho);
  CPPUNIT_TEST(testFailedEcho);
  CPPUNIT_TEST(testStreamRequest);
  CPPUNIT_TEST_SUITE_END();

  public:
    void setUp();
    void tearDown();
    void testEcho();
    void testFailedEcho();
    void testStreamRequest();
    void EchoComplete();
    void FailedEchoComplete();

  private:
    int m_fd_pair[2];
    SimpleRpcController m_controller;
    EchoRequest m_request;
    EchoReply m_reply;
    TestService_Stub *m_stub;
    SelectServer m_ss;
    TestServiceImpl *m_service;
    StreamRpcChannel *m_channel;
    LoopbackDescriptor *m_socket;
};


CPPUNIT_TEST_SUITE_REGISTRATION(StreamRpcChannelTest);


void TestServiceImpl::Echo(::google::protobuf::RpcController* controller,
                           const EchoRequest* request,
                           EchoReply* response,
                           ::google::protobuf::Closure* done) {
  response->set_data(request->data());
  done->Run();
  (void) controller;
  (void) request;
}


void TestServiceImpl::FailedEcho(::google::protobuf::RpcController* controller,
                                 const EchoRequest* request,
                                 EchoReply* response,
                                 ::google::protobuf::Closure* done) {
  controller->SetFailed("Error");
  done->Run();
  (void) request;
  (void) response;
}

void TestServiceImpl::Stream(::google::protobuf::RpcController* controller,
                             const ::ola::rpc::EchoRequest* request,
                             STREAMING_NO_RESPONSE* response,
                             ::google::protobuf::Closure* done) {
  CPPUNIT_ASSERT(!controller);
  CPPUNIT_ASSERT(!response);
  CPPUNIT_ASSERT(!done);
  CPPUNIT_ASSERT(request);
  CPPUNIT_ASSERT_EQUAL(string("foo"), request->data());
  m_ss->Terminate();
}


void StreamRpcChannelTest::setUp() {
  m_socket = new LoopbackDescriptor();
  m_socket->Init();

  m_service = new TestServiceImpl(&m_ss);
  m_channel = new StreamRpcChannel(m_service, m_socket);
  m_ss.AddReadDescriptor(m_socket);
  m_stub = new TestService_Stub(m_channel);
}


void StreamRpcChannelTest::tearDown() {
  m_ss.RemoveReadDescriptor(m_socket);
  delete m_socket;
  delete m_stub;
  delete m_channel;
  delete m_service;
}


void StreamRpcChannelTest::EchoComplete() {
  m_ss.Terminate();
  CPPUNIT_ASSERT(!m_controller.Failed());
  CPPUNIT_ASSERT_EQUAL(m_reply.data(), m_request.data());
}


void StreamRpcChannelTest::FailedEchoComplete() {
  m_ss.Terminate();
  CPPUNIT_ASSERT(m_controller.Failed());
}


/*
 * Check that we can call the echo method in the TestServiceImpl.
 */
void StreamRpcChannelTest::testEcho() {
  m_request.set_data("foo");
  m_stub->Echo(&m_controller,
               &m_request,
               &m_reply,
               NewCallback(this, &StreamRpcChannelTest::EchoComplete));

  m_ss.Run();
}


/*
 * Check that method that fail return correctly
 */
void StreamRpcChannelTest::testFailedEcho() {
  m_request.set_data("foo");
  m_stub->FailedEcho(
      &m_controller,
      &m_request,
      &m_reply,
      NewCallback(this, &StreamRpcChannelTest::FailedEchoComplete));
  m_ss.Run();
}

/*
 * Check stream requests work
 */
void StreamRpcChannelTest::testStreamRequest() {
  m_request.set_data("foo");
  m_stub->Stream(NULL, &m_request, NULL, NULL);
  m_ss.Run();
}