// TcpStreamBugTest.cpp : Defines the entry point for the console application.
//

#include "stdafx.h"
#include <iostream>
#include <iterator>
#include <cc++/socket.h>

using namespace ost;
using namespace std;

#define MAX_READ_BUF 1024

void test1(){
	char buf[MAX_READ_BUF];
	TCPStream m_str("194.87.0.50", 80); // www.ru
	m_str<<"GET"<<endl;

	do {
		(m_str).read(buf,MAX_READ_BUF);
		int readlen	= (m_str).gcount();
		cout.write(buf,readlen);
	}while((m_str).good());

	if(m_str.good())
		cout << "TCPStream good" << endl;
	else
		cout << "TCPStream fail" << endl;
}

void test2(){
	TCPStream m_str("194.87.0.50", 80); // www.ru
	m_str<<"GET"<<endl;

	std::copy(std::istreambuf_iterator<char>(m_str),
		std::istreambuf_iterator<char>(),
		std::ostream_iterator<char>(std::cout));

	if(m_str.good())
		cout << "TCPStream good" << endl;
	else
		cout << "TCPStream fail" << endl;

}

int _tmain(int argc, _TCHAR* argv[])
{
	cout << "TCPStream bug Test1..." << endl;
	test1();

	cout << endl << "TCPStream bug Test2..." << endl;
	test2();

	return 0;
}