Refactor and limit verification

michaeljoseph · Oct 2, 2014 · 0ad816b · 0ad816b
1 parent f408ec6
commit 0ad816b
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 9 deletions.
diff --git a/tests/__init__.py b/tests/__init__.py
@@ -41,6 +41,7 @@ class BaseTestCase(TestCase):
     title = 'Application To Platform'
     directory_name = 'application-to-platform'
     example_file = 'atp.md'
+    presentation_index = '%s/index.html' % directory_name
 
     def setUp(self):
         with open(self.example_file, 'w') as f:

diff --git a/tests/test_remarkable.py b/tests/test_remarkable.py
@@ -14,32 +14,30 @@
 class TestRemarkTestCase(BaseTestCase):
 
     def test_remark(self):
-        presentation_index = '%s/index.html' % self.directory_name
         sys.argv = [
             'remarkable',
             'remark',
             self.example_file,
             self.title,
         ]
         cli.main()
-        self.assertTrue(os.path.exists(presentation_index))
-        presentation_contents = io.open(presentation_index).read()
+        self.assertTrue(os.path.exists(self.presentation_index))
+        presentation_contents = io.open(self.presentation_index).read()
         self.assertTrue(self.title in presentation_contents)
-        self.assertTrue('michaeljoseph' in presentation_contents)
-        shutil.rmtree('application-to-platform')
+        shutil.rmtree(self.directory_name)
 
 
 class TestRevealTestCase(BaseTestCase):
 
     def test_reveal(self):
-        presentation_index = 'application-to-platform/index.html'
         sys.argv = [
             'remarkable',
             'reveal',
             self.example_file,
             self.title,
         ]
         cli.main()
-        self.assertTrue(os.path.exists(presentation_index))
-        self.assertTrue(self.title in io.open(presentation_index).read())
-        shutil.rmtree('application-to-platform')
+        self.assertTrue(os.path.exists(self.presentation_index))
+        presentation_contents = io.open(self.presentation_index).read()
+        self.assertTrue(self.title in presentation_contents)
+        shutil.rmtree(self.directory_name)