#include <fstream>
#include <memory>

#include <di/TCPClient.h>
#include <gtest/gtest.h>

using namespace Di;
using namespace std;

TEST(TCPClient, checkForConnection) {
	std::shared_ptr<TCPClient> c;

	c = make_shared<TCPClient>("google.com:80");
	EXPECT_TRUE(c->connected());
}

TEST(TCPClient, checkForNoConnection) {
	std::shared_ptr<TCPClient> c;

	c = make_shared<TCPClient>("127.0.0.1:40123");
	EXPECT_FALSE(c->connected());
}

TEST(TCPClient, checkSendData) {
	std::shared_ptr<TCPClient> c;

	c = make_shared<TCPClient>("google.com", 80);
	EXPECT_TRUE(c->connected());

	EXPECT_EQ(18, c->send("GET / HTTP/1.1\r\n\r\n"));
}

TEST(TCPClient, checkSendNoConn) {
	std::shared_ptr<TCPClient> c;

	c = make_shared<TCPClient>("google.com", 80);
	EXPECT_TRUE(c->connected());
	c->disconnect();
	EXPECT_FALSE(c->connected());

	EXPECT_EQ(-1, c->send("GET / HTTP/1.1\r\n\r\n"));
}