def setUp(self): super(TestDraftRegistrationApprovals, self).setUp() self.approval = DraftRegistrationApproval( initiated_by=self.user, meta={'registration_choice': 'immediate'}) self.authorizer1 = AuthUserFactory() self.authorizer2 = AuthUserFactory() self.approval.save() self.draft.registration_schema = MetaSchema.find_one( Q('name', 'eq', 'Prereg Challenge') & Q('schema_version', 'eq', 2)) self.draft.approval = self.approval self.draft.save()
def test_on_complete_embargo_creates_registration_for_draft_initiator( self, mock_enquque): end_date = dt.datetime.now() + dt.timedelta(days=366) # <- leap year self.approval = DraftRegistrationApproval(initiated_by=self.user, meta={ 'registration_choice': 'embargo', 'embargo_end_date': end_date.isoformat() }) self.authorizer1 = AuthUserFactory() self.authorizer2 = AuthUserFactory() self.approval.save() self.draft.approval = self.approval self.draft.save() self.approval._on_complete(self.user) registered_node = self.draft.registered_node assert_is_not_none(registered_node) assert_true(registered_node.is_pending_embargo) assert_equal(registered_node.registered_user, self.draft.initiator)