Lubuntu/ubuntu-dev-tools
3
0
Fork 0
mirror of https://git.launchpad.net/ubuntu-dev-tools synced 2025-05-12 09:21:29 +00:00
Code Issues Packages Projects Releases Wiki Activity

Port http_mock to mock.

Browse Source
This commit is contained in:
Dimitri John Ledkov 2014-12-15 04:24:41 +00:00
parent 522030be12
commit 88e3d1d6ba
1 changed files with 12 additions and 32 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

44
ubuntutools/test/test_archive.py
View File

@ -93,9 +93,8 @@ class LocalSourcePackageTestCase(mox.MoxTestBase, unittest.TestCase):
self._stubout('ubuntutools.archive.Distribution') self._stubout('ubuntutools.archive.Distribution')
self._stubout('ubuntutools.archive.rmadison') self._stubout('ubuntutools.archive.rmadison')
self.real_http = httplib2.Http() self.mock_http = self._stubout('httplib2.Http.request')
self.mox.StubOutWithMock(httplib2, 'Http') self.mock_http.side_effect = self.request_proxy
self.mock_http = self.mox.CreateMock(httplib2.Http)
# Silence the tests a little: # Silence the tests a little:
self._stubout('ubuntutools.logger.Logger.stdout') self._stubout('ubuntutools.logger.Logger.stdout')
@ -104,7 +103,7 @@ class LocalSourcePackageTestCase(mox.MoxTestBase, unittest.TestCase):
def _stubout(self, stub): def _stubout(self, stub):
patcher = mock.patch(stub) patcher = mock.patch(stub)
self.addCleanup(patcher.stop) self.addCleanup(patcher.stop)
patcher.start() return patcher.start()
def tearDown(self): def tearDown(self):
super(LocalSourcePackageTestCase, self).tearDown() super(LocalSourcePackageTestCase, self).tearDown()
@ -144,6 +143,12 @@ class LocalSourcePackageTestCase(mox.MoxTestBase, unittest.TestCase):
response = httplib2.Response({'status': 404}) response = httplib2.Response({'status': 404})
return response, "I'm a 404 Error" return response, "I'm a 404 Error"
def request_404_then_proxy(self, url, destname=None):
"mock side_effect callable to chain request 404 & proxy"
if self.mock_http.called:
return self.request_proxy(url, destname)
return self.request_404(url)
def test_local_copy(self): def test_local_copy(self):
pkg = self.SourcePackage('example', '1.0-1', 'main', pkg = self.SourcePackage('example', '1.0-1', 'main',
dscfile='test-data/example_1.0-1.dsc', dscfile='test-data/example_1.0-1.dsc',
@ -191,10 +196,6 @@ class LocalSourcePackageTestCase(mox.MoxTestBase, unittest.TestCase):
dist = self.SourcePackage.distribution dist = self.SourcePackage.distribution
mirror = UDTConfig.defaults['%s_MIRROR' % dist.upper()] mirror = UDTConfig.defaults['%s_MIRROR' % dist.upper()]
urlbase = '/pool/main/e/example/' urlbase = '/pool/main/e/example/'
httplib2.Http().AndReturn(self.mock_http)
self.mock_http.request('https://launchpad.net/%s/+archive/primary/'
'+files/example_1.0-1.dsc' % dist
).WithSideEffects(self.request_proxy)
url_opener = self.mox.CreateMock(urllib2.OpenerDirector) url_opener = self.mox.CreateMock(urllib2.OpenerDirector)
url_opener.open(mirror + urlbase + 'example_1.0.orig.tar.gz' url_opener.open(mirror + urlbase + 'example_1.0.orig.tar.gz'
).WithSideEffects(self.urlopen_proxy) ).WithSideEffects(self.urlopen_proxy)
@ -212,9 +213,6 @@ class LocalSourcePackageTestCase(mox.MoxTestBase, unittest.TestCase):
mirror = 'http://mirror' mirror = 'http://mirror'
lpbase = 'https://launchpad.net/ubuntu/+archive/primary/+files/' lpbase = 'https://launchpad.net/ubuntu/+archive/primary/+files/'
urlbase = '/pool/main/e/example/' urlbase = '/pool/main/e/example/'
httplib2.Http().AndReturn(self.mock_http)
self.mock_http.request(lpbase + 'example_1.0-1.dsc'
).WithSideEffects(self.request_proxy)
url_opener = self.mox.CreateMock(urllib2.OpenerDirector) url_opener = self.mox.CreateMock(urllib2.OpenerDirector)
url_opener.open(mirror + urlbase + 'example_1.0.orig.tar.gz' url_opener.open(mirror + urlbase + 'example_1.0.orig.tar.gz'
).WithSideEffects(self.urlopen_null) ).WithSideEffects(self.urlopen_null)
@ -232,12 +230,7 @@ class LocalSourcePackageTestCase(mox.MoxTestBase, unittest.TestCase):
pkg.pull() pkg.pull()
def test_dsc_missing(self): def test_dsc_missing(self):
lpbase = 'https://launchpad.net/ubuntu/+archive/primary/+files/' self.mock_http.side_effect = self.request_404
httplib2.Http().AndReturn(self.mock_http)
self.mock_http.request(lpbase + 'example_1.0-1.dsc'
).WithSideEffects(self.request_404)
self.mox.ReplayAll()
pkg = self.SourcePackage('example', '1.0-1', 'main', pkg = self.SourcePackage('example', '1.0-1', 'main',
workdir=self.workdir) workdir=self.workdir)
self.assertRaises(ubuntutools.archive.DownloadError, pkg.pull) self.assertRaises(ubuntutools.archive.DownloadError, pkg.pull)
@ -254,9 +247,6 @@ class DebianLocalSourcePackageTestCase(LocalSourcePackageTestCase):
lpbase = 'https://launchpad.net/debian/+archive/primary/+files/' lpbase = 'https://launchpad.net/debian/+archive/primary/+files/'
base = '/pool/main/e/example/' base = '/pool/main/e/example/'
httplib2.Http().AndReturn(self.mock_http)
self.mock_http.request(lpbase + 'example_1.0-1.dsc'
).WithSideEffects(self.request_proxy)
url_opener = self.mox.CreateMock(urllib2.OpenerDirector) url_opener = self.mox.CreateMock(urllib2.OpenerDirector)
url_opener.open(debian_mirror + base + 'example_1.0.orig.tar.gz' url_opener.open(debian_mirror + base + 'example_1.0.orig.tar.gz'
).WithSideEffects(self.urlopen_null) ).WithSideEffects(self.urlopen_null)
@ -291,12 +281,7 @@ class DebianLocalSourcePackageTestCase(LocalSourcePackageTestCase):
mirror = 'http://mirror' mirror = 'http://mirror'
lpbase = 'https://launchpad.net/debian/+archive/primary/+files/' lpbase = 'https://launchpad.net/debian/+archive/primary/+files/'
base = '/pool/main/e/example/' base = '/pool/main/e/example/'
httplib2.Http().AndReturn(self.mock_http) self.mock_http.side_effect = self.request_404_then_proxy
self.mock_http.request(lpbase + 'example_1.0-1.dsc'
).WithSideEffects(self.request_404)
httplib2.Http().AndReturn(self.mock_http)
self.mock_http.request(mirror + base + 'example_1.0-1.dsc'
).WithSideEffects(self.request_proxy)
url_opener = self.mox.CreateMock(urllib2.OpenerDirector) url_opener = self.mox.CreateMock(urllib2.OpenerDirector)
url_opener.open(mirror + base + 'example_1.0.orig.tar.gz' url_opener.open(mirror + base + 'example_1.0.orig.tar.gz'
).WithSideEffects(self.urlopen_proxy) ).WithSideEffects(self.urlopen_proxy)
@ -326,12 +311,7 @@ class DebianLocalSourcePackageTestCase(LocalSourcePackageTestCase):
mirror = 'http://mirror' mirror = 'http://mirror'
lpbase = 'https://launchpad.net/debian/+archive/primary/+files/' lpbase = 'https://launchpad.net/debian/+archive/primary/+files/'
base = '/pool/main/e/example/' base = '/pool/main/e/example/'
httplib2.Http().AndReturn(self.mock_http) self.mock_http.side_effect = self.request_404_then_proxy
self.mock_http.request(lpbase + 'example_1.0-1.dsc'
).WithSideEffects(self.request_404)
httplib2.Http().AndReturn(self.mock_http)
self.mock_http.request(mirror + base + 'example_1.0-1.dsc'
).WithSideEffects(self.request_proxy)
def fake_gpg_info(self, message, keyrings=None): def fake_gpg_info(self, message, keyrings=None):
return debian.deb822.GpgInfo.from_output( return debian.deb822.GpgInfo.from_output(