Move repo visibility tests into pytest

data/model/test/test_visible_repos.py

@@ -0,0 +1,89 @@
+from data import model
+
+from test.fixtures import *
+
+
+NO_ACCESS_USER = 'freshuser'
+READ_ACCESS_USER = 'reader'
+ADMIN_ACCESS_USER = 'devtable'
+PUBLIC_USER = 'public'
+RANDOM_USER = 'randomuser'
+OUTSIDE_ORG_USER = 'outsideorg'
+
+ADMIN_ROBOT_USER = 'devtable+dtrobot'
+
+ORGANIZATION = 'buynlarge'
+
+SIMPLE_REPO = 'simple'
+PUBLIC_REPO = 'publicrepo'
+RANDOM_REPO = 'randomrepo'
+
+OUTSIDE_ORG_REPO = 'coolrepo'
+
+ORG_REPO = 'orgrepo'
+ANOTHER_ORG_REPO = 'anotherorgrepo'
+
+# Note: The shared repo has devtable as admin, public as a writer and reader as a reader.
+SHARED_REPO = 'shared'
+
+
+def assertDoesNotHaveRepo(username, name):
+  repos = list(model.repository.get_visible_repositories(username))
+  names = [repo.name for repo in repos]
+  assert not name in names
+
+
+def assertHasRepo(username, name):
+  repos = list(model.repository.get_visible_repositories(username))
+  names = [repo.name for repo in repos]
+  assert name in names
+
+
+def test_noaccess(initialized_db):
+  repos = list(model.repository.get_visible_repositories(NO_ACCESS_USER))
+  names = [repo.name for repo in repos]
+  assert not names
+
+  # Try retrieving public repos now.
+  repos = list(model.repository.get_visible_repositories(NO_ACCESS_USER, include_public=True))
+  names = [repo.name for repo in repos]
+  assert PUBLIC_REPO in names
+
+
+def test_public(initialized_db):
+  assertHasRepo(PUBLIC_USER, PUBLIC_REPO)
+  assertHasRepo(PUBLIC_USER, SHARED_REPO)
+
+  assertDoesNotHaveRepo(PUBLIC_USER, SIMPLE_REPO)
+  assertDoesNotHaveRepo(PUBLIC_USER, RANDOM_REPO)
+  assertDoesNotHaveRepo(PUBLIC_USER, OUTSIDE_ORG_REPO)
+
+
+def test_reader(initialized_db):
+  assertHasRepo(READ_ACCESS_USER, SHARED_REPO)
+  assertHasRepo(READ_ACCESS_USER, ORG_REPO)
+
+  assertDoesNotHaveRepo(READ_ACCESS_USER, SIMPLE_REPO)
+  assertDoesNotHaveRepo(READ_ACCESS_USER, RANDOM_REPO)
+  assertDoesNotHaveRepo(READ_ACCESS_USER, OUTSIDE_ORG_REPO)
+  assertDoesNotHaveRepo(READ_ACCESS_USER, PUBLIC_REPO)
+
+
+def test_random(initialized_db):
+  assertHasRepo(RANDOM_USER, RANDOM_REPO)
+
+  assertDoesNotHaveRepo(RANDOM_USER, SIMPLE_REPO)
+  assertDoesNotHaveRepo(RANDOM_USER, SHARED_REPO)
+  assertDoesNotHaveRepo(RANDOM_USER, ORG_REPO)
+  assertDoesNotHaveRepo(RANDOM_USER, ANOTHER_ORG_REPO)
+  assertDoesNotHaveRepo(RANDOM_USER, PUBLIC_REPO)
+
+
+def test_admin(initialized_db):
+  assertHasRepo(ADMIN_ACCESS_USER, SIMPLE_REPO)
+  assertHasRepo(ADMIN_ACCESS_USER, SHARED_REPO)
+
+  assertHasRepo(ADMIN_ACCESS_USER, ORG_REPO)
+  assertHasRepo(ADMIN_ACCESS_USER, ANOTHER_ORG_REPO)
+
+  assertDoesNotHaveRepo(ADMIN_ACCESS_USER, OUTSIDE_ORG_REPO)