def test_extra_data_from_selected(self):
"""Ensure that keys starting with _ are not displayed,
but the valuesa retrievable.
"""
dicts = [{
"key1_1": "val1_1",
"key1_2": "val1_2",
"__extra": ["boo", "biz", "baz"]
}, {
"key1_1": "val2_1",
"key1_2": "val2_2",
"__extra": self
}]
grid = DictionaryGrid(dicts)
#make sure there are 2 columns
self.assertEqual(len(grid.get_model()), 2)
#ensure that none of the columns are named _extra
cols = grid.get_columns()
for c in cols:
self.assertEqual(c.get_title().startswith("key"), True)
#select the first row
selection = grid.get_selection()
selection.select_path((0, ))
selected_dict = grid.selected_rows[0]
self.assertEqual(selected_dict["__extra"], ["boo", "biz", "baz"])
def test_remove_selected_rows(self):
dicts = [{
"key1_1": "val1_1",
"key1_2": "val1_2",
"key1_3": "val1_3"
}, {
"key1_1": "val2_1",
"key1_2": "val2_2",
"key1_3": "val2_3"
}, {
"key1_1": "val3_1",
"key1_2": "val3_2",
"key1_3": "val3_3"
}, {
"key1_1": "val4_1",
"key1_2": "val4_2",
"key1_3": "val4_3"
}, {
"key1_1": "val5_1",
"key1_2": "val5_2",
"key1_3": "val5_3"
}]
grid = DictionaryGrid(dicts)
selection = grid.get_selection()
grid.remove_selected_rows()
#select the last row and remove it
selection.select_path((4, ))
grid.remove_selected_rows()
#the last row should then be selected and there should be 4 rows now
self.assertEqual(len(grid.get_model()), 4)
self.assertEqual(len(grid.selected_rows), 1)
self.assertEqual(grid.selected_rows[0]["key1_1"], "val4_1")
#select the first and third rows and remove them
selection = grid.get_selection()
selection.unselect_all()
selection.select_path((0, ))
selection.select_path((2, ))
grid.remove_selected_rows()
#make sure the now last row is selected, and there are 2 rows left
self.assertEqual(len(grid.get_model()), 2)
self.assertEqual(len(grid.selected_rows), 1)
self.assertEqual(grid.selected_rows[0]["key1_2"], "val4_2")
def test_remove_selected_with_filter(self):
dicts = [{"key1_1": "val1_1", "key1_2": "val1_2", "key1_3": "val1_3"},
{"key1_1": "val2_1", "key1_2": "val2_2", "key1_3": "val2_3"},
{"key1_1": "val3_1", "key1_2": "val3_2", "key1_3": "val3_3"},
{"key1_1": "val4_1", "key1_2": "val4_2", "key1_3": "val4_3"},
{"key1_1": "val5_1", "key1_2": "val5_2", "key1_3": "val5_3"}]
grid = DictionaryGrid(dicts)
grid_filter = GridFilter(grid)
filter_row = grid_filter.rows[0]
filter_combo = filter_row.get_children()[1].get_children()[0].get_children()[0]
filter_combo.set_active(1)
entry = filter_row.get_children()[1].get_children()[0].get_children()[1]
entry.set_text("val5_1")
selection = grid.get_selection()
selection.select_path((1,))
selection.select_path((2,))
grid.remove_selected_rows()
self.assertEqual(len(grid.get_model()),2)
