summaryrefslogtreecommitdiffstats
path: root/tests
diff options
context:
space:
mode:
authorTails developers <tails@boum.org>2011-07-08 17:54:37 +0200
committerTails developers <tails@boum.org>2011-07-08 17:55:47 +0200
commit0fcb8f677c36e236d18a73cac6f69d47521f7f89 (patch)
tree4dd0bbd2c5e4e2219ed07f9c92a43150e012f5ae /tests
parent16d5fa5db4827ed4fae3ae1b5f18a0efc0e67fc8 (diff)
Refactoring: create a set of Source classes and an utils module.
Diffstat (limited to 'tests')
-rw-r--r--tests/test_creator.py5
1 files changed, 3 insertions, 2 deletions
diff --git a/tests/test_creator.py b/tests/test_creator.py
index 8ac8b09..a09ed66 100644
--- a/tests/test_creator.py
+++ b/tests/test_creator.py
@@ -90,14 +90,15 @@ class TestLiveUSBCreator:
def test_extract_iso(self):
from glob import glob
+ from liveusb.source import LocalIsoSource
live = self._get_creator()
live.detect_removable_drives()
isos = filter(lambda x: x.endswith('.iso'),
filter(os.path.isfile, glob('*') + glob('*/*')))
assert isos, "No ISOs found. Put one in this directory"
for drive in live.drives:
- live.drive = drive
- live.iso = isos[0]
+ live.drive = drive
+ live.source = LocalIsoSource(path = isos[0])
live.mount_device()
if os.path.exists(live.get_liveos()):
live.delete_liveos()