Lock and Unlock device during test

AndroidManifest.xml

<?xml version="1.0" encoding="utf-8"?>
<manifest xmlns:android="http://schemas.android.com/apk/res/android">
...
    <uses-permission android:name="android.permission.DISABLE_KEYGUARD"/>
    <uses-permission android:name="android.permission.WAKE_LOCK"/>
...
</manifest>

OrdifyActivityTest.java

	@BeforeClass
	public static void setup() {
		unlockDevice();
	}

	@AfterClass
	public static void end() {
		lockDevice();
	}

	private static void lockDevice() {
		UiDevice uiDevice = UiDevice.getInstance(getInstrumentation());
		try {
			uiDevice.sleep();
		} catch (RemoteException e) {
			e.printStackTrace();
		}
	}

	protected static void unlockDevice() {
		UiDevice uiDevice = UiDevice.getInstance(getInstrumentation());
		try {
			uiDevice.wakeUp();
		} catch (RemoteException e) {
			e.printStackTrace();
		}

		Context app = getTargetContext().getApplicationContext();
		KeyguardManager keyguard = (KeyguardManager) app.getSystemService(Context.KEYGUARD_SERVICE);
		keyguard.newKeyguardLock("TAG").disableKeyguard();
	}