def test_after_import_gets_proc_unit(self):
        """ It should get unit for model """
        with mock.patch.multiple(
            self.unit, binder_for=mock.DEFAULT, unit_for=mock.DEFAULT
        ):
            self.unit.unit_for.side_effect = EndTestException
            with self.assertRaises(EndTestException):
                self.unit._after_import(self.record)
            self.unit.unit_for.assert_called_with(
                account_invoice_line.ProcurementOrderUnit,
                model='carepoint.procurement.order',
            )