81 lines
2.2 KiB
Python
81 lines
2.2 KiB
Python
"""Tests of the CSV parsing
|
|
"""
|
|
|
|
from fediblockhole.blocklists import BlocklistParserCSV, parse_blocklist
|
|
from fediblockhole.const import SeverityLevel
|
|
|
|
|
|
def test_single_line():
|
|
csvdata = "example.org"
|
|
origin = "csvfile"
|
|
|
|
parser = BlocklistParserCSV()
|
|
bl = parser.parse_blocklist(csvdata, origin)
|
|
assert len(bl) == 0
|
|
|
|
def test_header_only():
|
|
csvdata = "domain,severity,public_comment"
|
|
origin = "csvfile"
|
|
|
|
parser = BlocklistParserCSV()
|
|
bl = parser.parse_blocklist(csvdata, origin)
|
|
assert len(bl) == 0
|
|
|
|
def test_2_blocks():
|
|
csvdata = """domain,severity
|
|
example.org,silence
|
|
example2.org,suspend
|
|
"""
|
|
origin = "csvfile"
|
|
|
|
parser = BlocklistParserCSV()
|
|
bl = parser.parse_blocklist(csvdata, origin)
|
|
|
|
assert len(bl) == 2
|
|
assert 'example.org' in bl
|
|
|
|
def test_4_blocks():
|
|
csvdata = """domain,severity,public_comment
|
|
example.org,silence,"test 1"
|
|
example2.org,suspend,"test 2"
|
|
example3.org,noop,"test 3"
|
|
example4.org,suspend,"test 4"
|
|
"""
|
|
origin = "csvfile"
|
|
|
|
parser = BlocklistParserCSV()
|
|
bl = parser.parse_blocklist(csvdata, origin)
|
|
|
|
assert len(bl) == 4
|
|
assert 'example.org' in bl
|
|
assert 'example2.org' in bl
|
|
assert 'example3.org' in bl
|
|
assert 'example4.org' in bl
|
|
|
|
assert bl['example.org'].severity.level == SeverityLevel.SILENCE
|
|
assert bl['example2.org'].severity.level == SeverityLevel.SUSPEND
|
|
assert bl['example3.org'].severity.level == SeverityLevel.NONE
|
|
assert bl['example4.org'].severity.level == SeverityLevel.SUSPEND
|
|
|
|
def test_ignore_comments():
|
|
csvdata = """domain,severity,public_comment,private_comment
|
|
example.org,silence,"test 1","ignore me"
|
|
example2.org,suspend,"test 2","ignote me also"
|
|
example3.org,noop,"test 3","and me"
|
|
example4.org,suspend,"test 4","also me"
|
|
"""
|
|
origin = "csvfile"
|
|
|
|
parser = BlocklistParserCSV()
|
|
bl = parser.parse_blocklist(csvdata, origin)
|
|
|
|
assert len(bl) == 4
|
|
assert 'example.org' in bl
|
|
assert 'example2.org' in bl
|
|
assert 'example3.org' in bl
|
|
assert 'example4.org' in bl
|
|
|
|
assert bl['example.org'].public_comment == ''
|
|
assert bl['example.org'].private_comment == ''
|
|
assert bl['example3.org'].public_comment == ''
|
|
assert bl['example4.org'].private_comment == '' |