OLD | NEW |
(Empty) | |
| 1 #!/usr/bin/python |
| 2 |
| 3 """ |
| 4 Copyright 2014 Google Inc. |
| 5 |
| 6 Use of this source code is governed by a BSD-style license that can be |
| 7 found in the LICENSE file. |
| 8 |
| 9 Test url_utils.py |
| 10 """ |
| 11 |
| 12 # System-level imports |
| 13 import os |
| 14 import shutil |
| 15 import tempfile |
| 16 import unittest |
| 17 import urllib |
| 18 |
| 19 # Imports from within Skia |
| 20 import url_utils |
| 21 |
| 22 |
| 23 class UrlUtilsTest(unittest.TestCase): |
| 24 |
| 25 def test_create_filepath_url(self): |
| 26 """Tests create_filepath_url(). """ |
| 27 with self.assertRaises(Exception): |
| 28 url_utils.create_filepath_url('http://1.2.3.4/path') |
| 29 # Pass absolute filepath. |
| 30 self.assertEquals( |
| 31 url_utils.create_filepath_url( |
| 32 '%sdir%sfile' % (os.path.sep, os.path.sep)), |
| 33 'file:///dir/file') |
| 34 # Pass relative filepath. |
| 35 self.assertEquals( |
| 36 url_utils.create_filepath_url(os.path.join('dir', 'file')), |
| 37 'file://%s/dir/file' % urllib.pathname2url(os.getcwd())) |
| 38 |
| 39 def test_copy_contents(self): |
| 40 """Tests copy_contents(). """ |
| 41 contents = 'these are the contents' |
| 42 tempdir_path = tempfile.mkdtemp() |
| 43 try: |
| 44 source_path = os.path.join(tempdir_path, 'source') |
| 45 source_url = url_utils.create_filepath_url(source_path) |
| 46 with open(source_path, 'w') as source_handle: |
| 47 source_handle.write(contents) |
| 48 dest_path = os.path.join(tempdir_path, 'new_subdir', 'dest') |
| 49 # Destination subdir does not exist, so copy_contents() should fail |
| 50 # if create_subdirs_if_needed is False. |
| 51 with self.assertRaises(Exception): |
| 52 url_utils.copy_contents(source_url=source_url, |
| 53 dest_path=dest_path, |
| 54 create_subdirs_if_needed=False) |
| 55 # If create_subdirs_if_needed is True, it should work. |
| 56 url_utils.copy_contents(source_url=source_url, |
| 57 dest_path=dest_path, |
| 58 create_subdirs_if_needed=True) |
| 59 self.assertEquals(open(dest_path).read(), contents) |
| 60 finally: |
| 61 shutil.rmtree(tempdir_path) |
OLD | NEW |