Added more tests

dlib/test/iosockstream.cpp

@@ -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";
 
-    class test_iosockstream : public tester
+                out << "one two three four five";
+            }
+            catch (error& e)
             {
+                error_string = e.what();
+            }
+
+        }
+
+
     public:
-        test_iosockstream (
-        ) :
-            tester ("test_iosockstream",
-                    "Runs tests on the iosockstream component.")
-        {}
+        std::string error_string;
 
-        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;
 
 }