Added more tests

This commit is contained in:
Davis King 2012-11-17 16:19:44 -05:00
parent 8b22c9138f
commit da72b57608

View File

@ -38,7 +38,7 @@ namespace
{
try
{
dlog << LINFO << "serving connection";
dlog << LINFO << "serv1: serving connection";
std::string temp;
in >> temp;
@ -62,19 +62,40 @@ namespace
};
class serv2 : public server_iostream
{
virtual void on_connect (
std::istream& ,
std::ostream& out,
const std::string& ,
const std::string& ,
unsigned short ,
unsigned short ,
uint64
)
{
try
{
dlog << LINFO << "serv2: serving connection";
out << "one two three four five";
}
catch (error& e)
{
error_string = e.what();
}
}
public:
std::string error_string;
};
// ----------------------------------------------------------------------------------------
class test_iosockstream : public tester
{
public:
test_iosockstream (
) :
tester ("test_iosockstream",
"Runs tests on the iosockstream component.")
{}
void perform_test (
)
void test1()
{
serv theserv;
theserv.set_listening_port(12345);
@ -101,6 +122,46 @@ namespace
if (theserv.error_string.size() != 0)
throw error(theserv.error_string);
}
// ----------------------------------------------------------------------------------------
void test2()
{
serv2 theserv;
theserv.set_listening_port(12345);
theserv.start_async();
for (int i = 0; i < 1001; ++i)
{
print_spinner();
iosockstream stream("localhost:12345");
std::string temp;
stream >> temp; DLIB_TEST(temp == "one");
stream >> temp; DLIB_TEST(temp == "two");
stream >> temp; DLIB_TEST(temp == "three");
stream >> temp; DLIB_TEST(temp == "four");
stream >> temp; DLIB_TEST(temp == "five");
}
}
// ----------------------------------------------------------------------------------------
class test_iosockstream : public tester
{
public:
test_iosockstream (
) :
tester ("test_iosockstream",
"Runs tests on the iosockstream component.")
{}
void perform_test (
)
{
test1();
test2();
}
} a;
}