def test_given_list_of_sort_columns_then_returns_columns_for_sorting(self): table = Mock() column = Mock() column2 = Mock() config = SearchConfig(table=table, columns={'column': column, 'column2': column2, 'column3': Mock()}, sort=['column', 'column2'], default_sort='column') result = config.column_for_sorting('column2') assert_that(result, equal_to(column2))
def test_given_list_of_sort_columns_then_returns_columns_for_sorting(self): table = Mock() column = Mock() column2 = Mock() config = SearchConfig(table=table, columns={ 'column': column, 'column2': column2, 'column3': Mock() }, sort=['column', 'column2'], default_sort='column') result = config.column_for_sorting('column2') assert_that(result, equal_to(column2))