def test_get_sort_keys_whitelists_multiple_keys(self): sort_keys = ['foo', 'bar', 'nope'] mapping = {'foo': 'Foo', 'bar': 'Bar'} filtered_keys = db_api._get_sort_keys(sort_keys, mapping) self.assertIn('Foo', filtered_keys) self.assertIn('Bar', filtered_keys) self.assertEqual(2, len(filtered_keys))
def test_get_sort_keys_whitelists_single_key(self): sort_key = 'foo' mapping = {'foo': 'Foo'} filtered_keys = db_api._get_sort_keys(sort_key, mapping) self.assertEqual(['Foo'], filtered_keys)
def test_get_sort_keys_returns_empty_list_if_no_keys(self): sort_keys = None mapping = {} filtered_keys = db_api._get_sort_keys(sort_keys, mapping) self.assertEqual([], filtered_keys)