public void getSelectedColumnShouldReturnTheRightColumn() {
TreeViewerColumnSorter sorter = create(new TreeViewer(shell));
assertThat(sorter.getSelectedColumn(), nullValue());
TreeColumn column = new TreeColumn(sorter.getViewer().getTree(), SWT.NONE);
Event event = new Event();
event.widget = column;
SelectionEvent selectionEvent = new SelectionEvent(event);
sorter.widgetSelected(selectionEvent);
assertThat(sorter.getSelectedColumn(), sameInstance(column));