def test_export_workflow(self): from walkoff.helpers import strip_device_ids, strip_argument_ids playbook = execution_db_help.standard_load() response = self.get_with_status_check('/api/playbooks/{}?mode=export'.format(playbook.id), headers=self.headers) expected = PlaybookSchema().dump(playbook) strip_device_ids(expected) strip_argument_ids(expected) self.assertDictEqual(response, expected)
def test_export_workflow(self): from walkoff.helpers import strip_device_ids, strip_argument_ids playbook = execution_db_help.standard_load() response = self.get_with_status_check( '/api/playbooks/{}?mode=export'.format(playbook.id), headers=self.headers) expected = PlaybookSchema().dump(playbook).data strip_device_ids(expected) strip_argument_ids(expected) self.assertDictEqual(response, expected)
def __func(playbook): playbook_json = playbook_schema.dump(playbook) if mode == "export": strip_device_ids(playbook_json) strip_argument_ids(playbook_json) f = BytesIO() f.write(json.dumps(playbook_json, sort_keys=True, indent=4, separators=(',', ': ')).encode('utf-8')) f.seek(0) return send_file(f, attachment_filename=playbook.name + '.playbook', as_attachment=True), SUCCESS else: return playbook_json, SUCCESS
def __func(playbook): playbook_json = playbook_schema.dump(playbook).data if mode == "export": strip_device_ids(playbook_json) strip_argument_ids(playbook_json) f = BytesIO() f.write( json.dumps(playbook_json, sort_keys=True, indent=4, separators=(',', ': ')).encode('utf-8')) f.seek(0) return send_file(f, attachment_filename=playbook.name + '.playbook', as_attachment=True), SUCCESS else: return playbook_json, SUCCESS