def generate_script(self, files):
self.__prepare_output_folder()
mock_files = {}
output_file_name = os.path.join(self.work_dir, self.__MOCK_SCRIPT_NAME)
# generate file names and content
for k, v in files.items():
mock_files.update(self.__create_mock_file(k, v, write_files=False))
# add files to script
template = read_template('mock_script_template.py')
output = template.render(mocks=pickle.dumps(mock_files))
write_all_text(output_file_name, output)
def generate_script(self, files):
self.__prepare_output_folder()
mock_files = {}
output_file_name = os.path.join(self.work_dir, self.__MOCK_SCRIPT_NAME)
# generate file names and content
for k, v in files.items():
mock_files.update(self.__create_mock_file(k, v, write_files=False))
# add files to script
template = read_template('mock_script_template.py')
output = template.render(mocks=pickle.dumps(mock_files))
write_all_text(output_file_name, output)
def __init__(self, output_folder):
self.output_folder = output_folder
self.template = read_template('stubs_template.py')
def __init__(self, output_folder):
self.output_folder = output_folder
self.template = read_template('stubs_template.py')
