/*
 * ***** BEGIN LICENSE BLOCK *****
 * Version: MIT
 *
 * Copyright (c) 2010-2013 Alan Antonuk
 *
 * Permission is hereby granted, free of charge, to any person
 * obtaining a copy of this software and associated documentation
 * files (the "Software"), to deal in the Software without
 * restriction, including without limitation the rights to use, copy,
 * modify, merge, publish, distribute, sublicense, and/or sell copies
 * of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be
 * included in all copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
 * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
 * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
 * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS
 * BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN
 * ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
 * CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
 * SOFTWARE.
 * ***** END LICENSE BLOCK *****
 */

#include "connected_test.h"

using namespace AmqpClient;

TEST_F(connected_test, first_channel) {
  channel->DeclareExchange("test_channel_exchange",
                           Channel::EXCHANGE_TYPE_FANOUT, false, false, true);
  channel->DeleteExchange("test_channel_exchange");
}

// Check to see that channels are reused properly
TEST_F(connected_test, channel_reuse) {
  channel->DeclareExchange("test_channel_exchange1",
                           Channel::EXCHANGE_TYPE_FANOUT, false, false, true);
  channel->DeclareExchange("test_channel_exchange2",
                           Channel::EXCHANGE_TYPE_FANOUT, false, false, true);
  channel->DeleteExchange("test_channel_exchange1");
  channel->DeleteExchange("test_channel_exchange2");
}

// Check to see that a new channel is created when a channel is put in an
// exception state
TEST_F(connected_test, channel_recover_from_error) {
  EXPECT_THROW(channel->DeclareExchange("test_channel_exchangedoesnotexist",
                                        Channel::EXCHANGE_TYPE_FANOUT, true,
                                        false, true),
               ChannelException);

  channel->DeclareExchange("test_channel_exchange",
                           Channel::EXCHANGE_TYPE_FANOUT, false, false, true);
  channel->DeleteExchange("test_channel_exchange");
}

TEST_F(connected_test, channel_publish_success1) {
  BasicMessage::ptr_t message = BasicMessage::Create("Test message");

  channel->BasicPublish("", "test_channel_routingkey", message, false, false);
}

TEST_F(connected_test, channel_publish_success2) {
  BasicMessage::ptr_t message = BasicMessage::Create("Test message");

  channel->BasicPublish("", "test_channel_routingkey", message, false, false);
  channel->BasicPublish("", "test_channel_routingkey", message, false, false);
}

TEST_F(connected_test, channel_publish_returned_mandatory) {
  BasicMessage::ptr_t message = BasicMessage::Create("Test message");

  EXPECT_THROW(
      channel->BasicPublish("", "test_channel_noqueue", message, true, false),
      MessageReturnedException);
}

TEST_F(connected_test, channel_publish_full_rejected) {
  BasicMessage::ptr_t message = BasicMessage::Create("Test message");
  Table args;
  args.insert(TableEntry("x-max-length", 0));
  args.insert(TableEntry("x-overflow", "reject-publish"));
  std::string queue = channel->DeclareQueue("", false, false, true, true, args);

  EXPECT_THROW(channel->BasicPublish("", queue, message),
               MessageRejectedException);
}

TEST_F(connected_test, channel_publish_bad_exchange) {
  BasicMessage::ptr_t message = BasicMessage::Create("Test message");

  EXPECT_THROW(channel->BasicPublish("test_channel_badexchange",
                                     "test_channel_rk", message, false, false),
               ChannelException);
}

TEST_F(connected_test, channel_publish_bad_exchange_recover) {
  BasicMessage::ptr_t message = BasicMessage::Create("Test message");

  EXPECT_THROW(channel->BasicPublish("test_channel_badexchange",
                                     "test_channel_rk", message, false, false),
               ChannelException);

  channel->BasicPublish("", "test_channel_rk", message, false, false);
}

TEST_F(connected_test, channel_consume_success) {
  BasicMessage::ptr_t message = BasicMessage::Create("Test message");
  std::string queue = channel->DeclareQueue("");
  channel->BasicPublish("", queue, message);

  std::string consumer = channel->BasicConsume(queue);

  Envelope::ptr_t consumed_envelope;
  EXPECT_TRUE(channel->BasicConsumeMessage(consumer, consumed_envelope));
}

TEST_F(connected_test, channel_consume_success_timeout) {
  BasicMessage::ptr_t message = BasicMessage::Create("Test message");
  std::string queue = channel->DeclareQueue("");

  std::string consumer = channel->BasicConsume(queue, "", true, false);
  channel->BasicPublish("", queue, message);

  Envelope::ptr_t consumed_envelope;
  EXPECT_TRUE(channel->BasicConsumeMessage(consumer, consumed_envelope, 5000));
}

TEST(test_channels, big_message) {
  Channel::OpenOpts opts = connected_test::GetTestOpenOpts();
  opts.frame_max = 4096;
  Channel::ptr_t channel = Channel::Open(opts);
  BasicMessage::ptr_t message = BasicMessage::Create(std::string(4099, 'a'));

  std::string queue = channel->DeclareQueue("");

  std::string consumer = channel->BasicConsume(queue);
  channel->BasicPublish("", queue, message);

  Envelope::ptr_t consumed_envelope;
  EXPECT_TRUE(channel->BasicConsumeMessage(consumer, consumed_envelope));
}