Merge pull request #201 from QuickJack/master
[mono.git] / mcs / class / System / Test / System.Net / WebClientTest.cs
index fa57be61aec5571ed3fed578ea9a5c0ec43f6a02..84954f9be4c0b094513346263f1c7eb1debc5680 100644 (file)
@@ -5,12 +5,15 @@
 //
 
 using System;
+using System.Collections;
 using System.Collections.Specialized;
+using System.Globalization;
 using System.IO;
 using System.Net;
-using System.Collections;
+using System.Net.Sockets;
 using System.Runtime.Serialization;
-
+using System.Text;
+using System.Threading;
 using NUnit.Framework;
 
 namespace MonoTests.System.Net
@@ -382,6 +385,30 @@ namespace MonoTests.System.Net
                                Assert.IsNotNull (inner.Message, "#9");
                        }
                }
+
+               [Test]
+               public void EncodingTest ()
+               {
+                       WebClient wc = new WebClient ();
+                       Assert.AreSame (Encoding.Default, wc.Encoding, "#1");
+                       wc.Encoding = Encoding.ASCII;
+                       Assert.AreSame (Encoding.ASCII, wc.Encoding, "#2");
+               }
+
+               [Test]
+               public void Encoding_Value_Null ()
+               {
+                       WebClient wc = new WebClient ();
+                       try {
+                               wc.Encoding = null;
+                               Assert.Fail ("#1");
+                       } catch (ArgumentNullException ex) {
+                               Assert.AreEqual (typeof (ArgumentNullException), ex.GetType (), "#2");
+                               Assert.IsNull (ex.InnerException, "#3");
+                               Assert.IsNotNull (ex.Message, "#4");
+                               Assert.AreEqual ("Encoding", ex.ParamName, "#6");
+                       }
+               }
 #endif
 
                [Test] // OpenRead (string)
@@ -1667,6 +1694,30 @@ namespace MonoTests.System.Net
                }
 #endif
 
+               [Test]
+               public void UploadValues1 ()
+               {
+                       IPEndPoint ep = new IPEndPoint (IPAddress.Loopback, 8000);
+                       string url = "http://" + IPAddress.Loopback.ToString () + ":8000/test/";
+
+                       using (SocketResponder responder = new SocketResponder (ep, new SocketRequestHandler (EchoRequestHandler))) {
+                               responder.Start ();
+
+                               WebClient wc = new WebClient ();
+#if NET_2_0
+                               wc.Encoding = Encoding.ASCII;
+#endif
+
+                               NameValueCollection nvc = new NameValueCollection ();
+                               nvc.Add ("Name", "\u0041\u2262\u0391\u002E");
+                               nvc.Add ("Address", "\u002E\u2262\u0041\u0391");
+
+                               byte [] buffer = wc.UploadValues (url, nvc);
+                               string response = Encoding.UTF8.GetString (buffer);
+                               Assert.AreEqual ("Name=A%e2%89%a2%ce%91.&Address=.%e2%89%a2A%ce%91\r\n", response);
+                       }
+               }
+
                [Test] // UploadValues (string, NameValueCollection)
                public void UploadValues1_Address_Null ()
                {
@@ -1982,5 +2033,110 @@ namespace MonoTests.System.Net
                        }
                }
 #endif
+
+#if NET_2_0
+               [Test]
+               public void GetWebRequestOverriding ()
+               {
+                       GetWebRequestOverridingTestClass testObject = new GetWebRequestOverridingTestClass ();
+                       testObject.DownloadData ("http://www.mono-project.com");
+
+                       Assert.IsTrue (testObject.overridedCodeRan, "Overrided code wasn't called");
+               }
+               
+               class GetWebRequestOverridingTestClass : WebClient
+               {
+                       internal bool overridedCodeRan = false;
+                       protected override WebRequest GetWebRequest(Uri address)
+                       {
+                               overridedCodeRan = true;
+                               return base.GetWebRequest (address);
+                       }
+               }
+#endif
+
+               static byte [] EchoRequestHandler (Socket socket)
+               {
+                       MemoryStream ms = new MemoryStream ();
+                       byte [] buffer = new byte [4096];
+                       int bytesReceived = socket.Receive (buffer);
+                       while (bytesReceived > 0) {
+                               ms.Write (buffer, 0, bytesReceived);
+                                // We don't check for Content-Length or anything else here, so we give the client a little time to write
+                                // after sending the headers
+                               Thread.Sleep (200);
+                               if (socket.Available > 0) {
+                                       bytesReceived = socket.Receive (buffer);
+                               } else {
+                                       bytesReceived = 0;
+                               }
+                       }
+                       ms.Flush ();
+                       ms.Position = 0;
+
+                       StringBuilder sb = new StringBuilder ();
+
+                       string expect = null;
+
+                       StreamReader sr = new StreamReader (ms, Encoding.UTF8);
+                       string line = null;
+                       byte state = 0;
+                       while ((line = sr.ReadLine ()) != null) {
+                               if (state > 0) {
+                                       state = 2;
+                                       sb.Append (line);
+                                       sb.Append ("\r\n");
+                               } if (line.Length == 0) {
+                                       state = 1;
+                               } else if (line.StartsWith ("Expect:")) {
+                                       expect = line.Substring (8);
+                               }
+                       }
+
+                       StringWriter sw = new StringWriter ();
+
+                       if (expect == "100-continue" && state != 2) {
+                               sw.WriteLine ("HTTP/1.1 100 Continue");
+                               sw.WriteLine ();
+                               sw.Flush ();
+
+                               socket.Send (Encoding.UTF8.GetBytes (sw.ToString ()));
+
+                               // receive body
+                               ms = new MemoryStream ();
+                               buffer = new byte [4096];
+                               bytesReceived = socket.Receive (buffer);
+                               while (bytesReceived > 0) {
+                                       ms.Write (buffer, 0, bytesReceived);
+                                       Thread.Sleep (200);
+                                       if (socket.Available > 0) {
+                                               bytesReceived = socket.Receive (buffer);
+                                       } else {
+                                               bytesReceived = 0;
+                                       }
+                               }
+                               ms.Flush ();
+                               ms.Position = 0;
+
+                               sb = new StringBuilder ();
+                               sr = new StreamReader (ms, Encoding.UTF8);
+                               line = sr.ReadLine ();
+                               while (line != null) {
+                                       sb.Append (line);
+                                       sb.Append ("\r\n");
+                                       line = sr.ReadLine ();
+                               }
+                       }
+
+                       sw = new StringWriter ();
+                       sw.WriteLine ("HTTP/1.1 200 OK");
+                       sw.WriteLine ("Content-Type: text/xml");
+                       sw.WriteLine ("Content-Length: " + sb.Length.ToString (CultureInfo.InvariantCulture));
+                       sw.WriteLine ();
+                       sw.Write (sb.ToString ());
+                       sw.Flush ();
+
+                       return Encoding.UTF8.GetBytes (sw.ToString ());
+               }
        }
 }