# Copyright (C) 2008 Canonical Ltd
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
"""Tests for selection of the right Launchpad service by environment"""
import os
import xmlrpclib
from bzrlib import errors
from bzrlib.plugins.launchpad.lp_registration import (
InvalidLaunchpadInstance, LaunchpadService, NotLaunchpadBranch)
from bzrlib.plugins.launchpad.test_lp_directory import FakeResolveFactory
from bzrlib.tests import TestCase
class LaunchpadServiceTests(TestCase):
"""Test that the correct Launchpad instance is chosen."""
def setUp(self):
super(LaunchpadServiceTests, self).setUp()
# make sure we have a reproducible standard environment
self._captureVar('BZR_LP_XMLRPC_URL', None)
def test_default_service(self):
service = LaunchpadService()
self.assertEqual('https://xmlrpc.launchpad.net/bazaar/',
service.service_url)
def test_alter_default_service_url(self):
LaunchpadService.DEFAULT_SERVICE_URL = 'http://example.com/'
try:
service = LaunchpadService()
self.assertEqual('http://example.com/',
service.service_url)
finally:
LaunchpadService.DEFAULT_SERVICE_URL = \
LaunchpadService.LAUNCHPAD_INSTANCE['production']
def test_staging_service(self):
service = LaunchpadService(lp_instance='staging')
self.assertEqual('https://xmlrpc.staging.launchpad.net/bazaar/',
service.service_url)
def test_edge_service(self):
service = LaunchpadService(lp_instance='edge')
self.assertEqual('https://xmlrpc.edge.launchpad.net/bazaar/',
service.service_url)
def test_dev_service(self):
service = LaunchpadService(lp_instance='dev')
self.assertEqual('https://xmlrpc.launchpad.dev/bazaar/',
service.service_url)
def test_demo_service(self):
service = LaunchpadService(lp_instance='demo')
self.assertEqual('https://xmlrpc.demo.launchpad.net/bazaar/',
service.service_url)
def test_unknown_service(self):
error = self.assertRaises(InvalidLaunchpadInstance,
LaunchpadService,
lp_instance='fubar')
self.assertEqual('fubar is not a valid Launchpad instance.',
str(error))
def test_environment_overrides_default(self):
os.environ['BZR_LP_XMLRPC_URL'] = 'http://example.com/'
service = LaunchpadService()
self.assertEqual('http://example.com/',
service.service_url)
def test_environment_overrides_specified_service(self):
os.environ['BZR_LP_XMLRPC_URL'] = 'http://example.com/'
service = LaunchpadService(lp_instance='staging')
self.assertEqual('http://example.com/',
service.service_url)
class TestURLInference(TestCase):
"""Test the way we infer Launchpad web pages from branch URLs."""
def test_default_bzr_ssh_url(self):
service = LaunchpadService()
web_url = service.get_web_url_from_branch_url(
'bzr+ssh://bazaar.launchpad.net/~foo/bar/baz')
self.assertEqual(
'https://code.launchpad.net/~foo/bar/baz', web_url)
def test_product_bzr_ssh_url(self):
service = LaunchpadService(lp_instance='production')
web_url = service.get_web_url_from_branch_url(
'bzr+ssh://bazaar.launchpad.net/~foo/bar/baz')
self.assertEqual(
'https://code.launchpad.net/~foo/bar/baz', web_url)
def test_sftp_branch_url(self):
service = LaunchpadService(lp_instance='production')
web_url = service.get_web_url_from_branch_url(
'sftp://bazaar.launchpad.net/~foo/bar/baz')
self.assertEqual(
'https://code.launchpad.net/~foo/bar/baz', web_url)
def test_staging_branch_url(self):
service = LaunchpadService(lp_instance='production')
web_url = service.get_web_url_from_branch_url(
'bzr+ssh://bazaar.staging.launchpad.net/~foo/bar/baz')
self.assertEqual(
'https://code.launchpad.net/~foo/bar/baz', web_url)
def test_non_launchpad_url(self):
service = LaunchpadService()
error = self.assertRaises(
NotLaunchpadBranch, service.get_web_url_from_branch_url,
'bzr+ssh://example.com/~foo/bar/baz')
self.assertEqual(
'bzr+ssh://example.com/~foo/bar/baz is not registered on Launchpad.',
str(error))
def test_dodgy_launchpad_url(self):
service = LaunchpadService()
self.assertRaises(
NotLaunchpadBranch, service.get_web_url_from_branch_url,
'bzr+ssh://launchpad.net/~foo/bar/baz')
def test_lp_branch_url(self):
service = LaunchpadService(lp_instance='production')
factory = FakeResolveFactory(
self, '~foo/bar/baz',
dict(urls=['http://bazaar.launchpad.net/~foo/bar/baz']))
web_url = service.get_web_url_from_branch_url(
'lp:~foo/bar/baz', factory)
self.assertEqual(
'https://code.launchpad.net/~foo/bar/baz', web_url)
def test_lp_branch_shortcut(self):
service = LaunchpadService()
factory = FakeResolveFactory(
self, 'foo',
dict(urls=['http://bazaar.launchpad.net/~foo/bar/baz']))
web_url = service.get_web_url_from_branch_url('lp:foo', factory)
self.assertEqual(
'https://code.launchpad.net/~foo/bar/baz', web_url)
def test_lp_branch_fault(self):
service = LaunchpadService()
factory = FakeResolveFactory(self, 'foo', None)
def submit(service):
raise xmlrpclib.Fault(42, 'something went wrong')
factory.submit = submit
self.assertRaises(
errors.InvalidURL, service.get_web_url_from_branch_url, 'lp:foo',
factory)
def test_staging_url(self):
service = LaunchpadService(lp_instance='staging')
web_url = service.get_web_url_from_branch_url(
'bzr+ssh://bazaar.launchpad.net/~foo/bar/baz')
self.assertEqual(
'https://code.staging.launchpad.net/~foo/bar/baz', web_url)
def test_edge_url(self):
service = LaunchpadService(lp_instance='edge')
web_url = service.get_web_url_from_branch_url(
'bzr+ssh://bazaar.launchpad.net/~foo/bar/baz')
self.assertEqual(
'https://code.edge.launchpad.net/~foo/bar/baz', web_url)
def test_dev_url(self):
service = LaunchpadService(lp_instance='dev')
web_url = service.get_web_url_from_branch_url(
'bzr+ssh://bazaar.launchpad.net/~foo/bar/baz')
self.assertEqual(
'https://code.launchpad.dev/~foo/bar/baz', web_url)
def test_demo_url(self):
service = LaunchpadService(lp_instance='demo')
web_url = service.get_web_url_from_branch_url(
'bzr+ssh://bazaar.launchpad.net/~foo/bar/baz')
self.assertEqual(
'https://code.demo.launchpad.net/~foo/bar/baz', web_url)
|