def test_downloader_does_not_care_if_directory_already_exists(self):
        solnbr = 'FA-FOO-BAR-BAZ'
        downloader = AttachmentDownloader(shelf='test_attach',
                                          dl_dir='py_test_dls')

        downloader.create_dir_by_solnbr(solnbr)
        downloader.create_dir_by_solnbr(solnbr)

        # clean up
        os.rmdir(downloader.dir_for_solnbr(solnbr))
    def test_creates_solnbr_download_directory(self):
        solnbr = 'FA-FOO-BAR-BAZ'
        downloader = AttachmentDownloader(shelf='test_attach',
                                          dl_dir='py_test_dls')
        dirpath = downloader.dir_for_solnbr(solnbr)

        downloader.create_dir_by_solnbr(solnbr)
        self.assertTrue(os.path.isdir(dirpath))

        # clean up
        os.rmdir(dirpath)
        self.assertFalse(os.path.isdir(dirpath))
 def test_constructs_solnbr_download_directory_name(self):
     downloader = AttachmentDownloader(shelf='test_attach',
                                       dl_dir='py_test_dls')
     self.assertEqual(downloader.dir_for_solnbr('FA-FOO-BAR-BAZ'),
                      'py_test_dls/FA-FOO-BAR-BAZ')