def test_after_register_does_not_archive_pointers(self, mock_chain, mock_archive): proj = factories.ProjectFactory(creator=self.user) c1 = factories.ProjectFactory(creator=self.user, parent=proj) other = factories.ProjectFactory(creator=self.user) reg = factories.RegistrationFactory(project=proj) r1 = reg._nodes.first() proj.add_pointer(other, auth=Auth(self.user)) listeners.after_register(c1, r1, self.user) listeners.after_register(proj, reg, self.user) for kwargs in [dict(job_pk=n.archive_job._id,) for n in [reg, r1]]: mock_archive.assert_any_call(**kwargs)
def test_after_register_does_not_archive_pointers(self, mock_chain, mock_archive): proj = factories.ProjectFactory(creator=self.user) c1 = factories.ProjectFactory(creator=self.user, parent=proj) other = factories.ProjectFactory(creator=self.user) reg = factories.RegistrationFactory(project=proj) r1 = reg.nodes[0] proj.add_pointer(other, auth=Auth(self.user)) listeners.after_register(c1, r1, self.user) listeners.after_register(proj, reg, self.user) for kwargs in [dict(job_pk=n.archive_job._id,) for n in [reg, r1]]: mock_archive.assert_any_call(**kwargs)
def test_after_register_does_not_archive_pointers(self, mock_chain): proj = factories.ProjectFactory(creator=self.user) c1 = factories.ProjectFactory(creator=self.user, parent=proj) other = factories.ProjectFactory(creator=self.user) reg = factories.RegistrationFactory(project=proj) r1 = reg.nodes[0] proj.add_pointer(other, auth=Auth(self.user)) listeners.after_register(c1, r1, self.user) listeners.after_register(proj, reg, self.user) archive_sigs = [archive.si(**kwargs) for kwargs in [dict(job_pk=n.archive_job._id,) for n in [reg, r1]]] mock_chain.assert_called_with(*archive_sigs)
def test_after_register_archive_runs_only_for_root(self, mock_chain): proj = factories.ProjectFactory() c1 = factories.ProjectFactory(parent=proj) c2 = factories.ProjectFactory(parent=c1) reg = factories.RegistrationFactory(project=proj) rc1 = reg.nodes[0] rc2 = rc1.nodes[0] listeners.after_register(c1, rc1, self.user) mock_chain.assert_not_called() listeners.after_register(c2, rc2, self.user) mock_chain.assert_not_called() listeners.after_register(proj, reg, self.user) archive_sigs = [archive.si(**kwargs) for kwargs in [dict(job_pk=n.archive_job._id,) for n in [reg, rc1, rc2]]] mock_chain.assert_called_with(*archive_sigs)
def test_after_register_archive_runs_only_for_root(self, mock_chain, mock_archive): proj = factories.ProjectFactory() c1 = factories.ProjectFactory(parent=proj) c2 = factories.ProjectFactory(parent=c1) reg = factories.RegistrationFactory(project=proj) rc1 = reg.nodes[0] rc2 = rc1.nodes[0] listeners.after_register(c1, rc1, self.user) mock_chain.assert_not_called() listeners.after_register(c2, rc2, self.user) mock_chain.assert_not_called() listeners.after_register(proj, reg, self.user) for kwargs in [dict(job_pk=n.archive_job._id,) for n in [reg, rc1, rc2]]: mock_archive.assert_any_call(**kwargs)
def test_after_register_archive_runs_only_for_root(self, mock_chain, mock_archive): proj = factories.ProjectFactory() c1 = factories.ProjectFactory(parent=proj) c2 = factories.ProjectFactory(parent=c1) reg = factories.RegistrationFactory(project=proj) rc1 = reg.nodes[0] rc2 = rc1.nodes[0] mock_chain.reset_mock() listeners.after_register(c1, rc1, self.user) assert_false(mock_chain.called) listeners.after_register(c2, rc2, self.user) assert_false(mock_chain.called) listeners.after_register(proj, reg, self.user) for kwargs in [dict(job_pk=n.archive_job._id,) for n in [reg, rc1, rc2]]: mock_archive.assert_any_call(**kwargs)
def test_after_register(self, mock_before_archive, mock_chain): mock_chain.return_value = celery.chain() listeners.after_register(self.src, self.dst, self.user) mock_before_archive.assert_called_with(self.dst, self.user) archive_signature = archive.si(job_pk=self.archive_job._id) mock_chain.assert_called_with(archive_signature)
def test_after_register(self, mock_before_archive, mock_archive): listeners.after_register(self.src, self.dst, self.user) mock_before_archive.assert_called_with(self.dst, self.user) mock_archive.assert_called_with(job_pk=self.archive_job._id)