Попробуйте этот вспомогательный способ отключить Wi-Fi.И продолжайте выполнение теста как есть.(Только для Android)
static AppiumDriver driver;
public static void wifiOff() throws InterruptedException {
NetworkConnection mobileDriver = (NetworkConnection) driver;
if (mobileDriver.getNetworkConnection() !=
ConnectionType.AIRPLANE_MODE) {
// enabling Airplane mode
mobileDriver.setNetworkConnection(ConnectionType.AIRPLANE_MODE);
}
}