def test_query_set_get_with_valid_filter_param(self): resource = AddressableUserQuerySetResource(mock_orm.QuerySet( User(pk=1, name='Alice', age=31), User(pk=2, name='Bob', age=20) )) resource.allow_unfiltered_query = False resource.filters = [ParameterizedFilter('name', 'name')] data = resource.get(mock_context(), _ParamsImpl(QueryDict('name=Alice'))) self.assertEqual(data['objects'][0]['name'], 'Alice')
def test_get_list(self): get = Mock(name='mock') get.getlist.return_value = ['bar', 'baz'] params = _ParamsImpl(get) result = params.get_list('foo') self.assertEqual(result, ['bar', 'baz']) get.getlist.assert_called_with('foo')
def test_query_set_get_with_invalid_filter_param(self): resource = AddressableUserQuerySetResource(mock_orm.QuerySet( User(pk=1, name='Alice', age=31), User(pk=2, name='Bob', age=20) )) resource.allow_unfiltered_query = False resource.filters = [ParameterizedFilter('name', 'name')] data = None with self.assertRaises(SavoryPieError): data = resource.get(mock_context(), _ParamsImpl(QueryDict('[object Object]=12345'))) self.assertEqual(data, None)
def test_parameters_get_default(self): params = _ParamsImpl({'key1': 'value1'}) self.assertEqual(params.get('some', 'other'), 'other')
def test_get_item(self): params = _ParamsImpl({'key1': 'value1', 'key2': 'value2'}) self.assertEqual(params['key1'], 'value1')
def test_contains(self): params = _ParamsImpl({'key1': 'value1', 'key2': 'value2'}) self.assertTrue('key1' in params) self.assertFalse('key3' in params)
def test_parameters_get_as_wrong_type(self): params = _ParamsImpl({'key1': 1}) self.assertEqual(params.get_as('key1', str, 'other'), '1')
def test_get_list_of(self): params = _ParamsImpl({'key1': [1]}) self.assertEqual(params.get_list_of('key1', str), ['1'])
def test_parameters_get_as_default(self): params = _ParamsImpl({'key1': 1}) self.assertEqual(params.get_as('key2', str, 'other'), 'other')
def test_get_list_of_not_found(self): params = _ParamsImpl({'key1': [1]}) self.assertEqual(params.get_list_of('key2', str), [])
def test_parameters_get_list(self): get = mock.Mock() get.getlist.return_value = ['value1'] params = _ParamsImpl(get) self.assertEqual(params.get_list('key1'), ['value1'])
def test_parameters_get_as(self): params = _ParamsImpl({'key1': 'value1'}) self.assertEqual(params.get_as('key1', str, 'other'), 'value1')
def test_keys(self): params = _ParamsImpl({'key1': 'value1', 'key2': 'value2'}) self.assertEqual(sorted(params.keys()), ['key1', 'key2'])
def test_init(self): params = _ParamsImpl('some var') self.assertEqual(params._GET, 'some var')