Ensure we get the post-redirect url

1 files changed, 8 insertions, 0 deletions

diff --git a/components/script/webdriver_handlers.rs b/components/script/webdriver_handlers.rs
index 591e38eb52b..e3b50cb2aaf 100644
--- a/components/script/webdriver_handlers.rs
+++ b/components/script/webdriver_handlers.rs
@@ -24,6 +24,7 @@ use js::jsval::UndefinedValue;
 
 use ipc_channel::ipc::IpcSender;
 use std::rc::Rc;
+use url::Url;
 
 fn find_node_by_unique_id(page: &Rc<Page>, pipeline: PipelineId, node_id: String) -> Option<Root<Node>> {
     let page = get_page(&*page, pipeline);
@@ -179,3 +180,10 @@ pub fn handle_get_name(page: &Rc<Page>,
         None => Err(())
     }).unwrap();
 }
+
+pub fn handle_get_url(page: &Rc<Page>,
+                      _pipeline: PipelineId,
+                      reply: IpcSender<Url>) {
+    let url = page.document().r().url();
+    reply.send(url).unwrap();
+}