def test_turn_on_save(self): """ Tests turning on syncing for the save signal. """ turn_off_syncing() turn_on_syncing() with patch('entity.models.sync_entity_signal_handler') as mock_handler: Account.objects.create() self.assertTrue(mock_handler.called)
def test_turn_on_bulk(self): """ Tests turning on syncing for bulk operations. """ turn_off_syncing() turn_on_syncing() with patch('entity.models.sync_entities_signal_handler') as mock_handler: Account.objects.bulk_create([Account() for i in range(5)]) self.assertTrue(mock_handler.called)
def test_turn_on_delete(self): """ Tests turning on syncing for the delete signal. """ turn_off_syncing() turn_on_syncing() with patch('entity.models.sync_entity_signal_handler') as mock_handler: a = Account.objects.create() self.assertEquals(mock_handler.call_count, 1) a.delete() self.assertEquals(mock_handler.call_count, 2)
def tearDown(self): super(TestTurnOnOffSyncing, self).tearDown() # Make sure syncing is always on before every test turn_on_syncing()