コード例 #1
0
 def test_archiving_nodes_not_added_to_search_on_archive_incomplete(self, mock_send, mock_update_search):
     proj = factories.ProjectFactory()
     reg = factories.RegistrationFactory(project=proj)
     reg.save()
     with mock.patch('website.archiver.model.ArchiveJob.archive_tree_finished', mock.Mock(return_value=False)):
         listeners.archive_callback(reg)
     assert_false(mock_update_search.called)
コード例 #2
0
ファイル: test_archiver.py プロジェクト: ubiquitypress/osf.io
 def test_archive_callback_done_errors(self):
     self.dst.archive_job.update_target('dropbox', ARCHIVER_SUCCESS)
     self.dst.archive_job.update_target('osfstorage', ARCHIVER_FAILURE)
     self.dst.archive_job.save()
     with mock.patch('website.archiver.utils.handle_archive_fail') as mock_fail:
         listeners.archive_callback(self.dst)
     assert(mock_fail.called_with(ARCHIVER_NETWORK_ERROR, self.src, self.dst, self.user, self.dst.archive_job.target_addons))
コード例 #3
0
ファイル: test_archiver.py プロジェクト: bdyetton/osf.io
 def test_archive_callback_done_errors(self):
     self.dst.archive_job.update_target('dropbox', ARCHIVER_SUCCESS)
     self.dst.archive_job.update_target('osfstorage', ARCHIVER_FAILURE)
     self.dst.archive_job.save()
     with mock.patch('website.archiver.utils.handle_archive_fail') as mock_fail:
         listeners.archive_callback(self.dst)
     assert(mock_fail.called_with(ARCHIVER_NETWORK_ERROR, self.src, self.dst, self.user, self.dst.archive_job.target_addons))
コード例 #4
0
ファイル: test_archiver.py プロジェクト: wearpants/osf.io
 def test_archive_callback_done_success(self, mock_send,
                                        mock_archive_success):
     for addon in ['osfstorage', 'dropbox']:
         self.dst.archive_job.update_target(addon, ARCHIVER_SUCCESS)
     self.dst.archive_job.save()
     listeners.archive_callback(self.dst)
     mock_send.assert_called()
コード例 #5
0
ファイル: test_archiver.py プロジェクト: atelic/osf.io
 def test_archiving_nodes_not_added_to_search_on_archive_incomplete(self, mock_send, mock_update_search):
     proj = factories.ProjectFactory()
     reg = factories.RegistrationFactory(project=proj)
     reg.save()
     with mock.patch('website.archiver.model.ArchiveJob.archive_tree_finished', mock.Mock(return_value=False)):
         listeners.archive_callback(reg)
     assert_false(mock_update_search.called)
コード例 #6
0
ファイル: test_archiver.py プロジェクト: adlius/osf.io
 def test_archiving_nodes_not_added_to_search_on_archive_failure(self, mock_send, mock_delete_index_node):
     proj = factories.ProjectFactory()
     reg = factories.RegistrationFactory(project=proj)
     reg.save()
     with nested(
             mock.patch('osf.models.archive.ArchiveJob.archive_tree_finished', mock.Mock(return_value=True)),
             mock.patch('osf.models.archive.ArchiveJob.success', mock.PropertyMock(return_value=False))
     ) as (mock_finished, mock_success):
         listeners.archive_callback(reg)
     assert_true(mock_delete_index_node.called)
コード例 #7
0
ファイル: test_archiver.py プロジェクト: adlius/osf.io
 def test_archiving_nodes_added_to_search_on_archive_success_if_public(self, mock_update_search, mock_send, mock_archive_success):
     proj = factories.ProjectFactory()
     reg = factories.RegistrationFactory(project=proj)
     reg.save()
     with nested(
         mock.patch('osf.models.ArchiveJob.archive_tree_finished', mock.Mock(return_value=True)),
         mock.patch('osf.models.ArchiveJob.success', mock.PropertyMock(return_value=True))
     ) as (mock_finished, mock_success):
         listeners.archive_callback(reg)
     assert_equal(mock_update_search.call_count, 1)
コード例 #8
0
ファイル: test_archiver.py プロジェクト: bdyetton/osf.io
 def test_archive_callback_updates_archiving_state_when_done(self):
     proj = factories.NodeFactory()
     factories.NodeFactory(parent=proj)
     reg = factories.RegistrationFactory(project=proj)
     reg.archive_job.update_target('osfstorage', ARCHIVER_INITIATED)
     child = reg.nodes[0]
     child.archive_job.update_target('osfstorage', ARCHIVER_SUCCESS)
     child.save()
     listeners.archive_callback(child)
     assert_false(child.archiving)
コード例 #9
0
 def test_archive_callback_updates_archiving_state_when_done(self):
     proj = factories.NodeFactory()
     factories.NodeFactory(parent=proj)
     reg = factories.RegistrationFactory(project=proj)
     reg.archive_job.update_target('osfstorage', ARCHIVER_INITIATED)
     child = reg.nodes[0]
     child.archive_job.update_target('osfstorage', ARCHIVER_SUCCESS)
     child.save()
     listeners.archive_callback(child)
     assert_false(child.archiving)
コード例 #10
0
 def test_archiving_nodes_not_added_to_search_on_archive_failure(self, mock_send, mock_delete_index_node):
     proj = factories.ProjectFactory()
     reg = factories.RegistrationFactory(project=proj)
     reg.save()
     with nested(
             mock.patch('osf.models.archive.ArchiveJob.archive_tree_finished', mock.Mock(return_value=True)),
             mock.patch('osf.models.archive.ArchiveJob.success', mock.PropertyMock(return_value=False))
     ) as (mock_finished, mock_success):
         listeners.archive_callback(reg)
     assert_true(mock_delete_index_node.called)
コード例 #11
0
 def test_archiving_nodes_added_to_search_on_archive_success_if_public(self, mock_update_search, mock_send, mock_archive_success):
     proj = factories.ProjectFactory()
     reg = factories.RegistrationFactory(project=proj)
     reg.save()
     with nested(
         mock.patch('osf.models.ArchiveJob.archive_tree_finished', mock.Mock(return_value=True)),
         mock.patch('osf.models.ArchiveJob.success', mock.PropertyMock(return_value=True))
     ) as (mock_finished, mock_success):
         listeners.archive_callback(reg)
     assert_equal(mock_update_search.call_count, 1)
コード例 #12
0
ファイル: utils.py プロジェクト: Alpani/osf.io
def mock_archive(project, schema=None, auth=None, data=None, parent=None,
                 autocomplete=True, autoapprove=False):
    """ A context manager for registrations. When you want to call Node#register_node in
    a test but do not want to deal with any of this side effects of archiver, this
    helper allows for creating a registration in a safe fashion.

    :param bool autocomplete: automatically finish archival?
    :param bool autoapprove: automatically approve registration approval?

    Example use:

    project = ProjectFactory()
    with mock_archive(project) as registration:
        assert_true(registration.is_registration)
        assert_true(registration.archiving)
        assert_true(registration.is_pending_registration)

    with mock_archive(project, autocomplete=True) as registration:
        assert_true(registration.is_registration)
        assert_false(registration.archiving)
        assert_true(registration.is_pending_registration)

    with mock_archive(project, autocomplete=True, autoapprove=True) as registration:
        assert_true(registration.is_registration)
        assert_false(registration.archiving)
        assert_false(registration.is_pending_registration)
    """
    schema = schema or DEFAULT_METASCHEMA
    auth = auth or Auth(project.creator)
    data = data or ''

    with mock.patch('framework.tasks.handlers.enqueue_task'):
        registration = project.register_node(
            schema=schema,
            auth=auth,
            data=data,
            parent=parent,
        )
    registration.root.require_approval(project.creator)
    if autocomplete:
        root_job = registration.root.archive_job
        root_job.status = ARCHIVER_SUCCESS
        root_job.sent = False
        root_job.done = True
        root_job.save()
        sanction = registration.root.sanction
        with contextlib.nested(
            mock.patch.object(root_job, 'archive_tree_finished', mock.Mock(return_value=True)),
            mock.patch.object(sanction, 'ask')
        ):
            archiver_listeners.archive_callback(registration)
    if autoapprove:
        sanction = registration.root.sanction
        sanction._on_complete(project.creator)
    yield registration
コード例 #13
0
ファイル: test_archiver.py プロジェクト: rmoorman/osf.io
 def test_archive_callback_pending(self):
     for addon in ['osfstorage', 'dropbox']:
         self.archive_job.update_target(addon, ARCHIVER_INITIATED)
     self.dst.archive_job.update_target('osfstorage', ARCHIVER_SUCCESS)
     self.dst.archive_job.save()
     with mock.patch('website.mails.send_mail') as mock_send:
         with mock.patch(
                 'website.archiver.utils.handle_archive_fail') as mock_fail:
             listeners.archive_callback(self.dst)
     assert_false(mock_send.called)
     assert_false(mock_fail.called)
コード例 #14
0
ファイル: test_archiver.py プロジェクト: Alpani/osf.io
 def test_archiving_nodes_not_added_to_search_on_archive_failure(self, mock_send, mock_update_search):
     proj = factories.ProjectFactory()
     reg = factories.RegistrationFactory(project=proj)
     reg.save()
     with nested(
             mock.patch('website.archiver.model.ArchiveJob.archive_tree_finished', mock.Mock(return_value=True)),
             mock.patch('website.archiver.model.ArchiveJob.sent', mock.PropertyMock(return_value=False)),
             mock.patch('website.archiver.model.ArchiveJob.success', mock.PropertyMock(return_value=False))
     ) as (mock_finished, mock_sent, mock_success):
         listeners.archive_callback(reg)
     mock_update_search.assert_not_called()
コード例 #15
0
ファイル: test_archiver.py プロジェクト: ubiquitypress/osf.io
 def test_archiving_nodes_not_added_to_search_on_archive_failure(self, mock_send, mock_update_search):
     proj = factories.ProjectFactory()
     reg = factories.RegistrationFactory(project=proj)
     reg.save()
     with nested(
             mock.patch('website.archiver.model.ArchiveJob.archive_tree_finished', mock.Mock(return_value=True)),
             mock.patch('website.archiver.model.ArchiveJob.sent', mock.PropertyMock(return_value=False)),
             mock.patch('website.archiver.model.ArchiveJob.success', mock.PropertyMock(return_value=False))
     ) as (mock_finished, mock_sent, mock_success):
         listeners.archive_callback(reg)
     mock_update_search.assert_not_called()
コード例 #16
0
ファイル: test_archiver.py プロジェクト: adlius/osf.io
 def test_archive_callback_done_embargoed(self, mock_send, mock_archive_success):
     end_date = timezone.now() + datetime.timedelta(days=30)
     self.dst.archive_job.meta = {
         'embargo_urls': {
             contrib._id: None
             for contrib in self.dst.contributors
         }
     }
     self.dst.embargo_registration(self.user, end_date)
     self.dst.archive_job.update_target('osfstorage', ARCHIVER_SUCCESS)
     self.dst.save()
     listeners.archive_callback(self.dst)
     assert_equal(mock_send.call_count, 1)
コード例 #17
0
 def test_archive_callback_done_embargoed(self, mock_send, mock_archive_success):
     end_date = timezone.now() + datetime.timedelta(days=30)
     self.dst.archive_job.meta = {
         'embargo_urls': {
             contrib._id: None
             for contrib in self.dst.contributors
         }
     }
     self.dst.embargo_registration(self.user, end_date)
     self.dst.archive_job.update_target('osfstorage', ARCHIVER_SUCCESS)
     self.dst.save()
     listeners.archive_callback(self.dst)
     assert_equal(mock_send.call_count, 1)
コード例 #18
0
ファイル: test_archiver.py プロジェクト: rmoorman/osf.io
 def test_archive_callback_done_embargoed(self, mock_send):
     end_date = datetime.datetime.now() + datetime.timedelta(days=30)
     self.dst.archive_job.meta = {
         'embargo_urls':
         {contrib._id: None
          for contrib in self.dst.contributors}
     }
     self.dst.embargo_registration(self.user, end_date)
     for addon in ['osfstorage', 'dropbox']:
         self.dst.archive_job.update_target(addon, ARCHIVER_SUCCESS)
     self.dst.save()
     listeners.archive_callback(self.dst)
     mock_send.assert_called()
コード例 #19
0
def mock_archive(project, schema=None, auth=None, template=None, data=None, parent=None,
                 autocomplete=True, autoapprove=False):
    """ A context manager for registrations. When you want to call Node#register_node in
    a test but do not want to deal with any of this side effects of archiver, this
    helper allows for creating a registration in a safe fashion.

    :param bool autocomplete: automatically finish archival?
    :param bool autoapprove: automatically approve registration approval?

    Example use:

    project = ProjectFactory()
    with mock_archive(project) as registration:
        assert_true(registration.is_registration)
        assert_true(registration.archiving)
        assert_true(registration.is_pending_registration)

    with mock_archive(project, autocomplete=True) as registration:
        assert_true(registration.is_registration)
        assert_false(registration.archiving)
        assert_true(registration.is_pending_registration)

    with mock_archive(project, autocomplete=True, autoapprove=True) as registration:
        assert_true(registration.is_registration)
        assert_false(registration.archiving)
        assert_false(registration.is_pending_registration)
    """
    schema = schema or None
    auth = auth or Auth(project.creator)
    template = template or ''
    data = data or ''

    with mock.patch('framework.tasks.handlers.enqueue_task'):
        registration = project.register_node(schema, auth, template, data, parent)
    registration.root.require_approval(project.creator)
    if autocomplete:
        root_job = registration.root.archive_job
        root_job.status = ARCHIVER_SUCCESS
        root_job.sent = False
        root_job.done = True
        root_job.save()
        sanction = registration.root.sanction
        with contextlib.nested(
            mock.patch.object(root_job, 'archive_tree_finished', mock.Mock(return_value=True)),
            mock.patch.object(sanction, 'ask')
        ):
            archiver_listeners.archive_callback(registration)
    if autoapprove:
        sanction = registration.root.sanction
        sanction._on_complete(project.creator)
    yield registration
コード例 #20
0
ファイル: test_archiver.py プロジェクト: bdyetton/osf.io
 def test_archive_callback_done_embargoed(self, mock_send):
     end_date = datetime.datetime.now() + datetime.timedelta(days=30)
     self.dst.archive_job.meta = {
         'embargo_urls': {
             contrib._id: None
             for contrib in self.dst.contributors
         }
     }
     self.dst.embargo_registration(self.user, end_date)
     for addon in ['osfstorage', 'dropbox']:
         self.dst.archive_job.update_target(addon, ARCHIVER_SUCCESS)
     self.dst.save()
     listeners.archive_callback(self.dst)
     mock_send.assert_called_with(self.dst, self.user, urls=None)
コード例 #21
0
ファイル: test_archiver.py プロジェクト: adlius/osf.io
 def test_archive_callback_pending(self, mock_delay):
     self.archive_job.update_target(
         'osfstorage',
         ARCHIVER_INITIATED
     )
     self.dst.archive_job.update_target(
         'osfstorage',
         ARCHIVER_SUCCESS
     )
     self.dst.archive_job.save()
     with mock.patch('website.mails.send_mail') as mock_send:
         with mock.patch('website.archiver.utils.handle_archive_fail') as mock_fail:
             listeners.archive_callback(self.dst)
     assert_false(mock_send.called)
     assert_false(mock_fail.called)
     assert_true(mock_delay.called)
コード例 #22
0
ファイル: test_archiver.py プロジェクト: bdyetton/osf.io
 def test_archive_callback_pending(self):
     for addon in ['osfstorage', 'dropbox']:
         self.archive_job.update_target(
             addon,
             ARCHIVER_INITIATED
         )
     self.dst.archive_job.update_target(
         'osfstorage',
         ARCHIVER_SUCCESS
     )
     self.dst.archive_job.save()
     with mock.patch('website.archiver.utils.send_archiver_success_mail') as mock_send:
         with mock.patch('website.archiver.utils.handle_archive_fail') as mock_fail:
             listeners.archive_callback(self.dst)
     assert_false(mock_send.called)
     assert_false(mock_fail.called)
コード例 #23
0
 def test_archive_callback_on_tree_sends_only_one_email(
         self, mock_send_success, mock_arhive_success):
     proj = factories.NodeFactory()
     child = factories.NodeFactory(parent=proj)
     factories.NodeFactory(parent=child)
     reg = factories.RegistrationFactory(project=proj)
     rchild = reg._nodes.first()
     rchild2 = rchild._nodes.first()
     for node in [reg, rchild, rchild2]:
         node.archive_job._set_target('osfstorage')
     for node in [reg, rchild, rchild2]:
         node.archive_job.update_target('osfstorage', ARCHIVER_INITIATED)
     rchild.archive_job.update_target('osfstorage', ARCHIVER_SUCCESS)
     rchild.save()
     listeners.archive_callback(rchild)
     assert_false(mock_send_success.called)
     reg.archive_job.update_target('osfstorage', ARCHIVER_SUCCESS)
     reg.save()
     listeners.archive_callback(reg)
     assert_false(mock_send_success.called)
     rchild2.archive_job.update_target('osfstorage', ARCHIVER_SUCCESS)
     rchild2.save()
     listeners.archive_callback(rchild2)
     assert_equal(mock_send_success.call_count, 1)
     assert_true(mock_send_success.called)
コード例 #24
0
ファイル: test_archiver.py プロジェクト: ubiquitypress/osf.io
 def test_archive_callback_on_tree_sends_only_one_email(self, mock_send_success, mock_arhive_success):
     proj = factories.NodeFactory()
     child = factories.NodeFactory(parent=proj)
     factories.NodeFactory(parent=child)
     reg = factories.RegistrationFactory(project=proj)
     rchild = reg.nodes[0]
     rchild2 = rchild.nodes[0]
     for node in [reg, rchild, rchild2]:
         for addon in ['dropbox', 'osfstorage']:
             node.archive_job._set_target(addon)
     for node in [reg, rchild, rchild2]:
         for addon in ['dropbox', 'osfstorage']:
             node.archive_job.update_target(addon, ARCHIVER_INITIATED)
     for addon in ['dropbox', 'osfstorage']:
         rchild.archive_job.update_target(addon, ARCHIVER_SUCCESS)
     rchild.save()
     listeners.archive_callback(rchild)
     mock_send_success.assert_not_called()
     for addon in ['dropbox', 'osfstorage']:
         reg.archive_job.update_target(addon, ARCHIVER_SUCCESS)
     reg.save()
     listeners.archive_callback(reg)
     mock_send_success.assert_not_called()
     for addon in ['dropbox', 'osfstorage']:
         rchild2.archive_job.update_target(addon, ARCHIVER_SUCCESS)
     rchild2.save()
     listeners.archive_callback(rchild2)
     mock_send_success.assert_called()
コード例 #25
0
ファイル: test_archiver.py プロジェクト: bdyetton/osf.io
 def test_archive_callback_on_tree_sends_only_one_email(self, mock_send_success):
     proj = factories.NodeFactory()
     child = factories.NodeFactory(parent=proj)
     factories.NodeFactory(parent=child)
     reg = factories.RegistrationFactory(project=proj)
     rchild = reg.nodes[0]
     rchild2 = rchild.nodes[0]
     for node in [reg, rchild, rchild2]:
         for addon in ['dropbox', 'osfstorage']:
             node.archive_job._set_target(addon)
     for node in [reg, rchild, rchild2]:
         for addon in ['dropbox', 'osfstorage']:
             node.archive_job.update_target(addon, ARCHIVER_INITIATED)
     for addon in ['dropbox', 'osfstorage']:
         rchild.archive_job.update_target(addon, ARCHIVER_SUCCESS)
     rchild.save()
     listeners.archive_callback(rchild)
     mock_send_success.assert_not_called()
     for addon in ['dropbox', 'osfstorage']:
         reg.archive_job.update_target(addon, ARCHIVER_SUCCESS)
     reg.save()
     listeners.archive_callback(reg)
     mock_send_success.assert_not_called()
     for addon in ['dropbox', 'osfstorage']:
         rchild2.archive_job.update_target(addon, ARCHIVER_SUCCESS)
     rchild2.save()
     listeners.archive_callback(rchild2)
     mock_send_success.assert_called_with(reg)
コード例 #26
0
ファイル: test_archiver.py プロジェクト: adlius/osf.io
 def test_archive_callback_on_tree_sends_only_one_email(self, mock_send_success, mock_arhive_success):
     proj = factories.NodeFactory()
     child = factories.NodeFactory(parent=proj)
     factories.NodeFactory(parent=child)
     reg = factories.RegistrationFactory(project=proj)
     rchild = reg._nodes.first()
     rchild2 = rchild._nodes.first()
     for node in [reg, rchild, rchild2]:
         node.archive_job._set_target('osfstorage')
     for node in [reg, rchild, rchild2]:
         node.archive_job.update_target('osfstorage', ARCHIVER_INITIATED)
     rchild.archive_job.update_target('osfstorage', ARCHIVER_SUCCESS)
     rchild.save()
     listeners.archive_callback(rchild)
     assert_false(mock_send_success.called)
     reg.archive_job.update_target('osfstorage', ARCHIVER_SUCCESS)
     reg.save()
     listeners.archive_callback(reg)
     assert_false(mock_send_success.called)
     rchild2.archive_job.update_target('osfstorage', ARCHIVER_SUCCESS)
     rchild2.save()
     listeners.archive_callback(rchild2)
     assert_equal(mock_send_success.call_count, 1)
     assert_true(mock_send_success.called)
コード例 #27
0
ファイル: test_archiver.py プロジェクト: bdyetton/osf.io
 def test_archive_callback_done_success(self, mock_send):
     for addon in ['osfstorage', 'dropbox']:
         self.dst.archive_job.update_target(addon, ARCHIVER_SUCCESS)
     self.dst.archive_job.save()
     listeners.archive_callback(self.dst)
     mock_send.assert_called_with(self.dst)
コード例 #28
0
ファイル: utils.py プロジェクト: adlius/osf.io
def mock_archive(project, schema=None, auth=None, data=None, parent=None,
                 embargo=False, embargo_end_date=None,
                 retraction=False, justification=None, autoapprove_retraction=False,
                 autocomplete=True, autoapprove=False):
    """ A context manager for registrations. When you want to call Node#register_node in
    a test but do not want to deal with any of this side effects of archiver, this
    helper allows for creating a registration in a safe fashion.

    :param bool embargo: embargo the registration (rather than RegistrationApproval)
    :param bool autocomplete: automatically finish archival?
    :param bool autoapprove: automatically approve registration approval?
    :param bool retraction: retract the registration?
    :param str justification: a justification for the retraction
    :param bool autoapprove_retraction: automatically approve retraction?

    Example use:

    project = ProjectFactory()
    with mock_archive(project) as registration:
        assert_true(registration.is_registration)
        assert_true(registration.archiving)
        assert_true(registration.is_pending_registration)

    with mock_archive(project, autocomplete=True) as registration:
        assert_true(registration.is_registration)
        assert_false(registration.archiving)
        assert_true(registration.is_pending_registration)

    with mock_archive(project, autocomplete=True, autoapprove=True) as registration:
        assert_true(registration.is_registration)
        assert_false(registration.archiving)
        assert_false(registration.is_pending_registration)
    """
    schema = schema or get_default_metaschema()
    auth = auth or Auth(project.creator)
    data = data or ''

    with mock.patch('framework.celery_tasks.handlers.enqueue_task'):
        registration = project.register_node(
            schema=schema,
            auth=auth,
            data=data,
            parent=parent,
        )
    if embargo:
        embargo_end_date = embargo_end_date or (
            timezone.now() + dt.timedelta(days=20)
        )
        registration.embargo_registration(
            project.creator,
            embargo_end_date
        )
    else:
        registration.require_approval(project.creator)
    if autocomplete:
        root_job = registration.archive_job
        root_job.status = ARCHIVER_SUCCESS
        root_job.sent = False
        root_job.done = True
        root_job.save()
        sanction = registration.sanction
        with contextlib.nested(
            mock.patch.object(root_job, 'archive_tree_finished', mock.Mock(return_value=True)),
            mock.patch('website.archiver.tasks.archive_success.delay', mock.Mock())
        ):
            archiver_listeners.archive_callback(registration)
    if autoapprove:
        sanction = registration.sanction
        sanction.state = Sanction.APPROVED
        sanction.save()
        sanction._on_complete(project.creator)
        sanction.save()

    if retraction:
        justification = justification or 'Because reasons'
        registration.refresh_from_db()
        retraction = registration.retract_registration(project.creator, justification=justification)
        if autoapprove_retraction:
            retraction.state = Sanction.APPROVED
            retraction._on_complete(project.creator)
        retraction.save()
        registration.save()
    yield registration
コード例 #29
0
 def test_archive_callback_done_success(self, mock_send, mock_archive_success):
     self.dst.archive_job.update_target('osfstorage', ARCHIVER_SUCCESS)
     self.dst.archive_job.save()
     listeners.archive_callback(self.dst)
     assert_equal(mock_send.call_count, 1)
コード例 #30
0
def mock_archive(project,
                 schema=None,
                 auth=None,
                 data=None,
                 parent=None,
                 embargo=False,
                 embargo_end_date=None,
                 retraction=False,
                 justification=None,
                 autoapprove_retraction=False,
                 autocomplete=True,
                 autoapprove=False,
                 provider=None):
    """ A context manager for registrations. When you want to call Node#register_node in
    a test but do not want to deal with any of this side effects of archiver, this
    helper allows for creating a registration in a safe fashion.

    :param bool embargo: embargo the registration (rather than RegistrationApproval)
    :param bool autocomplete: automatically finish archival?
    :param bool autoapprove: automatically approve registration approval?
    :param bool retraction: retract the registration?
    :param str justification: a justification for the retraction
    :param bool autoapprove_retraction: automatically approve retraction?
    :param RegistrationProvider provider: provider to put the registration in

    Example use:

    project = ProjectFactory()
    with mock_archive(project) as registration:
        assert_true(registration.is_registration)
        assert_true(registration.archiving)
        assert_true(registration.is_pending_registration)

    with mock_archive(project, autocomplete=True) as registration:
        assert_true(registration.is_registration)
        assert_false(registration.archiving)
        assert_true(registration.is_pending_registration)

    with mock_archive(project, autocomplete=True, autoapprove=True) as registration:
        assert_true(registration.is_registration)
        assert_false(registration.archiving)
        assert_false(registration.is_pending_registration)
    """
    schema = schema or get_default_metaschema()
    auth = auth or Auth(project.creator)
    data = data or ''
    provider = provider or RegistrationProvider.objects.first(
    ) or RegistrationProviderFactory(_id='osf')

    with mock.patch('framework.celery_tasks.handlers.enqueue_task'):
        draft_reg = DraftRegistrationFactory(branched_from=project)
        registration = project.register_node(
            schema=schema,
            auth=auth,
            draft_registration=draft_reg,
            parent=parent,
            provider=provider,
        )
    if embargo:
        embargo_end_date = embargo_end_date or (timezone.now() +
                                                dt.timedelta(days=20))
        registration.embargo_registration(project.creator, embargo_end_date)
    else:
        registration.require_approval(project.creator)
    if autocomplete:
        root_job = registration.archive_job
        root_job.status = ARCHIVER_SUCCESS
        root_job.sent = False
        root_job.done = True
        root_job.save()
        sanction = registration.sanction
        mock.patch.object(root_job, 'archive_tree_finished',
                          mock.Mock(return_value=True)),
        mock.patch('website.archiver.tasks.archive_success.delay', mock.Mock())
        archiver_listeners.archive_callback(registration)

    if autoapprove:
        sanction = registration.sanction
        sanction.mode = Sanction.ANY
        sanction.approve(user=project.creator,
                         token=sanction.token_for_user(project.creator,
                                                       'approval'))

    if retraction:
        justification = justification or 'Because reasons'
        registration.refresh_from_db()
        retraction = registration.retract_registration(
            project.creator, justification=justification)
        if autoapprove_retraction:
            retraction.mode = Sanction.ANY
            retraction.approve(user=project.creator,
                               token=retraction.token_for_user(
                                   project.creator, 'approval'))
        registration.save()
    yield registration
コード例 #31
0
ファイル: utils.py プロジェクト: scooley/osf.io
def mock_archive(project,
                 schema=None,
                 auth=None,
                 data=None,
                 parent=None,
                 embargo=False,
                 embargo_end_date=None,
                 retraction=False,
                 justification=None,
                 autoapprove_retraction=False,
                 autocomplete=True,
                 autoapprove=False):
    """ A context manager for registrations. When you want to call Node#register_node in
    a test but do not want to deal with any of this side effects of archiver, this
    helper allows for creating a registration in a safe fashion.

    :param bool embargo: embargo the registration (rather than RegistrationApproval)
    :param bool autocomplete: automatically finish archival?
    :param bool autoapprove: automatically approve registration approval?
    :param bool retraction: retract the registration?
    :param str justification: a justification for the retraction
    :param bool autoapprove_retraction: automatically approve retraction?

    Example use:

    project = ProjectFactory()
    with mock_archive(project) as registration:
        assert_true(registration.is_registration)
        assert_true(registration.archiving)
        assert_true(registration.is_pending_registration)

    with mock_archive(project, autocomplete=True) as registration:
        assert_true(registration.is_registration)
        assert_false(registration.archiving)
        assert_true(registration.is_pending_registration)

    with mock_archive(project, autocomplete=True, autoapprove=True) as registration:
        assert_true(registration.is_registration)
        assert_false(registration.archiving)
        assert_false(registration.is_pending_registration)
    """
    schema = schema or DEFAULT_METASCHEMA
    auth = auth or Auth(project.creator)
    data = data or ''

    with mock.patch('framework.celery_tasks.handlers.enqueue_task'):
        registration = project.register_node(
            schema=schema,
            auth=auth,
            data=data,
            parent=parent,
        )
    if embargo:
        embargo_end_date = embargo_end_date or (datetime.datetime.now() +
                                                datetime.timedelta(days=20))
        registration.root.embargo_registration(project.creator,
                                               embargo_end_date)
    else:
        registration.root.require_approval(project.creator)
    if autocomplete:
        root_job = registration.root.archive_job
        root_job.status = ARCHIVER_SUCCESS
        root_job.sent = False
        root_job.done = True
        root_job.save()
        sanction = registration.root.sanction
        with contextlib.nested(
                mock.patch.object(root_job, 'archive_tree_finished',
                                  mock.Mock(return_value=True)),
                mock.patch('website.archiver.tasks.archive_success.delay',
                           mock.Mock())):
            archiver_listeners.archive_callback(registration)
    if autoapprove:
        sanction = registration.root.sanction
        sanction.state = Sanction.APPROVED
        sanction._on_complete(project.creator)
        sanction.save()

    if retraction:
        justification = justification or "Because reasons"
        retraction = registration.retract_registration(
            project.creator, justification=justification)
        if autoapprove_retraction:
            retraction.state = Sanction.APPROVED
            retraction._on_complete(project.creator)
        retraction.save()
        registration.save()
    yield registration
コード例 #32
0
ファイル: test_archiver.py プロジェクト: atelic/osf.io
 def test_archive_callback_done_success(self, mock_send, mock_archive_success):
     for addon in ['osfstorage', 'dropbox']:
         self.dst.archive_job.update_target(addon, ARCHIVER_SUCCESS)
     self.dst.archive_job.save()
     listeners.archive_callback(self.dst)
     assert_equal(mock_send.call_count, 1)