mirror
/
qutebrowser
mirror of https://github.com/qutebrowser/qutebrowser.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Use ResourceType enum for conversion

This commit is contained in:
Florian Bruhin 2020-12-19 13:24:07 +01:00 committed by Árni Dagur
parent 31dbffe35a
commit 8b7bb2da57
2 changed files with 1 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

BIN
tests/end2end/data/brave-adblock/ublock-matches.tsv.gz
View File

Binary file not shown.

13
tests/unit/components/test_braveadblock.py
View File

@ -95,23 +95,12 @@ def run_function_on_dataset(given_function):
contains tuples of (url, source_url, type) in each line. We give these
to values to the given function, row by row.
"""
dataset_type_to_enum = {
"unknown": ResourceType.unknown,
"image": ResourceType.image,
"stylesheet": ResourceType.stylesheet,
"media": ResourceType.media,
"script": ResourceType.script,
"font_resource": ResourceType.font_resource,
"xhr": ResourceType.xhr,
"sub_frame": ResourceType.sub_frame,
}
dataset = utils.adblock_dataset_tsv()
reader = csv.DictReader(dataset, delimiter="\t")
for row in reader:
url = QUrl(row["url"])
source_url = QUrl(row["source_url"])
resource_type = dataset_type_to_enum[row["type"]]
resource_type = ResourceType[row["type"]]
given_function(url, source_url, resource_type)