/usr/lib/python2.7/dist-packages/ubuntu-sso-client/ubuntu_sso/networkstate/tests/test_windows.py is in python-ubuntu-sso-client.tests 13.10-0ubuntu6.
This file is owned by root:root, with mode 0o644.
The actual contents of the file can be viewed below.
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 | # -*- coding: utf-8 -*-
#
# Copyright 2011-2012 Canonical Ltd.
#
# This program is free software: you can redistribute it and/or modify it
# under the terms of the GNU General Public License version 3, as published
# by the Free Software Foundation.
#
# This program is distributed in the hope that it will be useful, but
# WITHOUT ANY WARRANTY; without even the implied warranties of
# MERCHANTABILITY, SATISFACTORY QUALITY, or FITNESS FOR A PARTICULAR
# PURPOSE. See the GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License along
# with this program. If not, see <http://www.gnu.org/licenses/>.
#
# In addition, as a special exception, the copyright holders give
# permission to link the code of portions of this program with the
# OpenSSL library under certain conditions as described in each
# individual source file, and distribute linked combinations
# including the two.
# You must obey the GNU General Public License in all respects
# for all of the code used other than OpenSSL. If you modify
# file(s) with this exception, you may extend this exception to your
# version of the file(s), but you are not obligated to do so. If you
# do not wish to do so, delete this exception statement from your
# version. If you delete this exception statement from all source
# files in the program, then also delete it here.
"""Tests for the network manager."""
from ctypes import windll
from mocker import MockerTestCase
from twisted.internet.defer import inlineCallbacks
from ubuntu_sso.tests import TestCase
from ubuntu_sso.networkstate import NetworkFailException
from ubuntu_sso.networkstate.windows import (
is_machine_connected,
NetworkManager,
NetworkManagerState,
ONLINE,
OFFLINE)
class TestNetworkManager(MockerTestCase):
"""Test he Network Manager."""
def setUp(self):
super(TestNetworkManager, self).setUp()
self.connection_info = self.mocker.mock()
self.connection_no_info = self.mocker.mock()
self.disconnected = self.mocker.mock()
self.manager = NetworkManager(self.connection_no_info,
self.connection_info, self.disconnected)
def test_connection_made(self):
"""Ensure db is called."""
self.connection_info()
self.mocker.replay()
self.manager.ConnectionMade()
def test_connection_made_no_cb(self):
"""Ensure db is called."""
self.manager.connected_cb_info = None
self.mocker.replay()
self.manager.ConnectionMade()
def test_connection_made_no_info(self):
"""Ensure db is called."""
self.connection_no_info()
self.mocker.replay()
self.manager.ConnectionMadeNoQOCInfo()
def test_connection_made_no_info_no_cb(self):
"""Ensure db is called."""
self.manager.connected_cb = None
self.mocker.replay()
self.manager.ConnectionMadeNoQOCInfo()
def test_disconnection(self):
"""Ensure db is called."""
self.disconnected()
self.mocker.replay()
self.manager.ConnectionLost()
def test_disconnection_no_cb(self):
"""Ensure db is called."""
self.manager.disconnected_cb = None
self.mocker.replay()
self.manager.ConnectionLost()
class TestNetworkManagerState(MockerTestCase):
"""Test the Network Manager State."""
def setUp(self):
super(TestNetworkManagerState, self).setUp()
self.network_manager = self.mocker.mock()
self.is_connected = self.mocker.replace(
'ubuntu_sso.networkstate.windows.is_machine_connected')
self.thread = self.mocker.mock()
self.cb = self.mocker.mock()
self.state = NetworkManagerState(self.cb)
def test_connection_made(self):
"""Test that the cb is actually called."""
self.cb(ONLINE)
self.mocker.replay()
self.state.connection_made()
def test_connection_lost(self):
"""Test that the cb is actually called."""
self.cb(OFFLINE)
self.mocker.replay()
self.state.connection_lost()
def test_find_online_state_not_connected(self):
"""Test that we do find the online state correctly."""
self.is_connected()
self.mocker.result(False)
self.cb(OFFLINE)
self.mocker.result(self.thread)
self.thread.daemon = True
self.thread.start()
self.mocker.replay()
self.state.find_online_state(listener=self.network_manager,
listener_thread=self.thread)
def test_find_online_state_connected(self):
"""Test that we do find the online state correctly."""
self.is_connected()
self.mocker.result(ONLINE)
self.cb(ONLINE)
self.mocker.result(self.thread)
self.thread.daemon = True
self.thread.start()
self.mocker.replay()
self.state.find_online_state(listener=self.network_manager,
listener_thread=self.thread)
class FakeWininet(object):
"""Fake wininet for windll."""
connection_state = -1
# pylint: disable=C0103
def InternetGetConnectedState(self, *args, **kwargs):
"""Fake InternetGetConnectedState function from wininet."""
return self.connection_state
# pylint: enable=C0103
class FakeWininetException(object):
"""Fake wininet for windll."""
connection_state = -1
# pylint: disable=C0103
def InternetGetConnectedState(self, *args, **kwargs):
"""Fake InternetGetConnectedState function from wininet."""
raise Exception()
# pylint: enable=C0103
class TestConnection(TestCase):
"""Test the state of the connection."""
@inlineCallbacks
def setUp(self):
"""Setup the mocker dbus object tree."""
yield super(TestConnection, self).setUp()
self.patch(windll, "wininet", FakeWininet())
@inlineCallbacks
def test_is_machine_connected_connected(self):
"""Fake the NetworkManagerState."""
self.patch(FakeWininet, "connection_state", 1)
result = yield is_machine_connected()
self.assertTrue(result)
@inlineCallbacks
def test_is_machine_connected_disconnected(self):
"""Fake the NetworkManagerState."""
self.patch(FakeWininet, "connection_state", 0)
result = yield is_machine_connected()
self.assertFalse(result)
@inlineCallbacks
def test_is_machine_connected_error(self):
"""Fake the NetworkManagerState."""
self.patch(windll, "wininet", FakeWininetException())
yield self.assertFailure(is_machine_connected(), NetworkFailException)
|