initial import for Open Source 🎉

data/model/test/test_team.py

@@ -0,0 +1,61 @@
+import pytest
+
+from data.model.team import (add_or_invite_to_team, create_team, confirm_team_invite,
+                             list_team_users, validate_team_name)
+from data.model.organization import create_organization
+from data.model.user import get_user, create_user_noverify
+
+from test.fixtures import *
+
+
+@pytest.mark.parametrize('name, is_valid', [
+  ('', False),
+  ('f', False),
+  ('fo', True),
+  ('f' * 255, True),
+  ('f' * 256, False),
+  (' ', False),
+  ('helloworld', True),
+  ('hello_world', True),
+  ('hello-world', True),
+  ('hello world', False),
+  ('HelloWorld', False),
+])
+def test_validate_team_name(name, is_valid):
+  result, _ = validate_team_name(name)
+  assert result == is_valid
+
+
+def is_in_team(team, user):
+  return user.username in {u.username for u in list_team_users(team)}
+
+
+def test_invite_to_team(initialized_db):
+  first_user = get_user('devtable')
+  second_user = create_user_noverify('newuser', 'foo@example.com')
+
+  def run_invite_flow(orgname):
+    # Create an org owned by `devtable`.
+    org = create_organization(orgname, orgname + '@example.com', first_user)
+
+    # Create another team and add `devtable` to it. Since `devtable` is already
+    # in the org, it should be done directly.
+    other_team = create_team('otherteam', org, 'admin')
+    invite = add_or_invite_to_team(first_user, other_team, user_obj=first_user)
+    assert invite is None
+    assert is_in_team(other_team, first_user)
+
+    # Try to add `newuser` to the team, which should require an invite.
+    invite = add_or_invite_to_team(first_user, other_team, user_obj=second_user)
+    assert invite is not None
+    assert not is_in_team(other_team, second_user)
+
+    # Accept the invite.
+    confirm_team_invite(invite.invite_token, second_user)
+    assert is_in_team(other_team, second_user)
+
+  # Run for a new org.
+  run_invite_flow('firstorg')
+
+  # Create another org and repeat, ensuring the same operations perform the same way.
+  run_invite_flow('secondorg')