from breezy.tests import TestCaseWithTransport
from silver_platter import debian
pick_additional_colocated_branches = debian.pick_additional_colocated_branches
class AdditionalColocatedBranchesTests(TestCaseWithTransport):
def test_none(self):
a = self.make_branch("target")
self.assertEqual(
pick_additional_colocated_branches(a),
{},
)
def test_upstream(self):
a = self.make_branch("target")
a.controldir.create_branch("upstream")
self.assertEqual(
pick_additional_colocated_branches(a),
{"upstream": "upstream"},
)
def test_pristine_tar(self):
a = self.make_branch("target")
a.controldir.create_branch("pristine-tar")
self.assertEqual(
pick_additional_colocated_branches(a),
{"pristine-tar": "pristine-tar"},
)