def test_gets_file_bad_url(self, _urlopen, _exists, _unlink):
destfile = "foo"
_urlopen.return_value = StringIO("words words")
hh = HttpFileHandler(HttpURLs[0])
# Broken source
_open = mock_open()
_exists.return_value = False
_urlopen.side_effect = URLError('failed')
with patch('codetree.handlers.open', _open, create=True):
self.assertFalse(hh.get(destfile))
def test_gets_file_bad_url(self, _urlopen, _exists, _unlink):
destfile = "foo"
_urlopen.return_value = StringIO("words words")
hh = HttpFileHandler(HttpURLs[0])
# Broken source
_open = mock_open()
_exists.return_value = False
_urlopen.side_effect = URLError('failed')
with patch('codetree.handlers.open', _open, create=True):
self.assertFalse(hh.get(destfile))
def test_gets_file_with_overwite(self, _urlopen, _exists, _unlink):
destfile = "foo"
_urlopen.return_value = StringIO("words words")
hh = HttpFileHandler(HttpURLs[0])
# Overwrite existing file
_open = mock_open()
_exists.return_value = True
with patch('codetree.handlers.open', _open, create=True):
self.assertTrue(hh.get(destfile, options={"overwrite": True}))
_unlink.assert_called_with(destfile)
_open.assert_called_with(destfile, "w")
_urlopen.assert_called_with(HttpURLs[0])
def test_gets_file_no_overwrite(self, _urlopen, _exists, _unlink):
destfile = "foo"
_urlopen.return_value = StringIO("words words")
hh = HttpFileHandler(HttpURLs[0])
# Existing file
_open = mock_open()
_exists.return_value = True
with patch('codetree.handlers.open', _open, create=True):
self.assertFalse(hh.get(destfile))
self.assertFalse(_unlink.called)
self.assertFalse(_open.called)
self.assertFalse(_urlopen.called)
def test_gets_file(self, _urlopen, _exists, _unlink):
destfile = "foo"
_urlopen.return_value = StringIO("words words")
hh = HttpFileHandler(HttpURLs[0])
# New file
_open = mock_open()
_exists.return_value = False
with patch('codetree.handlers.open', _open, create=True):
self.assertTrue(hh.get(destfile))
self.assertFalse(_unlink.called)
_open.assert_called_with(destfile, "w")
_urlopen.assert_called_with(HttpURLs[0])
def test_gets_file_with_overwite(self, _urlopen, _exists, _unlink):
destfile = "foo"
_urlopen.return_value = StringIO("words words")
hh = HttpFileHandler(HttpURLs[0])
# Overwrite existing file
_open = mock_open()
_exists.return_value = True
with patch('codetree.handlers.open', _open, create=True):
self.assertTrue(hh.get(destfile, options={"overwrite": True}))
_unlink.assert_called_with(destfile)
_open.assert_called_with(destfile, "w")
_urlopen.assert_called_with(HttpURLs[0])
def test_gets_file_no_overwrite(self, _urlopen, _exists, _unlink):
destfile = "foo"
_urlopen.return_value = StringIO("words words")
hh = HttpFileHandler(HttpURLs[0])
# Existing file
_open = mock_open()
_exists.return_value = True
with patch('codetree.handlers.open', _open, create=True):
self.assertFalse(hh.get(destfile))
self.assertFalse(_unlink.called)
self.assertFalse(_open.called)
self.assertFalse(_urlopen.called)
def test_gets_file(self, _urlopen, _exists, _unlink):
destfile = "foo"
_urlopen.return_value = StringIO("words words")
hh = HttpFileHandler(HttpURLs[0])
# New file
_open = mock_open()
_exists.return_value = False
with patch('codetree.handlers.open', _open, create=True):
self.assertTrue(hh.get(destfile))
self.assertFalse(_unlink.called)
_open.assert_called_with(destfile, "w")
_urlopen.assert_called_with(HttpURLs[0])
