1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
|
# Copyright 2011 Canonical Ltd. This software is licensed under the
# GNU Affero General Public License version 3 (see the file LICENSE).
"""Functional tests for BranchCollection adapters."""
__metaclass__ = type
from lp.code.interfaces.branchcollection import IBranchCollection
from lp.testing import TestCaseWithFactory
from lp.registry.model.personproduct import PersonProduct
from canonical.testing.layers import DatabaseFunctionalLayer
class TestPersonProduct(TestCaseWithFactory):
layer = DatabaseFunctionalLayer
def test_person_product(self):
"""A PersonProduct can be adapted to a collection.
The collection will only find branches matching both the person and
the product.
"""
product = self.factory.makeProduct()
person = self.factory.makePerson()
person_product = PersonProduct(person, product)
random_branch = self.factory.makeBranch()
product_branch = self.factory.makeProductBranch(product=product)
person_branch = self.factory.makeBranch(owner=person)
person_product_branch = self.factory.makeProductBranch(
owner=person, product=product)
branches = IBranchCollection(person_product).getBranches()
self.assertNotIn(random_branch, branches)
self.assertNotIn(product_branch, branches)
self.assertNotIn(person_branch, branches)
self.assertIn(person_product_branch, branches)
|