Expose the real filename for SaveFile

1 files changed, 8 insertions, 0 deletions

diff --git a/morphlib/savefile_tests.py b/morphlib/savefile_tests.py
index 442a0779..63c80c9d 100644
--- a/morphlib/savefile_tests.py
+++ b/morphlib/savefile_tests.py
@@ -43,6 +43,14 @@ class SaveFileTests(unittest.TestCase):
     def test_there_are_no_files_initially(self):
         self.assertEqual(os.listdir(self.tempdir), [])
 
+    def test_sets_real_filename(self):
+        f = savefile.SaveFile(self.filename, 'w')
+        self.assertEqual(f.real_filename, self.filename)
+
+    def test_sets_name_to_temporary_name(self):
+        f = savefile.SaveFile(self.filename, 'w')
+        self.assertNotEqual(f.name, self.filename)
+
     def test_saves_new_file(self):
         f = savefile.SaveFile(self.filename, 'w')
         f.write('foo')