Add tests for enabling and disabling connections
Adds policy specific tests to check that the commands for starting and
stopping client connections work as expected, allowing or disallowing
connnections to happen, rejecting connecting requests if disabled,
ending connections when disabled, and indicating the state in status updates.
Bug: None
Change-Id: I98192246c886f412e69b0065545f873dca81d977
diff --git a/acts_tests/tests/google/fuchsia/wlan_policy/StartStopClientConnectionsTest.py b/acts_tests/tests/google/fuchsia/wlan_policy/StartStopClientConnectionsTest.py
new file mode 100644
index 0000000..7156a80
--- /dev/null
+++ b/acts_tests/tests/google/fuchsia/wlan_policy/StartStopClientConnectionsTest.py
@@ -0,0 +1,193 @@
+#!/usr/bin/env python3
+#
+# Copyright 2020 - 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.
+
+from acts import signals
+from acts.controllers.ap_lib import hostapd_constants
+from acts.controllers.ap_lib import hostapd_security
+from acts.test_utils.abstract_devices.utils_lib.wlan_utils import setup_ap
+from acts.test_utils.abstract_devices.utils_lib.wlan_policy_utils import setup_policy_tests, restore_state, save_network, start_connections, stop_connections
+from acts.test_utils.wifi.WifiBaseTest import WifiBaseTest
+from acts.utils import rand_ascii_str
+
+DISCONNECTED = "Disconnected"
+CONNECTION_STOPPED = "ConnectionStopped"
+CONNECTIONS_ENABLED = "ConnectionsEnabled"
+CONNECTIONS_DISABLED = "ConnectionsDisabled"
+WPA2 = "wpa2"
+
+
+class StartStopClientConnectionsTest(WifiBaseTest):
+ """ Tests that we see the expected behavior with enabling and disabling
+ client connections
+
+ Test Bed Requirement:
+ * One or more Fuchsia devices
+ * One Access Point
+ """
+ def setup_class(self):
+ super().setup_class()
+ # Start an AP with a hidden network
+ self.ssid = rand_ascii_str(hostapd_constants.AP_SSID_LENGTH_2G)
+ self.access_point = self.access_points[0]
+ self.password = rand_ascii_str(
+ hostapd_constants.AP_PASSPHRASE_LENGTH_2G)
+ self.security_type = WPA2
+ security = hostapd_security.Security(security_mode=self.security_type,
+ password=self.password)
+
+ self.access_point.stop_all_aps()
+ # TODO(63719) use varying values for AP that shouldn't affect the test.
+ setup_ap(self.access_point,
+ 'whirlwind',
+ hostapd_constants.AP_DEFAULT_CHANNEL_5G,
+ self.ssid,
+ security=security)
+
+ if len(self.fuchsia_devices) < 1:
+ raise EnvironmentError("No Fuchsia devices found.")
+ # Save the existing saved networks before we remove them for tests
+ # And remember whether client connections have started
+ self.preexisting_state = setup_policy_tests(self.fuchsia_devices)
+
+ def setup_test(self):
+ for fd in self.fuchsia_devices:
+ # Set new update listener so that the next test can always get the
+ # most recent udpdate immediately.
+ new_listener_result = fd.wlan_policy_lib.wlanSetNewListener()
+ if new_listener_result.get("error") != None:
+ self.log.warn(
+ "Error occurred initializing a new update listener for the facade, may"
+ "cause errors in updates for tests: %s" %
+ new_listener_result.get("error"))
+
+ resultRemove = fd.wlan_policy_lib.wlanRemoveAllNetworks()
+ if resultRemove.get("error") != None:
+ raise EnvironmentError(
+ "Failed to remove all networks in setup: %s" %
+ resultRemove.get("error"))
+
+ def teardown_class(self):
+ self.access_point.stop_all_aps()
+ for fd in self.fuchsia_devices:
+ # Put back the networks that were saved before tests began.
+ restore_state(self.fuchsia_devices, self.preexisting_state)
+
+ def connect_and_validate(self, fd, ssid, security_type, expected_response):
+ """ Sends a connect request to the device and verifies we get a response
+ without error. This does not validate that a connection will be
+ attempted. This will fail the test if there is an error sending the
+ connect request, or if we don't get the expected connect response."""
+ result_connect = fd.wlan_policy_lib.wlanConnect(ssid, security_type)
+ if result_connect.get("error") != None:
+ logging.error("Error occurred requesting a connection: %s" %
+ result_connect.get("error"))
+ raise EnvironmentError("Failed to send connect request")
+ response = result_connect.get("result")
+ if response != expected_response:
+ self.log.error(
+ "Incorrect connect request response. Expected: \"%s\", Actual: %s"
+ % (expected_response, response))
+ raise signals.TestFailure(
+ "Failed to get expected connect response")
+
+ def test_stop_client_connections_update(self):
+ for fd in self.fuchsia_devices:
+ if not stop_connections(fd):
+ raise EnvironmentError("Failed to stop client connecions")
+
+ # Check that the most recent update says that the device is not
+ # connected to anything and client connections are disabled
+ fd.wlan_policy_lib.wlanSetNewListener()
+ result_update = fd.wlan_policy_lib.wlanGetUpdate()
+ if result_update.get("error") != None:
+ self.log.error("Error occurred getting status update: %s" %
+ result_update.get("error"))
+ raise EnvironmentError("Failed to get update")
+
+ expected_update = {"networks": [], "state": CONNECTIONS_DISABLED}
+ if result_update.get("result") != expected_update:
+ self.log.error(
+ "Most recent status update does not indicate client "
+ "connections have stopped. Expected update: %s\nActual update:"
+ % (expected_update, result_update))
+ raise signals.TestFailure(
+ "Incorrect update after stopping client connections")
+
+ def test_start_client_connections_update(self):
+ for fd in self.fuchsia_devices:
+ if not start_connections(fd):
+ raise EnvironmentError("Failed to start client connecions")
+
+ # Check that the most recent update says that the device is not
+ # connected to anything and client connections are disabled
+ fd.wlan_policy_lib.wlanSetNewListener()
+ result_update = fd.wlan_policy_lib.wlanGetUpdate()
+ if result_update.get("error") != None:
+ self.log.error("Error occurred getting status update: %s" %
+ result_update.get("error"))
+ raise EnvironmentError("Failed to get update")
+
+ expected_update = {"networks": [], "state": CONNECTIONS_ENABLED}
+ if result_update.get("result") != expected_update:
+ self.log.error(
+ "Most recent status update does not indicate client "
+ "connections are enabled. Expected update: %s\nActual update:"
+ % (expected_update, result_update))
+ raise signals.TestFailure(
+ "Incorrect update after starting client connections")
+
+ def test_stop_client_connections_rejects_connections(self):
+ # Test that if we turn client connections off, our requests to connect
+ # are rejected.
+ for fd in self.fuchsia_devices:
+ if not stop_connections(fd):
+ raise EnvironmentError("Failed to stop client connecions")
+
+ # Save the network, otherwise connecting may fail because the
+ # network is not saved instead of client connections being off
+ if not save_network(fd, self.ssid, self.security_type,
+ self.password):
+ raise EnvironmentError("Failed to save network")
+ expected_response = "RejectedIncompatibleMode"
+ self.connect_and_validate(fd, self.ssid, self.security_type,
+ expected_response)
+
+ def test_start_stop_client_connections(self):
+ # Test that if we turn client connections on the device will connect,
+ # and if we turn of client connections the device will disconnect.
+ for fd in self.fuchsia_devices:
+ # Start client connections and check that we can
+ if not save_network(fd, self.ssid, self.security_type,
+ self.password):
+ raise EnvironmentError("Failed to save network")
+ if not start_connections(fd):
+ raise EnvironmentError("Failed to start client connections")
+
+ expected_response = "Acknowledged"
+ self.connect_and_validate(fd, self.ssid, self.security_type,
+ expected_response)
+
+ if not fd.wait_for_connect(self.ssid, self.security_type):
+ raise signals.TestFailure(
+ "Failed to connect after starting client connections")
+
+ # Stop client connections again and check that we disconnect
+ if not stop_connections(fd):
+ raise EnvironmentError("Failed to stop client connecions")
+ if not fd.wait_for_disconnect(self.ssid, self.security_type,
+ DISCONNECTED, CONNECTION_STOPPED):
+ raise signals.TestFailure(
+ "Failed to disconnect after client connections stopped")