def test_get_code_file():
# Test integration case
unifier = IntegrationScriptUnifier(f"{git_path()}/demisto_sdk/tests/test_files/VulnDB/")
assert unifier.get_code_file(".py") == f"{git_path()}/demisto_sdk/tests/test_files/VulnDB/VulnDB.py"
unifier = IntegrationScriptUnifier(f"{git_path()}/demisto_sdk/tests/test_files/Unifier/SampleNoPyFile")
with pytest.raises(Exception):
unifier.get_code_file(".py")
# Test script case
unifier = IntegrationScriptUnifier(f"{git_path()}/demisto_sdk/tests/test_files/CalculateGeoDistance/")
assert unifier.get_code_file(".py") == f"{git_path()}/demisto_sdk/tests/test_files/CalculateGeoDistance/" \
f"CalculateGeoDistance.py"
def test_get_code_file_case_insensative(tmp_path):
# Create an integration dir with some files
integration_dir = tmp_path / "TestDummyInt"
os.makedirs(integration_dir)
open(integration_dir / "Dummy.ps1", 'a')
open(integration_dir / "ADummy.tests.ps1", 'a') # a test file which is named such a way that it comes up first
unifier = IntegrationScriptUnifier(str(integration_dir))
assert unifier.get_code_file(".ps1") == str(integration_dir / "Dummy.ps1")