UiAutomationDriver.java revision 52107c27b6b0f2b0fdfec995784c73746bb95c4e 
1/*
2 * Copyright (C) 2013 DroidDriver committers
3 *
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
7 *
8 *      http://www.apache.org/licenses/LICENSE-2.0
9 *
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
15 */
16
17package com.google.android.droiddriver.uiautomation;
18
19import android.app.UiAutomation;
20import android.os.SystemClock;
21import android.view.accessibility.AccessibilityNodeInfo;
22
23import com.google.android.droiddriver.UiElement;
24import com.google.android.droiddriver.base.AbstractDroidDriver;
25import com.google.android.droiddriver.exceptions.ElementNotFoundException;
26
27/**
28 * Implementation of a DroidDriver that is driven via the accessibility layer.
29 */
30public class UiAutomationDriver extends AbstractDroidDriver {
31
32  private final UiAutomationContext context;
33
34  public UiAutomationDriver(UiAutomation uiAutomation) {
35    this.context = new UiAutomationContext(uiAutomation);
36  }
37
38  @Override
39  public UiElement getRootElement() {
40    return context.getUiElement(getRootNode());
41  }
42
43  private AccessibilityNodeInfo getRootNode() {
44    for (int i = 0; i < 3; i++) {
45      AccessibilityNodeInfo root = context.getUiAutomation().getRootInActiveWindow();
46      if (root != null) {
47        return root;
48      }
49      SystemClock.sleep(250);
50    }
51    throw new ElementNotFoundException("Could not find root node!");
52  }
53}
54