def test_save_file_with_existing_si_and_correct_we_should_return_true(self): si = ''.join(random.sample(string.letters+string.digits, 12)) we = ''.join(random.sample(string.letters+string.digits, 12)) result = save_file(si, StringIO.StringIO('NEW_FILE_WITH_WE'), 20, we) updated = save_file(si, StringIO.StringIO('UPDATED_FILE'), 20, we) fp = open(os.path.join(FILE_STORE, si), 'r') content = fp.read() fp.close() self.assertTrue(result) self.assertTrue(updated) self.assertEqual(content, 'UPDATED_FILE')
def test_save_file_with_existing_si_and_correct_we_should_return_true( self): si = ''.join(random.sample(string.letters + string.digits, 12)) we = ''.join(random.sample(string.letters + string.digits, 12)) result = save_file(si, StringIO.StringIO('NEW_FILE_WITH_WE'), 20, we) updated = save_file(si, StringIO.StringIO('UPDATED_FILE'), 20, we) fp = open(os.path.join(FILE_STORE, si), 'r') content = fp.read() fp.close() self.assertTrue(result) self.assertTrue(updated) self.assertEqual(content, 'UPDATED_FILE')
def test_save_new_file_should_return_true(self): si = ''.join(random.sample(string.letters + string.digits, 12)) return_code = save_file(si, StringIO.StringIO('TESTING'), 20) fp = open(os.path.join(FILE_STORE, si), 'r') result = fp.read() fp.close() self.assertTrue(return_code) self.assertEqual(result, 'TESTING')
def test_save_new_file_should_return_true(self): si = ''.join(random.sample(string.letters+string.digits, 12)) return_code = save_file(si, StringIO.StringIO('TESTING'), 20) fp = open(os.path.join(FILE_STORE, si), 'r') result = fp.read() fp.close() self.assertTrue(return_code) self.assertEqual(result, 'TESTING')
def test_save_file_with_existing_si_and_wrong_we_should_raise_exception(self): si = ''.join(random.sample(string.letters+string.digits, 12)) we = ''.join(random.sample(string.letters+string.digits, 12)) wrong_we = ''.join(random.sample(string.letters+string.digits, 12)) result = save_file(si, StringIO.StringIO('NEW_FILE_WITH_WE'), 20, we) file_obj = StringIO.StringIO('ERROR') self.assertRaises(FileSystemException, save_file, si, file_obj, 20, wrong_we) fp = open(os.path.join(FILE_STORE, si), 'r') content = fp.read() fp.close() self.assertTrue(result) self.assertEqual(content, 'NEW_FILE_WITH_WE')
def test_save_file_with_existing_si_and_wrong_we_should_raise_exception( self): si = ''.join(random.sample(string.letters + string.digits, 12)) we = ''.join(random.sample(string.letters + string.digits, 12)) wrong_we = ''.join(random.sample(string.letters + string.digits, 12)) result = save_file(si, StringIO.StringIO('NEW_FILE_WITH_WE'), 20, we) file_obj = StringIO.StringIO('ERROR') self.assertRaises(FileSystemException, save_file, si, file_obj, 20, wrong_we) fp = open(os.path.join(FILE_STORE, si), 'r') content = fp.read() fp.close() self.assertTrue(result) self.assertEqual(content, 'NEW_FILE_WITH_WE')
def put_file(request, storage_index=None, write_enabler=None): '''Handles HTTP PUT requests for new or existing files. If write_enabler is provided, it is further processed as an mutable file/directory. ''' if _clean_input(storage_index) is not None: if request.PUT is not None: try: content_length = request.ENV['CONTENT_LENGTH'] save_status = save_file(storage_index, request.PUT, content_length, _clean_input(write_enabler)) return Response(status_code=save_status) except FileSystemException, e: return Response(e.text, status_code=e.code)
def put_file(request, storage_index=None, write_enabler=None): '''Handles HTTP PUT requests for new or existing files. If write_enabler is provided, it is further processed as an mutable file/directory. ''' if _clean_input(storage_index) is not None: if request.PUT is not None: try: content_length = request.ENV['CONTENT_LENGTH'] save_status = save_file( storage_index, request.PUT, content_length, _clean_input(write_enabler) ) return Response(status_code=save_status) except FileSystemException, e: return Response(e.text, status_code=e.code)