def test_new_file_and_clobber(self): arg_parser = AppArgumentParser() args = arg_parser.parse_args([self.get_url('/static/my_file.txt')]) with cd_tempdir() as temp_dir: app = Builder(args).build() exit_code = yield app.run() self.assertEqual(0, exit_code) expected_filename = os.path.join(temp_dir, 'my_file.txt') self.assertTrue(os.path.exists(expected_filename)) with open(expected_filename, 'rb') as in_file: self.assertIn(b'END', in_file.read()) app = Builder(args).build() exit_code = yield app.run() self.assertEqual(0, exit_code) expected_filename = os.path.join(temp_dir, 'my_file.txt.1') self.assertTrue(os.path.exists(expected_filename))
def test_dir_or_file(self): arg_parser = AppArgumentParser() with cd_tempdir(): args = arg_parser.parse_args([ self.get_url('/dir_or_file'), '--recursive', '--no-host-directories', ]) app = Builder(args).build() os.mkdir('dir_or_file') exit_code = yield app.run() self.assertEqual(0, exit_code) print(list(os.walk('.'))) self.assertTrue(os.path.isdir('dir_or_file')) self.assertTrue(os.path.isfile('dir_or_file.f')) with cd_tempdir(): args = arg_parser.parse_args([ self.get_url('/dir_or_file/'), '--recursive', '--no-host-directories', ]) app = Builder(args).build() with open('dir_or_file', 'wb'): pass exit_code = yield app.run() self.assertEqual(0, exit_code) print(list(os.walk('.'))) self.assertTrue(os.path.isdir('dir_or_file.d')) self.assertTrue(os.path.isfile('dir_or_file.d/index.html')) self.assertTrue(os.path.isfile('dir_or_file'))
def test_file_continue(self): arg_parser = AppArgumentParser() args = arg_parser.parse_args([self.get_url('/static/my_file.txt'), '--continue', '--debug']) with cd_tempdir() as temp_dir: filename = os.path.join(temp_dir, 'my_file.txt') with open(filename, 'wb') as out_file: out_file.write(b'START') app = Builder(args).build() exit_code = yield app.run() self.assertEqual(0, exit_code) with open(filename, 'rb') as in_file: data = in_file.read() self.assertEqual('54388a281352fdb2cfa66009ac0e35dd8916af7c', hashlib.sha1(data).hexdigest())