page_load_notification_observer.cc

// Copyright 2013 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "chrome/test/remoting/page_load_notification_observer.h"

#include "content/public/browser/navigation_controller.h"
#include "content/public/browser/notification_service.h"
#include "content/public/browser/notification_types.h"
#include "content/public/browser/web_contents.h"

namespace remoting {

PageLoadNotificationObserver::PageLoadNotificationObserver(const GURL& target)
    : WindowedNotificationObserver(
          content::NOTIFICATION_LOAD_STOP,
          base::Bind(&PageLoadNotificationObserver::IsTargetLoaded,
                     base::Unretained(this))),
      target_(target),
      ignore_url_parameters_(false),
      matched_source_(content::NotificationService::AllSources()) {
}

PageLoadNotificationObserver::~PageLoadNotificationObserver() {}

bool PageLoadNotificationObserver::IsTargetLoaded() {
  content::NavigationController* controller =
      content::Source<content::NavigationController>(source()).ptr();
  GURL current_url = controller->GetWebContents()->GetURL();
  bool result = false;
  if (ignore_url_parameters_) {
    GURL::Replacements strip_parameters;
    strip_parameters.ClearQuery();
    strip_parameters.ClearRef();
    result = current_url.ReplaceComponents(strip_parameters) ==
        target_.ReplaceComponents(strip_parameters);
  } else {
    result = current_url == target_;
  }
  if (result) {
    matched_source_ = source();
  }
  return result;
}

}  // namespace remoting