def test_allocations (self):
project_1 = Mock(['id'])
project_1.id = "1"
project_2 = Mock(['id'])
project_2.id = "2"
resource = Mock(['id'])
resource.id = "1"
dt = datetime(2000, 1, 1)
Session.add(Allocation(project_1, resource, 0, dt, dt))
Session.add(Allocation(project_2, resource, 0, dt, dt))
assert_equal(
set(get_projects()),
set([Project.cached("1"), Project.cached("2")]))
def test_manager_projects (self):
project_1 = Mock(['id'])
project_1.id = "1"
project_2 = Mock(['id'])
project_2.id = "2"
resource = Mock(['id'])
resource.id = "1"
dt = datetime(2000, 1, 1)
Session.add(Allocation(project_1, resource, 0, dt, dt))
Session.add(Allocation(project_2, resource, 0, dt, dt))
assert_equal(
get_projects(manager=User.cached("1")),
[Project.cached("1")])
def test_no_allocations (self):
assert_equal(get_projects(), [])
