refactoring and fixes for Bug #11209
[wifi-assistant] / package / test / unit / gui / network_settings_view_test.py
1 from wifi_assistant.gui.network_settings_view import NetworkSettingsView
2 import gtk
3
4 import unittest
5 from unit.pie import *
6
7 class NetworkSettingsViewTest(unittest.TestCase):
8
9     def setUp(self):
10         self._list_store = gtk.ListStore('gboolean', str)
11         self._parent_container = Mock()
12         self._edit_callback = Mock()
13         
14         given(self._edit_callback).save().willReturn(True)
15         
16         self.testee = NetworkSettingsView(self._list_store, self._parent_container, self._edit_callback.save)
17
18
19     def test_bug_11209_addingAndRemovingNetworkSettingsChangesActiveWidget(self):
20         assert len(self._list_store) == 0
21         
22         self._replayMocks()
23         self._list_store.append([True, 'a row'])
24         
25         widget = self.testee._active_widget
26         verify(self._parent_container).add(widget)
27         
28         self._list_store.clear()
29         
30         verify(self._parent_container).remove(widget)
31         
32         
33     def _replayMocks(self):
34         self._parent_container.replay()
35         self._edit_callback.replay()
36
37         
38     
39 if __name__ == '__main__':
40     unittest.main()