def test_get_changes(self):
router = BaseModelRouter(self.connection)
router.serializer = Serializer(instance=TwoFieldModel())
current_state = {'changed': 'a', 'nochange': 123, 'id': 1}
past_state = {'changed': 'b', 'nochange': 123, 'id': 1}
changed_fields = router._get_changed_fields(current_state, past_state)
expected = ['changed']
self.assertListEqual(expected, changed_fields)
class TestBaseModelRouter(DragonTestCase):
def setUp(self):
self.router = BaseModelRouter(self.connection)
def test_on_error(self):
self.router.on_error({'key': 'error'})
actual = self.connection.last_message
expected = {'context': {'state': 'error'}, 'data': {'key': 'error'}}
self.assertDictEqual(actual, expected)
def setUp(self):
self.router = BaseModelRouter(self.connection)
