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