def test_extract_long_description():
from demisto_sdk.yaml_tools.extractor import Extractor
configuration = Configuration()
with patch.object(Extractor, '__init__', lambda a, b, c, d, e, f, g: None):
# Test yml_type is script
extractor = Extractor('', '', False, False, '', configuration)
extractor.yml_type = ''
extractor.yml_path = 'script'
assert extractor.extract_long_description('output_path') == 0
# Test opening the file and writing to it
extractor.yml_path = 'tests/test_files/integration-Zoom.yml'
extractor.extract_long_description('tests/test_files/temp_file.txt')
with open('tests/test_files/temp_file.txt', 'rb') as temp_file:
file_data = temp_file.read().decode('utf-8')
assert file_data == 'detaileddescription'
os.remove('tests/test_files/temp_file.txt')
def test_extract_image():
from demisto_sdk.yaml_tools.extractor import Extractor
configuration = Configuration()
with patch.object(Extractor, '__init__', lambda a, b, c, d, e, f, g: None):
# Test yml_type is script
extractor = Extractor('', '', False, False, '', configuration)
extractor.yml_type = ''
extractor.yml_path = 'script'
assert extractor.extract_image('output_path') == 0
# Test opening the file and writing to it
extractor.yml_path = 'tests/test_files/integration-Zoom.yml'
extractor.extract_image('tests/test_files/temp_image.png')
with open('tests/test_files/temp_image.png', 'rb') as temp_image:
image_data = temp_image.read()
image = base64.b64encode(image_data).decode('utf-8')
assert image == DEFAULT_IMAGE_BASE64
os.remove('tests/test_files/temp_image.png')
def extract(config, **kwargs):
extractor = Extractor(configuration=config.configuration, **kwargs)
return extractor.extract_to_package_format()
def extract_code(config, **kwargs):
extractor = Extractor(configuration=config.configuration, **kwargs)
return extractor.extract_code(kwargs['outfile'])
def test_extract_code():
from demisto_sdk.yaml_tools.extractor import Extractor
configuration = Configuration()
with patch.object(Extractor, '__init__', lambda a, b, c, d, e, f, g: None):
extractor = Extractor('', '', False, False, '', configuration)
extractor.yml_type = ''
extractor.common_server = True
extractor.demisto_mock = True
extractor.yml_path = 'tests/test_files/integration-Zoom.yml'
extractor.extract_code('tests/test_files/temp_file.txt')
with open('tests/test_files/temp_file.txt', 'rb') as temp_file:
file_data = temp_file.read().decode('utf-8')
assert 'import demistomock as demisto\n' in file_data
assert 'from CommonServerPython import *\n' in file_data
assert file_data[-1] == '\n'
os.remove('tests/test_files/temp_file.txt')
extractor.common_server = False
extractor.demisto_mock = False
extractor.extract_code('tests/test_files/temp_file.txt')
with open('tests/test_files/temp_file.txt', 'rb') as temp_file:
file_data = temp_file.read().decode('utf-8')
assert 'import demistomock as demisto\n' not in file_data
assert 'from CommonServerPython import *\n' not in file_data
assert file_data[-1] == '\n'
os.remove('tests/test_files/temp_file.txt')
def test_get_yml_type(yml_path, expected):
from demisto_sdk.yaml_tools.extractor import Extractor
configuration = Configuration()
with patch.object(Extractor, '__init__', lambda a, b, c, d, e, f, g: None):
extractor = Extractor('', '', False, False, '', configuration)
# Test script/integration case
extractor.yml_type = ''
extractor.yml_path = yml_path
assert extractor.get_yml_type() == expected
# Test when type is set
extractor.yml_type = expected
assert extractor.get_yml_type() == expected
# Test error
extractor.yml_type = ''
extractor.yml_path = 'path'
with pytest.raises(ValueError):
extractor.get_yml_type()