better TestSocketServer()

git-svn-id: https://svn.wxwidgets.org/svn/wx/wxWidgets/trunk@6760 c3d73ce0-8a6f-49c7-b76d-6d57e0e08775

samples/console/console.cpp

@@ -723,19 +723,22 @@ static void TestSocketServer()
 {
     puts("*** Testing wxSocketServer ***\n");
 
-    // we want to launch a server
+    static const int PORT = 3000;
+
     wxIPV4address addr;
-    addr.Service(3000);
+    addr.Service(PORT);
 
     wxSocketServer *server = new wxSocketServer(addr);
     if ( !server->Ok() )
     {
         puts("ERROR: failed to bind");
+
+        return;
     }
 
     for ( ;; )
     {
-        puts("Server: waiting for connection...");
+        printf("Server: waiting for connection on port %d...\n", PORT);
 
         wxSocketBase *socket = server->Accept();
         if ( !socket )
@@ -746,13 +749,21 @@ static void TestSocketServer()
 
         puts("Server: got a client.");
 
+        server->SetTimeout(60); // 1 min
+
+        while ( socket->IsConnected() )
+        {
             wxString s;
             char ch = '\0';
             for ( ;; )
             {
                 if ( socket->Read(&ch, sizeof(ch)).Error() )
+                {
+                    // don't log error if the client just close the connection
+                    if ( socket->IsConnected() )
                     {
                         puts("ERROR: in wxSocket::Read.");
+                    }
 
                     break;
                 }
@@ -782,11 +793,15 @@ static void TestSocketServer()
             socket->Write(s.MakeUpper().c_str(), s.length());
             socket->Write("\r\n", 2);
             printf("Server: wrote '%s'.\n", s.c_str());
-
-        delete socket;
         }
 
-    delete server;
+        puts("Server: lost a client.");
+
+        socket->Destroy();
+    }
+
+    // same as "delete server" but is consistent with GUI programs
+    server->Destroy();
 }
 
 static void TestSocketClient()
@@ -2608,9 +2623,9 @@ int main(int argc, char **argv)
 #endif // TEST_MIME
 
 #ifdef TEST_SOCKETS
-    if ( 0 )
-        TestSocketServer();
     if ( 1 )
+        TestSocketServer();
+    if ( 0 )
     {
         TestSocketClient();
         TestProtocolFtp();