def test_data_func(self):
iterator = self.model.append(None, ['/foo', 'Intel', 'Wireless'])
mock_cell = mock.Mock()
tv = nmwidgets.NetworkManagerTreeView()
tv.data_func(None, mock_cell, self.model, iterator, None)
mock_cell.set_property.assert_called_with('text', 'Intel Wireless')
i = self.model.append(iterator, ['Orange', True, 0])
tv.data_func(None, mock_cell, self.model, i, None)
mock_cell.set_property.assert_called_with('text', 'Orange')
def test_pixbuf_func(self):
iterator = self.model.append(None, ['/foo', 'Intel', 'Wireless'])
mock_cell = mock.Mock()
tv = nmwidgets.NetworkManagerTreeView()
tv.pixbuf_func(None, mock_cell, self.model, iterator, None)
mock_cell.set_property.assert_called_with('pixbuf', None)
# 0% strength, protected network
i = self.model.append(iterator, ['Orange', True, 0])
tv.pixbuf_func(None, mock_cell, self.model, i, None)
mock_cell.set_property.assert_called_with('pixbuf', tv.icons[5])
# 30% strength, protected network
self.model.set_value(i, 2, 30)
tv.pixbuf_func(None, mock_cell, self.model, i, None)
mock_cell.set_property.assert_called_with('pixbuf', tv.icons[6])
# 95% strength, unprotected network
self.model.set_value(i, 1, False)
self.model.set_value(i, 2, 95)
tv.pixbuf_func(None, mock_cell, self.model, i, None)
mock_cell.set_property.assert_called_with('pixbuf', tv.icons[4])
