def test_mode_n_on_both_bands(self): """Test that band is maintained when setting a mode N spec.""" spec = ap_spec.APSpec(band=ap_spec.BAND_5GHZ, mode=ap_spec.MODE_N) self.assertEquals(spec.band, ap_spec.BAND_5GHZ) self.assertEquals(spec.mode, ap_spec.MODE_N) spec = ap_spec.APSpec(band=ap_spec.BAND_2GHZ, mode=ap_spec.MODE_N) self.assertEquals(spec.band, ap_spec.BAND_2GHZ) self.assertEquals(spec.mode, ap_spec.MODE_N)
def testGetAPConfigurators_WithModeAPSpec(self): """Test with a mode only specified AP Spec""" self._build_ap_test_inventory() spec = ap_spec.APSpec(mode=ap_spec.DEFAULT_2GHZ_MODE) actual = self.factory.get_ap_configurators_by_spec(spec=spec) self.assertEquals([self.mock_ap1, self.mock_ap2].sort(), actual.sort()) spec = ap_spec.APSpec(mode=ap_spec.DEFAULT_5GHZ_MODE) actual = self.factory.get_ap_configurators_by_spec(spec=spec) self.assertEquals([self.mock_ap2], actual)
def testGetApConfigurators_WithBandAPSpec(self): """Test with a band only specified AP Spec""" self._build_ap_test_inventory() spec = ap_spec.APSpec(band=ap_spec.BAND_2GHZ) actual = self.factory.get_ap_configurators_by_spec(spec=spec) self.assertEquals([self.mock_ap1, self.mock_ap2].sort(), actual.sort()) spec = ap_spec.APSpec(band=ap_spec.BAND_5GHZ) actual = self.factory.get_ap_configurators_by_spec(spec=spec) self.assertEquals([self.mock_ap2], actual)
def testGetAPConfigurators_WithVisibilityAPSpec(self): """Test with a visibility specified AP Spec.""" self._build_ap_test_inventory() spec = ap_spec.APSpec(visible=True) actual = self.factory.get_ap_configurators_by_spec(spec=spec) self.assertEquals([self.mock_ap1, self.mock_ap2].sort(), actual.sort()) spec = ap_spec.APSpec(band=ap_spec.BAND_5GHZ, visible=False) actual = self.factory.get_ap_configurators_by_spec(spec=spec) self.assertEquals([self.mock_ap2], actual)
def testGetAPConfigurators_ByHostName(self): """Test obtaining a list of APs by hostname.""" self._build_ap_test_inventory() spec = ap_spec.APSpec(hostnames=['chromeos3-row2-rack1-host1']) actual = self.factory.get_ap_configurators_by_spec(spec=spec) self.assertEquals([self.mock_ap1], actual) spec = ap_spec.APSpec(hostnames=[ 'chromeos3-row2-rack1-host1', 'chromeos3-row2-rack1-host2' ]) actual = self.factory.get_ap_configurators_by_spec(spec=spec) self.assertEquals([self.mock_ap1, self.mock_ap2].sort(), actual.sort())
def testGetAPConfigurators_ByType(self): """Test obtaining configurators by type.""" self._build_ap_test_inventory() spec = ap_spec.APSpec(configurator_type=ap_spec.CONFIGURATOR_STATIC) actual = self.factory.get_ap_configurators_by_spec(spec=spec, pre_configure=True) self.assertEquals([self.mock_ap1], actual)
def test_set_security_psk_default(self): """Test setting security to WPAPSK.""" spec = ap_spec.APSpec(security=ap_spec.SECURITY_TYPE_WPAPSK) self.assertEquals(spec.visible, True) self.assertEquals(spec.security, ap_spec.SECURITY_TYPE_WPAPSK) self.assertEquals(spec.band, ap_spec.DEFAULT_BAND) self.assertEquals(spec.mode, ap_spec.DEFAULT_2GHZ_MODE) self.assertEquals(spec.channel, ap_spec.DEFAULT_2GHZ_CHANNEL)
def testGetAPConfigurators_ByLab(self): """Test obtaining configurators by location relative to the lab.""" self._build_ap_test_inventory() spec = ap_spec.APSpec(lab_ap=False) actual = self.factory.get_ap_configurators_by_spec(spec=spec, pre_configure=True) self.assertEquals([self.mock_ap3], actual)
def test_default_creation(self): """Test building a default ap_spec object.""" spec = ap_spec.APSpec() self.assertEquals(spec.visible, True) self.assertEquals(spec.security, ap_spec.DEFAULT_SECURITY_TYPE) self.assertEquals(spec.band, ap_spec.DEFAULT_BAND) self.assertEquals(spec.mode, ap_spec.DEFAULT_2GHZ_MODE) self.assertEquals(spec.channel, ap_spec.DEFAULT_2GHZ_CHANNEL) self.assertIsNone(spec.password)
def test_set_security_and_visibility(self): """Test setting visibility to hidden and security to WPAPSK.""" spec = ap_spec.APSpec(visible=False, security=ap_spec.SECURITY_TYPE_WPAPSK) self.assertEquals(spec.visible, False) self.assertEquals(spec.security, ap_spec.SECURITY_TYPE_WPAPSK) self.assertEquals(spec.band, ap_spec.DEFAULT_BAND) self.assertEquals(spec.mode, ap_spec.DEFAULT_2GHZ_MODE) self.assertEquals(spec.channel, ap_spec.DEFAULT_2GHZ_CHANNEL) self.assertIsNotNone(spec.password)
def testGetAndPreConfigureAPConfigurators(self): """Test preconfiguring APs.""" self._build_ap_test_inventory() # Pick something that is not the default channel. channel = ap_spec.VALID_5GHZ_CHANNELS[-1] spec = ap_spec.APSpec(channel=channel) actual = self.factory.get_ap_configurators_by_spec(spec=spec, pre_configure=True) self.assertEquals([self.mock_ap2], actual) self.assertEquals(actual[0].get_channel(), channel)
def test_only_set_band_5ghz(self): """Test setting only the band to 5GHz.""" spec = ap_spec.APSpec(band=ap_spec.BAND_5GHZ) self.assertEquals(spec.channel, ap_spec.DEFAULT_5GHZ_CHANNEL) self.assertEquals(spec.mode, ap_spec.DEFAULT_5GHZ_MODE)
def test_only_set_mode_5ghz(self): """Test setting only a 5GHz mode.""" spec = ap_spec.APSpec(mode=ap_spec.MODE_A) self.assertEquals(spec.band, ap_spec.BAND_5GHZ) self.assertEquals(spec.channel, ap_spec.DEFAULT_5GHZ_CHANNEL)
def test_only_set_mode_n(self): """Test setting the mode to N.""" spec = ap_spec.APSpec(mode=ap_spec.MODE_N) self.assertEquals(spec.band, ap_spec.DEFAULT_BAND) self.assertEquals(spec.channel, ap_spec.DEFAULT_2GHZ_CHANNEL)
def testGetAPConfigurators_WithSecurityAPSpec(self): """Test with a security only specified AP Spec""" self._build_ap_test_inventory() spec = ap_spec.APSpec(security=ap_spec.SECURITY_TYPE_WPAPSK) actual = self.factory.get_ap_configurators_by_spec(spec=spec) self.assertEquals([self.mock_ap1], actual)
def test_only_set_channel_5ghz(self): """Test setting only a 5GHz channel.""" spec = ap_spec.APSpec(channel=ap_spec.DEFAULT_5GHZ_CHANNEL) self.assertEquals(spec.band, ap_spec.BAND_5GHZ) self.assertEquals(spec.mode, ap_spec.DEFAULT_5GHZ_MODE)
def test_set_band_and_mode_5ghz(self): """Test setting the band and mode to valid 5GHz values.""" spec = ap_spec.APSpec(band=ap_spec.BAND_5GHZ, mode=ap_spec.MODE_A) self.assertEquals(spec.channel, ap_spec.DEFAULT_5GHZ_CHANNEL)
def test_set_band_mode_and_channel_5ghz(self): """Test setting the band and channel to valid 5GHz value.""" spec = ap_spec.APSpec(band=ap_spec.BAND_5GHZ, mode=ap_spec.MODE_N, channel=ap_spec.DEFAULT_5GHZ_CHANNEL) self.assertNotEquals(spec.mode, ap_spec.DEFAULT_2GHZ_MODE)