Rollback string manipulation in favor of fixed unit tests

This commit is contained in:
Jason Ertel
2022-07-05 16:21:27 -04:00
parent f6266b19cc
commit 9d43b7ec89

View File

@@ -17,7 +17,7 @@ def searchFile(artifact, csvfiles):
dir = os.path.dirname(os.path.realpath(__file__)) dir = os.path.dirname(os.path.realpath(__file__))
found = [] found = []
for f in csvfiles: for f in csvfiles:
filename = dir + "/" + f.strip("'") filename = dir + "/" + f
with open(filename, "r") as csvfile: with open(filename, "r") as csvfile:
csvdata = csv.DictReader(csvfile) csvdata = csv.DictReader(csvfile)
for row in csvdata: for row in csvdata:
@@ -58,7 +58,7 @@ def analyze(conf, input):
meta = helpers.loadMetadata(__file__) meta = helpers.loadMetadata(__file__)
data = helpers.parseArtifact(input) data = helpers.parseArtifact(input)
helpers.checkSupportedType(meta, data["artifactType"]) helpers.checkSupportedType(meta, data["artifactType"])
search = searchFile(data["value"], conf['file_path'].strip("[]").split(', ')) search = searchFile(data["value"], conf['file_path'])
results = prepareResults(search) results = prepareResults(search)
return results return results