Home | History | Annotate | Download | only in testing 
      1 /*
      2  * Copyright (C) 2014 Google Inc. All rights reserved.
      3  *
      4  * Redistribution and use in source and binary forms, with or without
      5  * modification, are permitted provided that the following conditions are
      6  * met:
      7  *
      8  *     * Redistributions of source code must retain the above copyright
      9  * notice, this list of conditions and the following disclaimer.
     10  *     * Redistributions in binary form must reproduce the above
     11  * copyright notice, this list of conditions and the following disclaimer
     12  * in the documentation and/or other materials provided with the
     13  * distribution.
     14  *     * Neither the name of Google Inc. nor the names of its
     15  * contributors may be used to endorse or promote products derived from
     16  * this software without specific prior written permission.
     17  *
     18  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
     19  * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
     20  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
     21  * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
     22  * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
     23  * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
     24  * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
     25  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
     26  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
     27  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
     28  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
     29  */
     30 
     31 #include "config.h"
     32 #include "InternalsGeolocation.h"
     33 
     34 #include "core/dom/Document.h"
     35 #include "core/page/Page.h"
     36 #include "core/testing/Internals.h"
     37 #include "modules/geolocation/GeolocationController.h"
     38 #include "modules/geolocation/GeolocationError.h"
     39 #include "modules/geolocation/GeolocationPosition.h"
     40 #include "modules/geolocation/testing/GeolocationClientMock.h"
     41 #include "wtf/CurrentTime.h"
     42 
     43 namespace WebCore {
     44 
     45 void InternalsGeolocation::setGeolocationClientMock(Internals&, Document* document)
     46 {
     47     ASSERT(document && document->frame());
     48     GeolocationClientMock* client = new GeolocationClientMock();
     49 
     50     for (Frame* childFrame = document->page()->mainFrame(); childFrame; childFrame = childFrame->tree().traverseNext()) {
     51         if (childFrame->isLocalFrame())
     52             GeolocationController::from(toLocalFrame(childFrame))->setClientForTest(client);
     53     }
     54 }
     55 
     56 void InternalsGeolocation::setGeolocationPosition(Internals&, Document* document, double latitude, double longitude, double accuracy)
     57 {
     58     ASSERT(document && document->frame());
     59     GeolocationClientMock* client = geolocationClient(document);
     60     if (!client)
     61         return;
     62     client->setPosition(GeolocationPosition::create(currentTime(), latitude, longitude, accuracy));
     63 }
     64 
     65 void InternalsGeolocation::setGeolocationPositionUnavailableError(Internals&, Document* document, const String& message)
     66 {
     67     ASSERT(document && document->frame());
     68     GeolocationClientMock* client = geolocationClient(document);
     69     if (!client)
     70         return;
     71     client->setPositionUnavailableError(message);
     72 }
     73 
     74 void InternalsGeolocation::setGeolocationPermission(Internals&, Document* document, bool allowed)
     75 {
     76     ASSERT(document && document->frame());
     77     GeolocationClientMock* client = geolocationClient(document);
     78     if (!client)
     79         return;
     80     client->setPermission(allowed);
     81 }
     82 
     83 int InternalsGeolocation::numberOfPendingGeolocationPermissionRequests(Internals&, Document* document)
     84 {
     85     ASSERT(document && document->frame());
     86     GeolocationClientMock* client = geolocationClient(document);
     87     if (!client)
     88         return -1;
     89     return client->numberOfPendingPermissionRequests();
     90 }
     91 
     92 GeolocationClientMock* InternalsGeolocation::geolocationClient(Document* document)
     93 {
     94     GeolocationController* controller = GeolocationController::from(document->frame());
     95     if (!controller->hasClientForTest())
     96         return 0;
     97     return static_cast<GeolocationClientMock*>(controller->client());
     98 }
     99 
    100 } // namespace WebCore
    101