#!/usr/bin/env python
# Copyright 2011 Google Inc. All Rights Reserved.
#
# 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.
"""Unit tests for platformsettings.
Usage:
$ ./platformsettings_test.py
"""
import unittest
import platformsettings
WINDOWS_7_IP = '172.11.25.170'
WINDOWS_7_MAC = '00-1A-44-DA-88-C0'
WINDOWS_7_IPCONFIG = """
Windows IP Configuration
Host Name . . . . . . . . . . . . : THEHOST1-W
Primary Dns Suffix . . . . . . . : something.example.com
Node Type . . . . . . . . . . . . : Hybrid
IP Routing Enabled. . . . . . . . : No
WINS Proxy Enabled. . . . . . . . : No
DNS Suffix Search List. . . . . . : example.com
another.example.com
Ethernet adapter Local Area Connection:
Connection-specific DNS Suffix . : somethingexample.com
Description . . . . . . . . . . . : Int PRO/1000 MT Network Connection
Physical Address. . . . . . . . . : %(mac_addr)s
DHCP Enabled. . . . . . . . . . . : Yes
Autoconfiguration Enabled . . . . : Yes
IPv6 Address. . . . . . . . . . . : 1234:0:1000:1200:839f:d256:3a6c:210(Preferred)
Temporary IPv6 Address. . . . . . : 2143:0:2100:1800:38f9:2d65:a3c6:120(Preferred)
Link-local IPv6 Address . . . . . : abcd::1234:1a33:b2cc:238%%18(Preferred)
IPv4 Address. . . . . . . . . . . : %(ip_addr)s(Preferred)
Subnet Mask . . . . . . . . . . . : 255.255.248.0
Lease Obtained. . . . . . . . . . : Thursday, April 28, 2011 9:40:22 PM
Lease Expires . . . . . . . . . . : Tuesday, May 10, 2011 12:15:48 PM
Default Gateway . . . . . . . . . : abcd::2:37ee:ef70:56%%18
172.11.25.254
DHCP Server . . . . . . . . . . . : 172.11.22.33
DNS Servers . . . . . . . . . . . : 8.8.4.4
NetBIOS over Tcpip. . . . . . . . : Enabled
""" % {'ip_addr': WINDOWS_7_IP, 'mac_addr': WINDOWS_7_MAC}
WINDOWS_XP_IP = '172.1.2.3'
WINDOWS_XP_MAC = '00-34-B8-1F-FA-70'
WINDOWS_XP_IPCONFIG = """
Windows IP Configuration
Host Name . . . . . . . . . . . . : HOSTY-0
Primary Dns Suffix . . . . . . . :
Node Type . . . . . . . . . . . . : Unknown
IP Routing Enabled. . . . . . . . : No
WINS Proxy Enabled. . . . . . . . : No
DNS Suffix Search List. . . . . . : example.com
Ethernet adapter Local Area Connection 2:
Connection-specific DNS Suffix . : example.com
Description . . . . . . . . . . . : Int Adapter (PILA8470B)
Physical Address. . . . . . . . . : %(mac_addr)s
Dhcp Enabled. . . . . . . . . . . : Yes
Autoconfiguration Enabled . . . . : Yes
IP Address. . . . . . . . . . . . : %(ip_addr)s
Subnet Mask . . . . . . . . . . . : 255.255.254.0
Default Gateway . . . . . . . . . : 172.1.2.254
DHCP Server . . . . . . . . . . . : 172.1.3.241
DNS Servers . . . . . . . . . . . : 172.1.3.241
8.8.8.8
8.8.4.4
Lease Obtained. . . . . . . . . . : Thursday, April 07, 2011 9:14:55 AM
Lease Expires . . . . . . . . . . : Thursday, April 07, 2011 1:14:55 PM
""" % {'ip_addr': WINDOWS_XP_IP, 'mac_addr': WINDOWS_XP_MAC}
# scutil show State:/Network/Global/IPv4
OSX_IPV4_STATE = """
<dictionary> {
PrimaryInterface : en1
PrimaryService : 8824452C-FED4-4C09-9256-40FB146739E0
Router : 192.168.1.1
}
"""
# scutil show State:/Network/Service/[PRIMARY_SERVICE_KEY]/DNS
OSX_DNS_STATE_LION = """
<dictionary> {
DomainName : mtv.corp.google.com
SearchDomains : <array> {
0 : mtv.corp.google.com
1 : corp.google.com
2 : prod.google.com
3 : prodz.google.com
4 : google.com
}
ServerAddresses : <array> {
0 : 172.72.255.1
1 : 172.49.117.57
2 : 172.54.116.57
}
}
"""
OSX_DNS_STATE_SNOW_LEOPARD = """
<dictionary> {
ServerAddresses : <array> {
0 : 172.27.1.1
1 : 172.94.117.57
2 : 172.45.116.57
}
DomainName : mtv.corp.google.com
SearchDomains : <array> {
0 : mtv.corp.google.com
1 : corp.google.com
2 : prod.google.com
3 : prodz.google.com
4 : google.com
}
}
"""
class SystemProxyTest(unittest.TestCase):
def test_basic(self):
system_proxy = platformsettings.SystemProxy(None, None)
self.assertEqual(None, system_proxy.host)
self.assertEqual(None, system_proxy.port)
self.assertFalse(system_proxy)
def test_from_url_empty(self):
system_proxy = platformsettings.SystemProxy.from_url('')
self.assertEqual(None, system_proxy.host)
self.assertEqual(None, system_proxy.port)
self.assertFalse(system_proxy)
def test_from_url_basic(self):
system_proxy = platformsettings.SystemProxy.from_url('http://pxy.com:8888/')
self.assertEqual('pxy.com', system_proxy.host)
self.assertEqual(8888, system_proxy.port)
self.assertTrue(system_proxy)
def test_from_url_no_port(self):
system_proxy = platformsettings.SystemProxy.from_url('http://pxy.com/')
self.assertEqual('pxy.com', system_proxy.host)
self.assertEqual(None, system_proxy.port)
self.assertTrue(system_proxy)
def test_from_url_empty_string(self):
system_proxy = platformsettings.SystemProxy.from_url('')
self.assertEqual(None, system_proxy.host)
self.assertEqual(None, system_proxy.port)
self.assertFalse(system_proxy)
def test_from_url_bad_string(self):
system_proxy = platformsettings.SystemProxy.from_url('foo:80')
self.assertEqual(None, system_proxy.host)
self.assertEqual(None, system_proxy.port)
self.assertFalse(system_proxy)
class HasSniTest(unittest.TestCase):
def test_has_sni(self):
# Check that no exception is raised.
platformsettings.HasSniSupport()
# pylint: disable=abstract-method
class Win7Settings(platformsettings._WindowsPlatformSettings):
@classmethod
def _ipconfig(cls, *args):
if args == ('/all',):
return WINDOWS_7_IPCONFIG
raise RuntimeError
class WinXpSettings(platformsettings._WindowsPlatformSettings):
@classmethod
def _ipconfig(cls, *args):
if args == ('/all',):
return WINDOWS_XP_IPCONFIG
raise RuntimeError
class WindowsPlatformSettingsTest(unittest.TestCase):
def test_get_mac_address_xp(self):
self.assertEqual(WINDOWS_XP_MAC,
WinXpSettings()._get_mac_address(WINDOWS_XP_IP))
def test_get_mac_address_7(self):
self.assertEqual(WINDOWS_7_MAC,
Win7Settings()._get_mac_address(WINDOWS_7_IP))
class OsxSettings(platformsettings._OsxPlatformSettings):
def __init__(self):
super(OsxSettings, self).__init__()
self.ipv4_state = OSX_IPV4_STATE
self.dns_state = None # varies by test
def _scutil(self, cmd):
if cmd == 'show State:/Network/Global/IPv4':
return self.ipv4_state
elif cmd.startswith('show State:/Network/Service/'):
return self.dns_state
raise RuntimeError("Unrecognized cmd: %s", cmd)
class OsxPlatformSettingsTest(unittest.TestCase):
def setUp(self):
self.settings = OsxSettings()
def test_get_primary_nameserver_lion(self):
self.settings.dns_state = OSX_DNS_STATE_LION
self.assertEqual('172.72.255.1', self.settings._get_primary_nameserver())
def test_get_primary_nameserver_snow_leopard(self):
self.settings.dns_state = OSX_DNS_STATE_SNOW_LEOPARD
self.assertEqual('172.27.1.1', self.settings._get_primary_nameserver())
def test_get_primary_nameserver_unexpected_ipv4_state_raises(self):
self.settings.ipv4_state = 'Some error'
self.settings.dns_state = OSX_DNS_STATE_SNOW_LEOPARD
self.assertRaises(platformsettings.DnsReadError,
self.settings._get_primary_nameserver)
def test_get_primary_nameserver_unexpected_dns_state_raises(self):
self.settings.dns_state = 'Some other error'
self.assertRaises(platformsettings.DnsReadError,
self.settings._get_primary_nameserver)
if __name__ == '__main__':
unittest.main()