Skip to content

Commit 02df27d

Browse files
authored
Update Phabricator mock to have libmozdata User-Agent set (#107)
1 parent 6973fde commit 02df27d

File tree

1 file changed

+13
-0
lines changed

1 file changed

+13
-0
lines changed

tests/conftest.py

+13
Original file line numberDiff line numberDiff line change
@@ -5,8 +5,10 @@
55

66
import asyncio
77
import collections
8+
from configparser import ConfigParser
89
import json
910
import os.path
11+
import tempfile
1012
import urllib.parse
1113
from contextlib import contextmanager
1214
from datetime import datetime, timedelta
@@ -345,6 +347,17 @@ def mock_taskcluster():
345347

346348
class MockBuild(PhabricatorBuild):
347349
def __init__(self, diff_id, repo_phid, revision_id, target_phid, diff):
350+
config_file = tempfile.NamedTemporaryFile()
351+
with open(config_file.name, "w") as f:
352+
custom_conf = ConfigParser()
353+
custom_conf.add_section("User-Agent")
354+
custom_conf.set("User-Agent", "name", "libmozdata")
355+
custom_conf.write(f)
356+
f.seek(0)
357+
from libmozdata import config
358+
359+
config.set_config(config.ConfigIni(config_file.name))
360+
348361
self.diff_id = diff_id
349362
self.repo_phid = repo_phid
350363
self.revision_id = revision_id

0 commit comments

Comments
 (0)