class TestWebhookRepository(Test):
payload = dict(foo='bar')
def setUp(self):
super(TestWebhookRepository, self).setUp()
self.webhook_repo = WebhookRepository(db)
TaskRunFactory.create()
webhook = Webhook(project_id=1, payload=self.payload)
self.webhook_repo.save(webhook)
def test_get_return_none_if_no_webhook(self):
"""Test get method returns None if there is no log with the
specified id."""
assert self.webhook_repo.get(2) is None
def test_get_return_obj(self):
"""Test get method returns obj."""
tmp = self.webhook_repo.get(1)
assert tmp.id == 1
assert tmp.project_id == 1
def test_get_by_return_obj(self):
"""Test get_by method returns obj."""
tmp = self.webhook_repo.get_by(project_id=1)
assert tmp.id == 1
assert tmp.project_id == 1
def test_filter_by_return_obj(self):
"""Test filter_by method returns obj."""
tmp = self.webhook_repo.filter_by(project_id=1)
assert len(tmp) == 1
tmp = tmp[0]
assert tmp.id == 1
assert tmp.project_id == 1
tmp = self.webhook_repo.filter_by(limit=1, offset=1, project_id=1)
assert len(tmp) == 0
def test_save(self):
"""Test save."""
webhook = self.webhook_repo.get(1)
assert webhook.id == 1
def test_save_fails_if_integrity_error(self):
"""Test save raises a DBIntegrityError if the instance to be saved lacks
a required value"""
wh = Webhook(project_id=None)
assert_raises(DBIntegrityError, self.webhook_repo.save, wh)
def test_save_only_saves_webhooks(self):
"""Test save raises a WrongObjectError when an object which is not
a Webhok instance is saved"""
bad_object = dict()
assert_raises(WrongObjectError, self.webhook_repo.save, bad_object)
class TestWebhookRepository(Test):
payload = dict(foo='bar')
@with_context
def setUp(self):
super(TestWebhookRepository, self).setUp()
self.webhook_repo = WebhookRepository(db)
TaskRunFactory.create()
webhook = Webhook(project_id=1, payload=self.payload)
self.webhook_repo.save(webhook)
@with_context
def test_get_return_none_if_no_webhook(self):
"""Test get method returns None if there is no log with the
specified id."""
assert self.webhook_repo.get(2) is None
@with_context
def test_get_return_obj(self):
"""Test get method returns obj."""
tmp = self.webhook_repo.get(1)
assert tmp.id == 1
assert tmp.project_id == 1
@with_context
def test_get_by_return_obj(self):
"""Test get_by method returns obj."""
tmp = self.webhook_repo.get_by(project_id=1)
assert tmp.id == 1
assert tmp.project_id == 1
@with_context
def test_filter_by_return_obj(self):
"""Test filter_by method returns obj."""
tmp = self.webhook_repo.filter_by(project_id=1)
assert len(tmp) == 1
tmp = tmp[0]
assert tmp.id == 1
assert tmp.project_id == 1
tmp = self.webhook_repo.filter_by(limit=1, offset=1, project_id=1)
assert len(tmp) == 0
@with_context
def test_save(self):
"""Test save."""
webhook = self.webhook_repo.get(1)
assert webhook.id == 1
@with_context
def test_save_fails_if_integrity_error(self):
"""Test save raises a DBIntegrityError if the instance to be saved lacks
a required value"""
wh = Webhook(project_id=None)
assert_raises(DBIntegrityError, self.webhook_repo.save, wh)
@with_context
def test_save_only_saves_webhooks(self):
"""Test save raises a WrongObjectError when an object which is not
a Webhok instance is saved"""
bad_object = dict()
assert_raises(WrongObjectError, self.webhook_repo.save, bad_object)
@with_context
@patch('pybossa.repositories.webhook_repository.clean_project')
def test_delete_entries_from_project(self, clean_project_mock):
"""Test delete entries from project works."""
project = ProjectFactory.create()
wh = Webhook(project_id=project.id)
self.webhook_repo.save(wh)
self.webhook_repo.delete_entries_from_project(project)
res = self.webhook_repo.get_by(project_id=project.id)
assert res is None
clean_project_mock.assert_called_with(project.id)
class TestWebhookRepository(Test):
payload = dict(foo='bar')
@with_context
def setUp(self):
super(TestWebhookRepository, self).setUp()
self.webhook_repo = WebhookRepository(db)
TaskRunFactory.create()
webhook = Webhook(project_id=1, payload=self.payload)
self.webhook_repo.save(webhook)
@with_context
def test_get_return_none_if_no_webhook(self):
"""Test get method returns None if there is no log with the
specified id."""
assert self.webhook_repo.get(2) is None
@with_context
def test_get_return_obj(self):
"""Test get method returns obj."""
tmp = self.webhook_repo.get(1)
assert tmp.id == 1
assert tmp.project_id == 1
@with_context
def test_get_by_return_obj(self):
"""Test get_by method returns obj."""
tmp = self.webhook_repo.get_by(project_id=1)
assert tmp.id == 1
assert tmp.project_id == 1
@with_context
def test_filter_by_return_obj(self):
"""Test filter_by method returns obj."""
tmp = self.webhook_repo.filter_by(project_id=1)
assert len(tmp) == 1
tmp = tmp[0]
assert tmp.id == 1
assert tmp.project_id == 1
tmp = self.webhook_repo.filter_by(limit=1, offset=1, project_id=1)
assert len(tmp) == 0
@with_context
def test_save(self):
"""Test save."""
webhook = self.webhook_repo.get(1)
assert webhook.id == 1
@with_context
def test_save_fails_if_integrity_error(self):
"""Test save raises a DBIntegrityError if the instance to be saved lacks
a required value"""
wh = Webhook(project_id=None)
assert_raises(DBIntegrityError, self.webhook_repo.save, wh)
@with_context
def test_save_only_saves_webhooks(self):
"""Test save raises a WrongObjectError when an object which is not
a Webhok instance is saved"""
bad_object = dict()
assert_raises(WrongObjectError, self.webhook_repo.save, bad_object)
@with_context
@patch('pybossa.repositories.webhook_repository.clean_project')
def test_delete_entries_from_project(self, clean_project_mock):
"""Test delete entries from project works."""
project = ProjectFactory.create()
wh = Webhook(project_id=project.id)
self.webhook_repo.save(wh)
self.webhook_repo.delete_entries_from_project(project)
res = self.webhook_repo.get_by(project_id=project.id)
assert res is None
clean_project_mock.assert_called_with(project.id)
