1- import unittest
1+ import unittest2
22from datetime import datetime , timedelta , date as dt_date
33import zfslib as zfs
44from zfslib_test_tools import *
@@ -52,7 +52,7 @@ def _checkPath(path):
5252 if b_cr : _checkPath (diff .snap_path_right )
5353
5454
55- class Diff_Tests (unittest .TestCase ):
55+ class Diff_Tests (unittest2 .TestCase ):
5656 def setUp (self ):
5757 if len (snaps_varlog ) < 2 :
5858 raise Exception ("Testing cannot be done. Please ensure that Dataset rpool/ROOT/<computer>/var/log has more than one snapshot" )
@@ -159,7 +159,7 @@ def _verify_diffs(self, step, diffs):
159159
160160
161161
162- class Poolset_Tests (unittest .TestCase ):
162+ class Poolset_Tests (unittest2 .TestCase ):
163163 def test_find_dataset_for_path (self ):
164164 path = '/var/log/apt/history.log'
165165 tup = poolset .find_dataset_for_path (path )
@@ -200,7 +200,7 @@ def test_find_dataset_for_path(self):
200200
201201
202202
203- class Dataset_Tests (unittest .TestCase ):
203+ class Dataset_Tests (unittest2 .TestCase ):
204204
205205 def test_get_rel_path (self ):
206206 self .assertEqual (ds_varlog .get_rel_path ('/var/log/apt/history.log' ), '/apt/history.log' )
@@ -212,7 +212,7 @@ def test_get_rel_path(self):
212212
213213
214214
215- class Snapshot_Tests (unittest .TestCase ):
215+ class Snapshot_Tests (unittest2 .TestCase ):
216216
217217 def test_resolve_snap_path (self ):
218218 snap = snaps_varlog [1 ]
@@ -240,4 +240,5 @@ def test_resolve_snap_path(self):
240240
241241
242242if __name__ == "__main__" :
243- unittest .main ()
243+ unittest2 .main ()
244+ sys .exit (0 )
0 commit comments