def setUp(self):
super(TestNetworkManagerState, self).setUp()
self.network_manager = self.mocker.mock()
self.is_connected = self.mocker.replace(
'ubuntuone.networkstate.windows.is_machine_connected')
self.thread = self.mocker.mock()
self.cb = self.mocker.mock()
self.state = NetworkManagerState(self.cb)
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(
'ubuntuone.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)