from unittest import mock from django.core.urlresolvers import reverse from taiga.projects import services from taiga.base.utils import json from .. import factories as f import pytest pytestmark = pytest.mark.django_db def test_get_members_from_bulk(): data = [{"role_id": "1", "email": "member1@email.com"}, {"role_id": "1", "email": "member2@email.com"}] members = services.get_members_from_bulk(data, project_id=1) assert len(members) == 2 assert members[0].email == "member1@email.com" assert members[1].email == "member2@email.com" @mock.patch("taiga.projects.services.members.db") def test_create_members_in_bulk(db): data = [{"role_id": "1", "email": "member1@email.com"}, {"role_id": "1", "email": "member2@email.com"}] members = services.create_members_in_bulk(data, project_id=1) db.save_in_bulk.assert_called_once_with(members, None, None) def test_api_create_bulk_members(client): project = f.ProjectFactory() john = f.UserFactory.create() joseph = f.UserFactory.create() tester = f.RoleFactory(project=project, name="Tester") gamer = f.RoleFactory(project=project, name="Gamer") url = reverse("memberships-bulk-create") data = { "project_id": project.id, "bulk_memberships": [ {"role_id": tester.pk, "email": john.email}, {"role_id": gamer.pk, "email": joseph.email}, ] } client.login(project.owner) response = client.json.post(url, json.to_json(data)) assert response.status_code == 200 assert response.data[0]["email"] == john.email assert response.data[1]["email"] == joseph.email def test_api_resend_invitation(client, outbox): invitation = f.create_invitation() url = reverse("memberships-resend-invitation", kwargs={"pk": invitation.pk}) client.login(invitation.project.owner) response = client.post(url) assert response.status_code == 204 assert len(outbox) == 1 assert outbox[0].to == [invitation.email] def test_api_invite_existing_user(client, outbox): "Should create the invitation linked to that user" user = f.UserFactory.create() role = f.RoleFactory.create() client.login(role.project.owner) url = reverse("memberships-list") data = {"role": role.pk, "project": role.project.pk, "email": user.email} response = client.json.post(url, json.to_json(data)) assert response.status_code == 201, response.data assert len(outbox) == 1 assert user.memberships.count() == 1 message = outbox[0] assert message.to == [user.email] assert "Added to the project" in message.subject def test_api_create_invalid_membership(client): "Should create the invitation linked to that user" user = f.UserFactory.create() role = f.RoleFactory.create() client.login(role.project.owner) url = reverse("memberships-list") data = {"role": role.pk, "project": role.project.pk} response = client.json.post(url, data) assert response.status_code == 400, response.data assert user.memberships.count() == 0