blob: 82de5098e66bea7619774e0e20f9094ad9c3a1e2 [file] [log] [blame]
/*
* Copyright (C) 2010 The Android Open Source Project
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package android.bluetooth;
import android.content.BroadcastReceiver;
import android.content.Context;
import android.content.Intent;
import android.content.IntentFilter;
import android.os.Environment;
import android.util.Log;
import junit.framework.Assert;
import java.io.BufferedWriter;
import java.io.File;
import java.io.FileWriter;
import java.io.IOException;
public class BluetoothTestUtils extends Assert {
/**
* Timeout for {@link BluetoothAdapter#disable()} in ms.
*/
private static final int DISABLE_TIMEOUT = 5000;
/**
* Timeout for {@link BluetoothAdapter#enable()} in ms.
*/
private static final int ENABLE_TIMEOUT = 20000;
/**
* Timeout for {@link BluetoothAdapter#setScanMode(int)} in ms.
*/
private static final int SET_SCAN_MODE_TIMEOUT = 5000;
/**
* Timeout for {@link BluetoothAdapter#startDiscovery()} in ms.
*/
private static final int START_DISCOVERY_TIMEOUT = 5000;
/**
* Timeout for {@link BluetoothAdapter#cancelDiscovery()} in ms.
*/
private static final int CANCEL_DISCOVERY_TIMEOUT = 5000;
private static final int DISCOVERY_STARTED_FLAG = 1;
private static final int DISCOVERY_FINISHED_FLAG = 1 << 1;
private static final int SCAN_MODE_NONE_FLAG = 1 << 2;
private static final int SCAN_MODE_CONNECTABLE_FLAG = 1 << 3;
private static final int SCAN_MODE_CONNECTABLE_DISCOVERABLE_FLAG = 1 << 4;
private static final int STATE_OFF_FLAG = 1 << 5;
private static final int STATE_TURNING_ON_FLAG = 1 << 6;
private static final int STATE_ON_FLAG = 1 << 7;
private static final int STATE_TURNING_OFF_FLAG = 1 << 8;
/**
* Time between polls in ms.
*/
private static final int POLL_TIME = 100;
private Context mContext;
private BufferedWriter mOutputWriter;
private String mOutputFile;
private String mTag;
private class BluetoothReceiver extends BroadcastReceiver {
private int mFiredFlags = 0;
@Override
public void onReceive(Context context, Intent intent) {
synchronized (this) {
if (BluetoothAdapter.ACTION_DISCOVERY_STARTED.equals(intent.getAction())) {
mFiredFlags |= DISCOVERY_STARTED_FLAG;
} else if (BluetoothAdapter.ACTION_DISCOVERY_FINISHED.equals(intent.getAction())) {
mFiredFlags |= DISCOVERY_FINISHED_FLAG;
} else if (BluetoothAdapter.ACTION_SCAN_MODE_CHANGED.equals(intent.getAction())) {
int mode = intent.getIntExtra(BluetoothAdapter.EXTRA_SCAN_MODE,
BluetoothAdapter.ERROR);
assertNotSame(mode, BluetoothAdapter.ERROR);
switch (mode) {
case BluetoothAdapter.SCAN_MODE_NONE:
mFiredFlags |= SCAN_MODE_NONE_FLAG;
break;
case BluetoothAdapter.SCAN_MODE_CONNECTABLE:
mFiredFlags |= SCAN_MODE_CONNECTABLE_FLAG;
break;
case BluetoothAdapter.SCAN_MODE_CONNECTABLE_DISCOVERABLE:
mFiredFlags |= SCAN_MODE_CONNECTABLE_DISCOVERABLE_FLAG;
break;
}
} else if (BluetoothAdapter.ACTION_STATE_CHANGED.equals(intent.getAction())) {
int state = intent.getIntExtra(BluetoothAdapter.EXTRA_STATE,
BluetoothAdapter.ERROR);
assertNotSame(state, BluetoothAdapter.ERROR);
switch (state) {
case BluetoothAdapter.STATE_OFF:
mFiredFlags |= STATE_OFF_FLAG;
break;
case BluetoothAdapter.STATE_TURNING_ON:
mFiredFlags |= STATE_TURNING_ON_FLAG;
break;
case BluetoothAdapter.STATE_ON:
mFiredFlags |= STATE_ON_FLAG;
break;
case BluetoothAdapter.STATE_TURNING_OFF:
mFiredFlags |= STATE_TURNING_OFF_FLAG;
break;
}
}
}
}
public int getFiredFlags() {
synchronized (this) {
return mFiredFlags;
}
}
public void resetFiredFlags() {
synchronized (this) {
mFiredFlags = 0;
}
}
}
private BluetoothReceiver mReceiver = new BluetoothReceiver();
public BluetoothTestUtils(Context context, String tag) {
this(context, tag, null);
}
public BluetoothTestUtils(Context context, String tag, String outputFile) {
mContext = context;
mTag = tag;
mOutputFile = outputFile;
if (mOutputFile == null) {
mOutputWriter = null;
} else {
try {
mOutputWriter = new BufferedWriter(new FileWriter(new File(
Environment.getExternalStorageDirectory(), mOutputFile), true));
} catch (IOException e) {
Log.w(mTag, "Test output file could not be opened", e);
mOutputWriter = null;
}
}
IntentFilter filter = new IntentFilter();
filter.addAction(BluetoothAdapter.ACTION_DISCOVERY_FINISHED);
filter.addAction(BluetoothAdapter.ACTION_DISCOVERY_STARTED);
filter.addAction(BluetoothAdapter.ACTION_SCAN_MODE_CHANGED);
filter.addAction(BluetoothAdapter.ACTION_STATE_CHANGED);
mContext.registerReceiver(mReceiver, filter);
}
public void close() {
mContext.unregisterReceiver(mReceiver);
if (mOutputWriter != null) {
try {
mOutputWriter.close();
} catch (IOException e) {
Log.w(mTag, "Test output file could not be closed", e);
}
}
}
public void enable(BluetoothAdapter adapter) {
int mask = STATE_TURNING_ON_FLAG | STATE_ON_FLAG | SCAN_MODE_CONNECTABLE_FLAG;
mReceiver.resetFiredFlags();
int state = adapter.getState();
switch (state) {
case BluetoothAdapter.STATE_ON:
assertTrue(adapter.isEnabled());
return;
case BluetoothAdapter.STATE_OFF:
case BluetoothAdapter.STATE_TURNING_OFF:
assertFalse(adapter.isEnabled());
assertTrue(adapter.enable());
break;
case BluetoothAdapter.STATE_TURNING_ON:
assertFalse(adapter.isEnabled());
mask = 0; // Don't check for received intents since we might have missed them.
break;
default:
fail("enable() invalid state: state=" + state);
}
long s = System.currentTimeMillis();
while (System.currentTimeMillis() - s < ENABLE_TIMEOUT) {
state = adapter.getState();
if (state == BluetoothAdapter.STATE_ON) {
assertTrue(adapter.isEnabled());
if ((mReceiver.getFiredFlags() & mask) == mask) {
mReceiver.resetFiredFlags();
writeOutput(String.format("enable() completed in %d ms",
(System.currentTimeMillis() - s)));
return;
}
} else {
assertFalse(adapter.isEnabled());
assertEquals(BluetoothAdapter.STATE_TURNING_ON, state);
}
sleep(POLL_TIME);
}
int firedFlags = mReceiver.getFiredFlags();
mReceiver.resetFiredFlags();
fail(String.format("enable() timeout: state=%d (expected %d), flags=0x%x (expected 0x%x)",
state, BluetoothAdapter.STATE_ON, firedFlags, mask));
}
public void disable(BluetoothAdapter adapter) {
int mask = STATE_TURNING_OFF_FLAG | STATE_OFF_FLAG | SCAN_MODE_NONE_FLAG;
mReceiver.resetFiredFlags();
int state = adapter.getState();
switch (state) {
case BluetoothAdapter.STATE_OFF:
assertFalse(adapter.isEnabled());
return;
case BluetoothAdapter.STATE_ON:
assertTrue(adapter.isEnabled());
assertTrue(adapter.disable());
break;
case BluetoothAdapter.STATE_TURNING_ON:
assertFalse(adapter.isEnabled());
assertTrue(adapter.disable());
break;
case BluetoothAdapter.STATE_TURNING_OFF:
assertFalse(adapter.isEnabled());
mask = 0; // Don't check for received intents since we might have missed them.
break;
default:
fail("disable() invalid state: state=" + state);
}
long s = System.currentTimeMillis();
while (System.currentTimeMillis() - s < DISABLE_TIMEOUT) {
state = adapter.getState();
if (state == BluetoothAdapter.STATE_OFF) {
assertFalse(adapter.isEnabled());
if ((mReceiver.getFiredFlags() & mask) == mask) {
mReceiver.resetFiredFlags();
writeOutput(String.format("disable() completed in %d ms",
(System.currentTimeMillis() - s)));
return;
}
} else {
assertFalse(adapter.isEnabled());
assertEquals(BluetoothAdapter.STATE_TURNING_OFF, state);
}
sleep(POLL_TIME);
}
int firedFlags = mReceiver.getFiredFlags();
mReceiver.resetFiredFlags();
fail(String.format("disable() timeout: state=%d (expected %d), flags=0x%x (expected 0x%x)",
state, BluetoothAdapter.STATE_OFF, firedFlags, mask));
}
public void discoverable(BluetoothAdapter adapter) {
int mask = SCAN_MODE_CONNECTABLE_DISCOVERABLE_FLAG;
mReceiver.resetFiredFlags();
if (!adapter.isEnabled()) {
fail("discoverable() bluetooth not enabled");
}
int scanMode = adapter.getScanMode();
if (scanMode == BluetoothAdapter.SCAN_MODE_CONNECTABLE_DISCOVERABLE) {
return;
}
assertEquals(scanMode, BluetoothAdapter.SCAN_MODE_CONNECTABLE);
assertTrue(adapter.setScanMode(BluetoothAdapter.SCAN_MODE_CONNECTABLE_DISCOVERABLE));
long s = System.currentTimeMillis();
while (System.currentTimeMillis() - s < SET_SCAN_MODE_TIMEOUT) {
scanMode = adapter.getScanMode();
if (scanMode == BluetoothAdapter.SCAN_MODE_CONNECTABLE_DISCOVERABLE) {
if ((mReceiver.getFiredFlags() & mask) == mask) {
mReceiver.resetFiredFlags();
writeOutput(String.format("discoverable() completed in %d ms",
(System.currentTimeMillis() - s)));
return;
}
} else {
assertEquals(scanMode, BluetoothAdapter.SCAN_MODE_CONNECTABLE);
}
sleep(POLL_TIME);
}
int firedFlags = mReceiver.getFiredFlags();
mReceiver.resetFiredFlags();
fail(String.format("discoverable() timeout: scanMode=%d (expected %d), flags=0x%x "
+ "(expected 0x%x)", scanMode, BluetoothAdapter.SCAN_MODE_CONNECTABLE_DISCOVERABLE,
firedFlags, mask));
}
public void undiscoverable(BluetoothAdapter adapter) {
int mask = SCAN_MODE_CONNECTABLE_FLAG;
mReceiver.resetFiredFlags();
if (!adapter.isEnabled()) {
fail("undiscoverable() bluetooth not enabled");
}
int scanMode = adapter.getScanMode();
if (scanMode == BluetoothAdapter.SCAN_MODE_CONNECTABLE) {
return;
}
assertEquals(scanMode, BluetoothAdapter.SCAN_MODE_CONNECTABLE_DISCOVERABLE);
assertTrue(adapter.setScanMode(BluetoothAdapter.SCAN_MODE_CONNECTABLE));
long s = System.currentTimeMillis();
while (System.currentTimeMillis() - s < SET_SCAN_MODE_TIMEOUT) {
scanMode = adapter.getScanMode();
if (scanMode == BluetoothAdapter.SCAN_MODE_CONNECTABLE) {
if ((mReceiver.getFiredFlags() & mask) == mask) {
mReceiver.resetFiredFlags();
writeOutput(String.format("undiscoverable() completed in %d ms",
(System.currentTimeMillis() - s)));
return;
}
} else {
assertEquals(scanMode, BluetoothAdapter.SCAN_MODE_CONNECTABLE_DISCOVERABLE);
}
sleep(POLL_TIME);
}
int firedFlags = mReceiver.getFiredFlags();
mReceiver.resetFiredFlags();
fail(String.format("undiscoverable() timeout: scanMode=%d (expected %d), flags=0x%x "
+ "(expected 0x%x)", scanMode, BluetoothAdapter.SCAN_MODE_CONNECTABLE, firedFlags,
mask));
}
public void startScan(BluetoothAdapter adapter) {
int mask = DISCOVERY_STARTED_FLAG;
mReceiver.resetFiredFlags();
if (!adapter.isEnabled()) {
fail("startScan() bluetooth not enabled");
}
if (adapter.isDiscovering()) {
return;
}
assertTrue(adapter.startDiscovery());
long s = System.currentTimeMillis();
while (System.currentTimeMillis() - s < START_DISCOVERY_TIMEOUT) {
if (adapter.isDiscovering() && ((mReceiver.getFiredFlags() & mask) == mask)) {
mReceiver.resetFiredFlags();
writeOutput(String.format("startScan() completed in %d ms",
(System.currentTimeMillis() - s)));
return;
}
sleep(POLL_TIME);
}
int firedFlags = mReceiver.getFiredFlags();
mReceiver.resetFiredFlags();
fail(String.format("startScan() timeout: isDiscovering=%b, flags=0x%x (expected 0x%x)",
adapter.isDiscovering(), firedFlags, mask));
}
public void stopScan(BluetoothAdapter adapter) {
int mask = DISCOVERY_FINISHED_FLAG;
mReceiver.resetFiredFlags();
if (!adapter.isEnabled()) {
fail("stopScan() bluetooth not enabled");
}
if (!adapter.isDiscovering()) {
return;
}
// TODO: put assertTrue() around cancelDiscovery() once it starts returning true.
adapter.cancelDiscovery();
long s = System.currentTimeMillis();
while (System.currentTimeMillis() - s < CANCEL_DISCOVERY_TIMEOUT) {
if (!adapter.isDiscovering() && ((mReceiver.getFiredFlags() & mask) == mask)) {
mReceiver.resetFiredFlags();
writeOutput(String.format("stopScan() completed in %d ms",
(System.currentTimeMillis() - s)));
return;
}
sleep(POLL_TIME);
}
int firedFlags = mReceiver.getFiredFlags();
mReceiver.resetFiredFlags();
fail(String.format("stopScan() timeout: isDiscovering=%b, flags=0x%x (expected 0x%x)",
adapter.isDiscovering(), firedFlags, mask));
}
public void writeOutput(String s) {
Log.i(mTag, s);
if (mOutputWriter == null) {
return;
}
try {
mOutputWriter.write(s + "\n");
mOutputWriter.flush();
} catch (IOException e) {
Log.w(mTag, "Could not write to output file", e);
}
}
private void sleep(long time) {
try {
Thread.sleep(time);
} catch (InterruptedException e) {
}
}
}