#include "CppUTestExt/MockSupport.h"
#include "socket.hh"
#include <stdexcept>

Socket::Socket(const char * device) {
	mock().actualCall("socket_constructor")
			.withStringParameter("device", device);
	if(!mock().returnBoolValueOrDefault(true)) {
		throw std::runtime_error("forced creation failure");
	}
}

Socket::~Socket() {}

int Socket::get_descriptor(void) {
	return mock().actualCall("get_descriptor").returnIntValue();
}

int Socket::send_message(const unsigned char *destination_address,
		const char *message, int message_length) {
	mock().actualCall("send_message")
			.withMemoryBufferParameter("destination_address", destination_address, ETHER_ADDR_LEN)
			.withMemoryBufferParameter("message", (const unsigned char*)message, message_length)
			.withIntParameter("message_length", message_length);
	int return_value = mock().returnIntValueOrDefault(6);
	if(return_value == -1) {
		throw std::runtime_error("forced send_message failure");
	}
	return return_value;
}

int Socket::receive_message(unsigned char *source_address,
			unsigned char *destination_address,
			char *buffer, int buffer_size) {
	mock().actualCall("receive_message")
			.withOutputParameter("source_address", source_address)
			.withOutputParameter("destination_address", destination_address)
			.withOutputParameter("buffer", buffer)
			.withIntParameter("buffer_size", buffer_size);
	int return_value = mock().returnIntValueOrDefault(6);
	if(return_value == -1) {
		throw std::runtime_error("forced receive_message failure");
	}
	return return_value;
}

void Socket::add_membership(Socket::MembershipType type,
		const unsigned char *multicast_address) {
	mock().actualCall("add_membership")
		.withIntParameter("type", type)
		.withMemoryBufferParameter("multicast_address", multicast_address, ETHER_ADDR_LEN);
	if(!mock().returnBoolValueOrDefault(true)) {
		throw std::runtime_error("forced add_membership failure");
	}
}

void Socket::drop_membership(Socket::MembershipType type,
		const unsigned char *multicast_address) {
	mock().actualCall("drop_membership")
		.withIntParameter("type", type)
		.withMemoryBufferParameter("multicast_address", multicast_address, ETHER_ADDR_LEN);
	if(!mock().returnBoolValueOrDefault(true)) {
		throw std::runtime_error("forced drop_membership failure");
	}
}