def test_set_dueDate_reporting_auto(self):
self.prepare_mocked_contract()
manager = CeasefireMilestoneManager(Mock())
approval_milestone_mock = Mock()
approval_milestone_mock.type_ = 'approval'
approval_milestone_mock.get.return_value = 'approval'
approval_milestone_mock.dateMet = datetime.now()
reporting_milestone_mock = Mock()
reporting_milestone_mock.type_ = 'reporting'
reporting_milestone_mock.dueDate = None
self.contract.milestones = (approval_milestone_mock,
reporting_milestone_mock)
self.contract.dateSigned = datetime.now()
target_dueDate = datetime.combine(
date(
approval_milestone_mock.dateMet.year +
MILESTONE_REPORTING_DUEDATE_OFFSET_YEARS,
approval_milestone_mock.dateMet.month,
approval_milestone_mock.dateMet.day,
), approval_milestone_mock.dateMet.time())
manager.set_dueDate(reporting_milestone_mock, self.contract)
self.assertEqual(reporting_milestone_mock.dueDate, target_dueDate,
'dueDate has been calculated incorrectly')
def test_choose_status_to_partially_met(self):
manager = CeasefireMilestoneManager(Mock())
mocked_milestone = Mock()
mocked_milestone.dueDate = datetime.now()
# set `dateMet` after `dueDate` to acquire `partiallyMet` status
dateMet_to_set = mocked_milestone.dueDate + timedelta(days=2)
manager.choose_status(mocked_milestone, dateMet_to_set)
assert mocked_milestone.status == 'partiallyMet', 'Milestone status was not choosed correctly'
def test_create_milestones(self):
self.prepare_mocked_contract()
manager = CeasefireMilestoneManager(Mock())
self.contract.dateSigned = datetime.now()
manager.create_milestones(self.mocked_request)
self.assertEqual(len(self.contract.milestones), 3, 'milestones were not created')
def setUp(self):
self.manager = CeasefireMilestoneManager(Mock())
self.contract = Mock()
financing_milestone = MagicMock()
financing_milestone.type_ = financing_milestone.__getitem__.return_value = 'financing'
approval_milestone = MagicMock()
approval_milestone.type_ = approval_milestone.__getitem__.return_value = 'approval'
reporting_milestone = MagicMock()
reporting_milestone.type_ = reporting_milestone.__getitem__.return_value = 'reporting'
self.contract.milestones = [financing_milestone, approval_milestone, reporting_milestone]
def test_change_milestone_to_not_met(self):
manager = CeasefireMilestoneManager(Mock())
mocked_request = Mock()
mocked_milestone = Mock()
mocked_contract = Mock()
mocked_milestone.__parent__ = mocked_contract
mocked_milestone.status = 'processing'
mocked_request.context = mocked_milestone
mocked_request.validated = {'data': {}}
mocked_request.json = {'data': {'status': 'notMet'}}
manager.change_milestone(mocked_request)
self.assertEqual(mocked_milestone.status, 'notMet')
self.assertEqual(mocked_contract.status, 'unsuccessful')
def test_set_dueDate_financing(self):
self.prepare_mocked_contract()
manager = CeasefireMilestoneManager(Mock())
milestone_mock = Mock()
milestone_mock.type_ = 'financing'
self.contract.dateSigned = datetime.now()
target_dueDate = calculate_business_date(
self.contract.dateSigned,
MILESTONE_FINANCING_DUEDATE_OFFSET,
context=None,
working_days=True,
specific_hour=18)
manager.set_dueDate(milestone_mock, self.contract)
self.assertEqual(milestone_mock.dueDate, target_dueDate,
'dueDate has been calculated incorrectly')
def test_populate_milestones(self):
self.prepare_mocked_contract()
manager = CeasefireMilestoneManager(Mock())
self.contract.dateSigned = datetime.now()
milestones = manager.populate_milestones(self.contract)
self.assertEqual(len(milestones), 3, '3 milestones must be generated')
generated_types = [milestone.type_ for milestone in milestones]
self.assertEqual(set(MILESTONE_TYPES), set(generated_types),
'types of generated milestones are wrong')
financing_milestone = get_milestone_by_type(milestones, 'financing')
self.assertNotEqual(
financing_milestone.dueDate, None,
'dueDate of financing milestone has not been generated')
def test_set_dueDate_financing_with_accelerator(self):
self.prepare_mocked_contract()
manager = CeasefireMilestoneManager(Mock())
milestone_mock = Mock()
milestone_mock.type_ = 'financing'
if SANDBOX_MODE:
self.contract.sandbox_parameters = 'quick, accelerator=1440'
self.contract.dateSigned = datetime.now()
target_dueDate = calculate_business_date(
self.contract.dateSigned,
MILESTONE_FINANCING_DUEDATE_OFFSET,
context=self.contract,
working_days=True,
specific_hour=18)
manager.set_dueDate(milestone_mock, self.contract)
self.assertEqual(milestone_mock.dueDate, target_dueDate,
'dueDate has been calculated incorrectly')
def test_set_dueDate_approval(self):
self.prepare_mocked_contract()
manager = CeasefireMilestoneManager(Mock())
financing_milestone_mock = Mock()
financing_milestone_mock.type_ = 'financing'
financing_milestone_mock.get.return_value = 'financing'
financing_milestone_mock.dateMet = datetime.now()
approval_milestone_mock = Mock()
approval_milestone_mock.type_ = 'approval'
self.contract.milestones = (financing_milestone_mock, approval_milestone_mock)
self.contract.dateSigned = datetime.now()
target_dueDate = calculate_business_date(
financing_milestone_mock.dateMet,
MILESTONE_APPROVAL_DUEDATE_OFFSET,
context=None,
working_days=False,
specific_hour=18,
result_is_working_day=True)
manager.set_dueDate(approval_milestone_mock, self.contract)
self.assertEqual(approval_milestone_mock.dueDate, target_dueDate, 'dueDate has been calculated incorrectly')
class SetContractStatusTest(unittest.TestCase):
def setUp(self):
self.manager = CeasefireMilestoneManager(Mock())
self.contract = Mock()
financing_milestone = MagicMock()
financing_milestone.type_ = financing_milestone.__getitem__.return_value = 'financing'
approval_milestone = MagicMock()
approval_milestone.type_ = approval_milestone.__getitem__.return_value = 'approval'
reporting_milestone = MagicMock()
reporting_milestone.type_ = reporting_milestone.__getitem__.return_value = 'reporting'
self.contract.milestones = [
financing_milestone, approval_milestone, reporting_milestone
]
def test_set_by_parameters_table(self):
statuses_table = (
(('met', 'processing', 'scheduled'), 'active.approval'),
(('partiallyMet', 'processing', 'scheduled'), 'active.approval'),
(('met', 'met', 'processing'), 'active'),
(('met', 'partiallyMet', 'processing'), 'active'),
(('met', 'partiallyMet', 'met'), 'pending.terminated'),
(('met', 'partiallyMet', 'notMet'), 'pending.unsuccessful'),
(('notMet', 'partiallyMet', 'met'), 'pending.unsuccessful'),
)
for row in statuses_table:
input_statuses = row[0]
target_contract_status = row[1]
# init milestone statuses
for i, milestone in enumerate(self.contract.milestones):
milestone.status = input_statuses[i]
self.manager.contract_status_based_on_milestones(self.contract)
assert self.contract.status == target_contract_status
def test_change_milestone_to_not_met(self):
manager = CeasefireMilestoneManager(Mock())
mocked_request = Mock()
mocked_milestone = Mock()
mocked_milestone.id = '1'
mocked_contract = Mock()
doc_mock = Mock()
doc_mock.documentOf = 'milestone'
doc_mock.relatedItem = '1'
mocked_contract.documents = [doc_mock]
mocked_milestone.__parent__ = mocked_contract
mocked_milestone.status = 'processing'
mocked_request.context = mocked_milestone
mocked_request.validated = {'data': {}}
mocked_request.json = {'data': {'status': 'notMet'}}
manager.change_milestone(mocked_request)
self.assertEqual(mocked_milestone.status, 'notMet')
self.assertEqual(mocked_contract.status, 'pending.unsuccessful')
def change_contract(self, request):
new_status = request.validated['data'].get('status')
if new_status == 'active.payment':
milestone_manager = CeasefireMilestoneManager(request.context)
milestone_manager.create_milestones(request)