def test_data_dir(self):
project = Project(path=sample_project_dir)
self.assertEqual(project.data_dir, '.crawlmi')
project.set_data_dir('crawlmi_data')
self.assertEqual(project.data_dir, join(sample_project_dir, 'crawlmi_data'))
self.assertFalse(exists(project.data_dir))
project.data_path('.', create_dir=True)
self.assertTrue(exists(project.data_dir))
os.rmdir(project.data_dir)
project.set_data_dir(None)
self.assertEqual(project.data_dir, '.crawlmi')
def test_data_path(self):
project = Project(path=sample_project_dir)
project.set_data_dir('crawlmi_data')
# relative path
expected = join(sample_project_dir, 'crawlmi_data', 'a', 'b')
relative = project.data_path(join('a', 'b'), create_dir=False)
self.assertEqual(relative, expected)
self.assertFalse(exists(expected))
# create dir
project.data_path(join('a', 'b'), create_dir=True)
self.assertTrue(exists(expected))
os.rmdir(join(sample_project_dir, 'crawlmi_data', 'a', 'b'))
os.rmdir(join(sample_project_dir, 'crawlmi_data', 'a'))
# absolute path
expected = os.path.abspath(__file__)
absolute = project.data_path(expected, create_dir=False)
self.assertEqual(absolute, expected)
os.rmdir(project.data_dir)
