/*
 *
 * Copyright (c) 2006 The Apache Software Foundation
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *    http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 *
 */

#include "Cluster.h"
#include "test_tools.h"

#include "qpid/framing/SessionPingBody.h"
#include "qpid/framing/SessionPongBody.h"
#include "qpid/cluster/ClassifierHandler.h"

#define BOOST_AUTO_TEST_MAIN    // Must come before #include<boost/test/*>
#include <boost/test/auto_unit_test.hpp>

#include <sys/wait.h>

static const ProtocolVersion VER;

/** Verify membership in a cluster with one member. */
BOOST_AUTO_TEST_CASE(testClusterOne) {
    TestCluster cluster("clusterOne", "amqp:one:1");
    AMQFrame send(VER, 1, new SessionPingBody(VER));
    cluster.handle(send);
    AMQFrame received;
    BOOST_REQUIRE(cluster.received.waitPop(received));
    BOOST_CHECK_TYPEID_EQUAL(SessionPingBody, *received.getBody());
    BOOST_CHECK_EQUAL(1u, cluster.size());
    Cluster::MemberList members = cluster.getMembers();
    BOOST_CHECK_EQUAL(1u, members.size());
    shared_ptr<const Cluster::Member> me=members.front();
    BOOST_REQUIRE_EQUAL(me->url, "amqp:one:1");
}

/** Fork a process to test a cluster with two members */
BOOST_AUTO_TEST_CASE(testClusterTwo) {
    bool nofork=getenv("NOFORK") != 0;
    pid_t pid=0;
    if (!nofork) {
        pid = fork();
        BOOST_REQUIRE(pid >= 0);
    }
    if (pid || nofork) {        // Parent
        BOOST_MESSAGE("Parent start");
        TestCluster cluster("clusterTwo", "amqp:parent:1");
        BOOST_REQUIRE(cluster.waitFor(2)); // Myself and child.

        // Exchange frames with child.
        AMQFrame send(VER, 1, new SessionPingBody(VER));
        cluster.handle(send);
        AMQFrame received;
        BOOST_REQUIRE(cluster.received.waitPop(received));
        BOOST_CHECK_TYPEID_EQUAL(SessionPingBody, *received.getBody());
        
        BOOST_REQUIRE(cluster.received.waitPop(received));
        BOOST_CHECK_TYPEID_EQUAL(SessionPongBody, *received.getBody());

        if (!nofork) {
            // Wait for child to exit.
            int status;
            BOOST_CHECK_EQUAL(::wait(&status), pid);
            BOOST_CHECK_EQUAL(0, status);
            BOOST_CHECK(cluster.waitFor(1));
            BOOST_CHECK_EQUAL(1u, cluster.size());
        }
    }
    else {                      // Child
        BOOST_REQUIRE(execl("./Cluster_child", "./Cluster_child", NULL));
    }
}

struct CountHandler : public FrameHandler {
    CountHandler() : count(0) {}
    void handle(AMQFrame&) { count++; }
    size_t count;
};
    
/** Test the ClassifierHandler */
BOOST_AUTO_TEST_CASE(testClassifierHandlerWiring) {
    AMQFrame queueDecl(VER, 0, new QueueDeclareBody(VER));
    AMQFrame messageTrans(VER, 0, new MessageTransferBody(VER));
    shared_ptr<CountHandler> wiring(new CountHandler());
    shared_ptr<CountHandler> other(new CountHandler());
    
    ClassifierHandler classify(wiring, other);

    classify.handle(queueDecl);
    BOOST_CHECK_EQUAL(1u, wiring->count);
    BOOST_CHECK_EQUAL(0u, other->count);
    
    classify.handle(messageTrans);
    BOOST_CHECK_EQUAL(1u, wiring->count);
    BOOST_CHECK_EQUAL(1u, other->count);
}