def test_redact_everything():
test_words = 'The quick brown fox jumped lazily over the fence.'
test_string = 'The quick brown fox jumped lazily over the fence.'
expected = 'XXXXXXXX XXXXXXXX XXXXXXXX XXXXXXXX XXXXXXXX XXXXXXXX XXXXXXXX XXXXXXXX XXXXXXXX.'
res = redactor.redact(test_words, test_string)
assert res == expected
def test_space_then_comma():
test_words = 'fox ,lazily ,fence' # Spaces and then commas between the words
test_string = 'The quick brown fox jumped lazily over the fence.'
expected = 'The quick brown XXXXXXXX jumped XXXXXXXX over the XXXXXXXX.'
res = redactor.redact(test_words, test_string)
assert res == expected
def test_simple_redact():
test_words = 'fox, lazily, fence'
test_string = 'The quick brown fox jumped lazily over the fence.'
expected = 'The quick brown XXXXXXXX jumped XXXXXXXX over the XXXXXXXX.'
res = redactor.redact(test_words, test_string)
assert res == expected
def test_spaces_only():
test_words = 'fox lazily fence' # NO commas between the words
test_string = 'The quick brown fox jumped lazily over the fence.'
expected = 'The quick brown XXXXXXXX jumped XXXXXXXX over the XXXXXXXX.'
res = redactor.redact(test_words, test_string)
assert res == expected
def test_no_redactions():
test_words = 'python, apples, firefox'
test_string = 'The quick brown fox jumped lazily over the fence.'
expected = 'The quick brown fox jumped lazily over the fence.'
res = redactor.redact(test_words, test_string)
assert res == expected
def index():
if request.method == 'GET':
return render_template("index.html")
elif request.method == 'POST':
inFile = request.files['secretFile']
if inFile.filename == '':
abort(404, 'No File provided')
inFile = inFile.read().decode('UTF-8')
words = request.form['sensitiveWords']
if words == '':
abort(404, 'No words provided to redact.')
redactedText = redactor.redact(words, inFile)
FileOps = FileOperations.FileOperations()
FileOps.save_redacted(redactedText)
return render_template("results.html",
original=inFile,
redacted=redactedText,
filename=FileOps.out_file)