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())
