def test_setup_fields_processors(self, mock_eng, mock_resolve): from ramses import models config = Mock() schema = { 'properties': { 'stories': { "_db_settings": { "type": "relationship", "document": "Story", "backref_name": "owner", }, "_processors": ["lowercase"], "_backref_processors": ["backref_lowercase"] } } } models.setup_fields_processors(config, 'mymodel', schema) mock_resolve.assert_has_calls([ call('lowercase'), call('backref_lowercase')]) mock_eng.get_document_cls.assert_called_once_with('Story') config.add_field_processors.assert_has_calls([ call([mock_resolve()], model='mymodel', field='stories'), call([mock_resolve()], model=mock_eng.get_document_cls(), field='owner'), ])
def test_setup_fields_processors(self, mock_eng, mock_resolve): from ramses import models config = Mock() schema = { 'properties': { 'stories': { "_db_settings": { "type": "relationship", "document": "Story", "backref_name": "owner", }, "_processors": ["lowercase"], "_backref_processors": ["backref_lowercase"] } } } models.setup_fields_processors(config, 'mymodel', schema) mock_resolve.assert_has_calls( [call('lowercase'), call('backref_lowercase')]) mock_eng.get_document_cls.assert_called_once_with('Story') config.add_field_processors.assert_has_calls([ call([mock_resolve()], model='mymodel', field='stories'), call([mock_resolve()], model=mock_eng.get_document_cls(), field='owner'), ])
def test_setup_fields_processors_backref_no_backname( self, mock_eng, mock_resolve): from ramses import models config = Mock() schema = { 'properties': { 'stories': { "_db_settings": { "type": "relationship", "document": "Story", }, "_backref_processors": ["backref_lowercase"] } } } models.setup_fields_processors(config, 'mymodel', schema) assert not config.add_field_processors.called
def test_setup_fields_processors_backref_no_backname( self, mock_eng, mock_resolve): from ramses import models config = Mock() schema = { 'properties': { 'stories': { "_db_settings": { "type": "relationship", "document": "Story", }, "_backref_processors": ["backref_lowercase"] } } } models.setup_fields_processors(config, 'mymodel', schema) assert not config.add_field_processors.called