1 files changed, 10 insertions, 15 deletions
diff --git a/specs/XmlTimeline.py b/specs/XmlTimeline.py
index 9e14d47..ce4ea2b 100644
--- a/specs/XmlTimeline.py
+++ b/specs/XmlTimeline.py
@@ -17,14 +17,12 @@
 # along with Timeline.  If not, see <http://www.gnu.org/licenses/>.
 
 
-import codecs
-import tempfile
-import os.path
-import shutil
 from datetime import datetime
-import unittest
+import codecs
+
 import wx
 
+from specs.utils import TmpDirTestCase
 from timelinelib.db.backends.xmlfile import XmlTimeline
 from timelinelib.db import db_open
 from timelinelib.db.objects import Category
@@ -35,32 +33,32 @@ from timelinelib.meta.version import get_version
 from timelinelib.time import WxTimeType
 
 
-class XmlTimelineSpec(unittest.TestCase):
+class XmlTimelineSpec(TmpDirTestCase):
 
     IO = True
 
     def testUseWxTimeTypeWhenUseWideDateRangeIsTrue(self):
         timeline = XmlTimeline(None, load=False, use_wide_date_range=True)
-        self.assertTrue(isinstance(timeline.time_type, WxTimeType))
+        self.assertTrue(isinstance(timeline.get_time_type(), WxTimeType))
 
     def testAlertStringParsingGivesAlertData(self):
         timeline = XmlTimeline(None, load=False, use_wide_date_range=True)
         time, text = timeline._parse_alert_string("2012-11-11 00:00:00;Now is the time")
         self.assertEqual("Now is the time", text)
-        self.assertEqual("2012-11-11 00:00:00", "%s" % timeline.time_type.time_string(time))
+        self.assertEqual("2012-11-11 00:00:00", "%s" % timeline.get_time_type().time_string(time))
 
     def testAlertDataConversionGivesAlertString(self):
         timeline = XmlTimeline(None, load=False, use_wide_date_range=False)
         alert = (datetime(2010, 8, 31, 0, 0, 0), "Hoho")
         alert_text = timeline.alert_string(alert)
         self.assertEqual("2010-8-31 0:0:0;Hoho", alert_text)
-        
+
     def testWxTimeAlertDataConversionGivesAlertString(self):
         timeline = XmlTimeline(None, load=False, use_wide_date_range=True)
         alert = (wx.DateTimeFromDMY(30, 8, 2010, 0, 0, 0), "Hoho")
         alert_text = timeline.alert_string(alert)
         self.assertEqual("2010-09-30 00:00:00;Hoho", alert_text)
-        
+
     def testDisplayedPeriodTagNotWrittenIfNotSet(self):
         # Create a new db and add one event
         db = db_open(self.tmp_path)
@@ -162,8 +160,5 @@ class XmlTimelineSpec(unittest.TestCase):
                 self.fail("Unknown category.")
 
     def setUp(self):
-        self.tmp_dir = tempfile.mkdtemp(prefix="timeline-test")
-        self.tmp_path = os.path.join(self.tmp_dir, "test.timeline")
-
-    def tearDown(self):
-        shutil.rmtree(self.tmp_dir)
+        TmpDirTestCase.setUp(self)
+        self.tmp_path = self.get_tmp_path("test.timeline")