def api():
from auslib.web.admin.base import app
app.config["SECRET_KEY"] = "notasecret"
app.config["CORS_ORIGINS"] = "*"
app.config["AUTH_DOMAIN"] = "balrog.test.dev"
app.config["AUTH_AUDIENCE"] = "balrog test"
app.config["M2M_ACCOUNT_MAPPING"] = {}
app.config["WHITELISTED_DOMAINS"] = {
"download.mozilla.org": ("Firefox", ),
"archive.mozilla.org": ("Firefox", ),
"cdmdownload.adobe.com": ("CDM", ),
}
return app.test_client()
def setUp(self):
self.version_fd, self.version_file = mkstemp()
cache.reset()
cache.make_copies = True
app.config["SECRET_KEY"] = 'abc123'
app.config['DEBUG'] = True
app.config["WTF_CSRF_ENABLED"] = False
app.config['WHITELISTED_DOMAINS'] = {'good.com': ('a', 'b', 'c', 'd')}
app.config["VERSION_FILE"] = self.version_file
with open(self.version_file, "w+") as f:
f.write("""
{
"source":"https://github.com/mozilla/balrog",
"version":"1.0",
"commit":"abcdef123456"
}
""")
dbo.setDb('sqlite:///:memory:')
dbo.setDomainWhitelist({'good.com': ('a', 'b', 'c', 'd')})
dbo.create()
dbo.permissions.t.insert().execute(permission='admin', username='******', data_version=1)
dbo.permissions.t.insert().execute(permission='permission', username='******', data_version=1)
dbo.permissions.t.insert().execute(permission='release', username='******',
options=dict(products=['fake', "a", 'b'], actions=["create", "modify"]), data_version=1)
dbo.permissions.t.insert().execute(permission='release_read_only', username='******', options=dict(actions=["set"], products=["a", "b"]), data_version=1)
dbo.permissions.t.insert().execute(permission='rule', username='******', options=dict(actions=["modify"], products=['a', "b"]), data_version=1)
dbo.permissions.t.insert().execute(permission='release', username='******', options=dict(actions=["modify"], products=['a']), data_version=1)
dbo.permissions.t.insert().execute(permission="scheduled_change", username="******", options=dict(actions=["enact"]), data_version=1)
dbo.permissions.t.insert().execute(permission='release_locale', username='******',
options=dict(actions=["modify"], products=['a']), data_version=1)
dbo.permissions.t.insert().execute(permission='admin', username='******',
options=dict(products=['a']), data_version=1)
dbo.permissions.user_roles.t.insert().execute(username="******", role="releng", data_version=1)
dbo.permissions.user_roles.t.insert().execute(username="******", role="qa", data_version=1)
dbo.permissions.user_roles.t.insert().execute(username="******", role="relman", data_version=1)
dbo.permissions.user_roles.t.insert().execute(username="******", role="releng", data_version=1)
dbo.permissions.user_roles.t.insert().execute(username="******", role="relman", data_version=1)
dbo.productRequiredSignoffs.t.insert().execute(product="fake", channel="a", role="releng", signoffs_required=1, data_version=1)
dbo.productRequiredSignoffs.t.insert().execute(product="fake", channel="e", role="releng", signoffs_required=1, data_version=1)
dbo.productRequiredSignoffs.t.insert().execute(product="fake", channel="j", role="releng", signoffs_required=1, data_version=1)
dbo.productRequiredSignoffs.t.insert().execute(product="fake", channel="k", role="relman", signoffs_required=1, data_version=2)
dbo.productRequiredSignoffs.history.t.insert().execute(change_id=1, changed_by="bill", timestamp=10, product="fake", channel="k", role="relman")
dbo.productRequiredSignoffs.history.t.insert().execute(change_id=2, changed_by="bill", timestamp=11, product="fake", channel="k", role="relman",
signoffs_required=2, data_version=1)
dbo.productRequiredSignoffs.history.t.insert().execute(change_id=3, changed_by="bill", timestamp=25, product="fake", channel="k", role="relman",
signoffs_required=1, data_version=2)
dbo.permissionsRequiredSignoffs.t.insert().execute(product="fake", role="releng", signoffs_required=1, data_version=1)
dbo.permissionsRequiredSignoffs.t.insert().execute(product="bar", role="releng", signoffs_required=1, data_version=1)
dbo.permissionsRequiredSignoffs.t.insert().execute(product="blah", role="releng", signoffs_required=1, data_version=1)
dbo.permissionsRequiredSignoffs.t.insert().execute(product="doop", role="releng", signoffs_required=1, data_version=2)
dbo.permissionsRequiredSignoffs.t.insert().execute(product="superfake", role="relman", signoffs_required=1, data_version=1)
dbo.permissionsRequiredSignoffs.history.t.insert().execute(change_id=1, changed_by="bill", timestamp=10, product="doop", role="releng")
dbo.permissionsRequiredSignoffs.history.t.insert().execute(change_id=2, changed_by="bill", timestamp=11, product="doop", role="releng",
signoffs_required=2, data_version=1)
dbo.permissionsRequiredSignoffs.history.t.insert().execute(change_id=3, changed_by="bill", timestamp=25, product="doop", role="releng",
signoffs_required=1, data_version=2)
dbo.releases.t.insert().execute(
name='a', product='a', data=createBlob(dict(name='a', hashFunction="sha512", schema_version=1)), data_version=1)
dbo.releases.t.insert().execute(
name='ab', product='a', data=createBlob(dict(name='ab', hashFunction="sha512", schema_version=1)), data_version=1)
dbo.releases.t.insert().execute(
name='b', product='b', data=createBlob(dict(name='b', hashFunction="sha512", schema_version=1)), data_version=1)
dbo.releases.t.insert().execute(
name='c', product='c', data=createBlob(dict(name='c', hashFunction="sha512", schema_version=1)), data_version=1)
dbo.releases.t.insert().execute(name='d', product='d', data_version=1, data=createBlob("""
{
"name": "d",
"schema_version": 1,
"hashFunction": "sha512",
"platforms": {
"p": {
"locales": {
"d": {
"complete": {
"filesize": 1234,
"from": "*",
"hashValue": "abc"
}
}
}
}
}
}
"""))
dbo.rules.t.insert().execute(
rule_id=1, priority=100, version='3.5', buildTarget='d', backgroundRate=100, mapping='c', update_type='minor',
product="a", channel="a", data_version=1
)
dbo.rules.t.insert().execute(
rule_id=2, alias="frodo", priority=100, version='3.3', buildTarget='d', backgroundRate=100, mapping='b', update_type='minor',
product="a", channel="a", data_version=1
)
dbo.rules.t.insert().execute(
rule_id=3, product='a', priority=100, version='3.5', buildTarget='a', backgroundRate=100, mapping='a', update_type='minor',
channel="a", data_version=1
)
dbo.rules.t.insert().execute(
rule_id=4, product='fake', priority=80, buildTarget='d', backgroundRate=100, mapping='a', update_type='minor', channel="a",
data_version=1
)
dbo.rules.t.insert().execute(
rule_id=5, priority=80, buildTarget='d', version='3.3', backgroundRate=0, mapping='c', update_type='minor',
product="a", channel="a", data_version=1
)
dbo.rules.t.insert().execute(rule_id=6, product='fake', priority=40, backgroundRate=50, mapping='a', update_type='minor', channel="e", data_version=1)
dbo.rules.t.insert().execute(rule_id=7, product='fake', priority=30, backgroundRate=85, mapping='a', update_type='minor', channel="c", data_version=1)
self.client = app.test_client()
def setUp(self):
from auslib.web.admin.views import base as view_base
self.view_base = view_base
# Mock out verified_userinfo, because we don't want to talk to Auth0
# or need to provide real credentials in tests.
# We don't do this with "mock" because this is a base to all of other
# tests, and "mock" must be applied in the tests itself.
self.orig_base_verified_userinfo = view_base.verified_userinfo
self.mocked_user = None
def my_userinfo(*args, **kwargs):
return {"email": self.mocked_user}
view_base.verified_userinfo = my_userinfo
self.version_fd, self.version_file = mkstemp()
cache.reset()
cache.make_copies = True
app.config["SECRET_KEY"] = "abc123"
app.config["DEBUG"] = True
app.config["WTF_CSRF_ENABLED"] = False
app.config["WHITELISTED_DOMAINS"] = {"good.com": ("a", "b", "c", "d")}
app.config["VERSION_FILE"] = self.version_file
app.config["AUTH_DOMAIN"] = "balrog.test.dev"
app.config["AUTH_AUDIENCE"] = "balrog test"
app.config["M2M_ACCOUNT_MAPPING"] = {}
app.config["CORS_ORIGINS"] = "*"
with open(self.version_file, "w+") as f:
f.write("""
{
"source":"https://github.com/mozilla/balrog",
"version":"1.0",
"commit":"abcdef123456"
}
""")
dbo.setDb("sqlite:///:memory:")
dbo.setDomainWhitelist({"good.com": ("a", "b", "c", "d")})
self.metadata.create_all(dbo.engine)
dbo.permissions.t.insert().execute(permission="admin",
username="******",
data_version=1)
dbo.permissions.t.insert().execute(permission="permission",
username="******",
data_version=1)
dbo.permissions.t.insert().execute(permission="release",
username="******",
options=dict(
products=["fake", "a", "b"],
actions=["create", "modify"]),
data_version=1)
dbo.permissions.t.insert().execute(permission="release",
username="******",
options=dict(products=["a"],
actions=["modify"]),
data_version=1)
dbo.permissions.t.insert().execute(permission="rule",
username="******",
options=dict(products=["fake"],
actions=["create"]),
data_version=1)
dbo.permissions.t.insert().execute(permission="release_read_only",
username="******",
options=dict(actions=["set"],
products=["a", "b"]),
data_version=1)
dbo.permissions.t.insert().execute(permission="rule",
username="******",
options=dict(actions=["modify"],
products=["a", "b"]),
data_version=1)
dbo.permissions.t.insert().execute(permission="release",
username="******",
options=dict(actions=["modify"],
products=["a"]),
data_version=1)
dbo.permissions.t.insert().execute(permission="scheduled_change",
username="******",
options=dict(actions=["enact"]),
data_version=1)
dbo.permissions.t.insert().execute(permission="release_locale",
username="******",
options=dict(actions=["modify"],
products=["a"]),
data_version=1)
dbo.permissions.t.insert().execute(permission="admin",
username="******",
options=dict(products=["a"]),
data_version=1)
dbo.permissions.user_roles.t.insert().execute(username="******",
role="releng",
data_version=1)
dbo.permissions.user_roles.t.insert().execute(username="******",
role="qa",
data_version=1)
dbo.permissions.user_roles.t.insert().execute(username="******",
role="relman",
data_version=1)
dbo.permissions.user_roles.t.insert().execute(username="******",
role="releng",
data_version=1)
dbo.permissions.user_roles.t.insert().execute(username="******",
role="relman",
data_version=1)
dbo.productRequiredSignoffs.t.insert().execute(product="fake",
channel="a",
role="releng",
signoffs_required=1,
data_version=1)
dbo.productRequiredSignoffs.t.insert().execute(product="fake",
channel="e",
role="releng",
signoffs_required=1,
data_version=1)
dbo.productRequiredSignoffs.t.insert().execute(product="fake",
channel="j",
role="releng",
signoffs_required=1,
data_version=1)
dbo.productRequiredSignoffs.t.insert().execute(product="fake",
channel="k",
role="relman",
signoffs_required=1,
data_version=2)
dbo.productRequiredSignoffs.history.t.insert().execute(
change_id=1,
changed_by="bill",
timestamp=10,
product="fake",
channel="k",
role="relman")
dbo.productRequiredSignoffs.history.t.insert().execute(
change_id=2,
changed_by="bill",
timestamp=11,
product="fake",
channel="k",
role="relman",
signoffs_required=2,
data_version=1)
dbo.productRequiredSignoffs.history.t.insert().execute(
change_id=3,
changed_by="bill",
timestamp=25,
product="fake",
channel="k",
role="relman",
signoffs_required=1,
data_version=2)
dbo.permissionsRequiredSignoffs.t.insert().execute(product="fake",
role="releng",
signoffs_required=1,
data_version=1)
dbo.permissionsRequiredSignoffs.t.insert().execute(product="bar",
role="releng",
signoffs_required=1,
data_version=1)
dbo.permissionsRequiredSignoffs.t.insert().execute(product="blah",
role="releng",
signoffs_required=1,
data_version=1)
dbo.permissionsRequiredSignoffs.t.insert().execute(product="doop",
role="releng",
signoffs_required=1,
data_version=2)
dbo.permissionsRequiredSignoffs.t.insert().execute(product="superfake",
role="relman",
signoffs_required=1,
data_version=1)
dbo.permissionsRequiredSignoffs.history.t.insert().execute(
change_id=1,
changed_by="bill",
timestamp=10,
product="doop",
role="releng")
dbo.permissionsRequiredSignoffs.history.t.insert().execute(
change_id=2,
changed_by="bill",
timestamp=11,
product="doop",
role="releng",
signoffs_required=2,
data_version=1)
dbo.permissionsRequiredSignoffs.history.t.insert().execute(
change_id=3,
changed_by="bill",
timestamp=25,
product="doop",
role="releng",
signoffs_required=1,
data_version=2)
dbo.releases.t.insert().execute(name="a",
product="a",
data=createBlob(
dict(name="a",
hashFunction="sha512",
schema_version=1)),
data_version=1)
dbo.releases.t.insert().execute(name="ab",
product="a",
data=createBlob(
dict(name="ab",
hashFunction="sha512",
schema_version=1)),
data_version=1)
dbo.releases.history.t.insert().execute(change_id=1,
timestamp=5,
changed_by="bill",
name="ab")
dbo.releases.history.t.insert().execute(
change_id=2,
timestamp=6,
changed_by="bill",
name="ab",
product="a",
data=createBlob(
dict(name="ab", hashFunction="sha512", schema_version=1)),
data_version=1,
)
dbo.releases.t.insert().execute(name="b",
product="b",
data=createBlob(
dict(name="b",
hashFunction="sha512",
schema_version=1)),
data_version=1)
dbo.releases.history.t.insert().execute(change_id=5,
timestamp=15,
changed_by="bill",
name="b")
dbo.releases.history.t.insert().execute(
change_id=6,
timestamp=16,
changed_by="bill",
name="b",
product="b",
data=createBlob(
dict(name="b", hashFunction="sha512", schema_version=1)),
data_version=1,
)
dbo.releases.t.insert().execute(name="c",
product="c",
data=createBlob(
dict(name="c",
hashFunction="sha512",
schema_version=1)),
data_version=1)
dbo.releases.t.insert().execute(
name="d",
product="d",
data_version=1,
data=createBlob("""
{
"name": "d",
"schema_version": 1,
"hashFunction": "sha512",
"platforms": {
"p": {
"locales": {
"d": {
"complete": {
"filesize": 1234,
"from": "*",
"hashValue": "abc"
}
}
}
}
}
}
"""),
)
dbo.releases.history.t.insert().execute(change_id=3,
timestamp=9,
changed_by="bill",
name="d")
dbo.releases.history.t.insert().execute(
change_id=4,
timestamp=10,
changed_by="bill",
name="d",
product="d",
data_version=1,
data=createBlob("""
{
"name": "d",
"schema_version": 1,
"hashFunction": "sha512",
"platforms": {
"p": {
"locales": {
"d": {
"complete": {
"filesize": 1234,
"from": "*",
"hashValue": "abc"
}
}
}
}
}
}
"""),
)
dbo.rules.t.insert().execute(
rule_id=1,
priority=100,
version="3.5",
buildTarget="d",
backgroundRate=100,
mapping="c",
update_type="minor",
product="a",
channel="a",
data_version=1,
)
dbo.rules.t.insert().execute(
rule_id=2,
alias="frodo",
priority=100,
version="3.3",
buildTarget="d",
backgroundRate=100,
mapping="b",
update_type="minor",
product="a",
channel="a",
data_version=1,
)
dbo.rules.t.insert().execute(
rule_id=3,
product="a",
priority=100,
version="3.5",
buildTarget="a",
backgroundRate=100,
mapping="a",
update_type="minor",
channel="a",
data_version=1,
)
dbo.rules.t.insert().execute(rule_id=4,
product="fake",
priority=80,
buildTarget="d",
backgroundRate=100,
mapping="a",
update_type="minor",
channel="a",
data_version=1)
dbo.rules.t.insert().execute(rule_id=5,
priority=80,
buildTarget="d",
version="3.3",
backgroundRate=0,
mapping="c",
update_type="minor",
product="a",
channel="a",
data_version=1)
dbo.rules.t.insert().execute(rule_id=6,
product="fake",
priority=40,
backgroundRate=50,
mapping="a",
update_type="minor",
channel="e",
data_version=1)
dbo.rules.t.insert().execute(rule_id=7,
product="fake",
priority=30,
backgroundRate=85,
mapping="a",
update_type="minor",
channel="c",
data_version=1)
dbo.rules.t.insert().execute(rule_id=8,
product="fake2",
priority=25,
backgroundRate=100,
mapping="a",
update_type="minor",
channel="c",
mig64=True,
data_version=1)
dbo.rules.t.insert().execute(rule_id=9,
product="fake3",
priority=25,
backgroundRate=100,
mapping="a",
update_type="minor",
channel="c",
jaws=True,
data_version=1)
self.client = app.test_client()
def setUp(self):
self.version_fd, self.version_file = mkstemp()
cache.reset()
cache.make_copies = True
app.config["SECRET_KEY"] = 'abc123'
app.config['DEBUG'] = True
app.config["WTF_CSRF_ENABLED"] = False
app.config['WHITELISTED_DOMAINS'] = {'good.com': ('a', 'b', 'c', 'd')}
app.config["VERSION_FILE"] = self.version_file
with open(self.version_file, "w+") as f:
f.write("""
{
"source":"https://github.com/mozilla/balrog",
"version":"1.0",
"commit":"abcdef123456"
}
""")
dbo.setDb('sqlite:///:memory:')
dbo.setDomainWhitelist({'good.com': ('a', 'b', 'c', 'd')})
dbo.create()
dbo.permissions.t.insert().execute(permission='admin',
username='******',
data_version=1)
dbo.permissions.t.insert().execute(permission='permission',
username='******',
data_version=1)
dbo.permissions.t.insert().execute(permission='release',
username='******',
options=dict(
products=['fake', "a", 'b'],
actions=["create", "modify"]),
data_version=1)
dbo.permissions.t.insert().execute(permission='release_read_only',
username='******',
options=dict(actions=["set"],
products=["a", "b"]),
data_version=1)
dbo.permissions.t.insert().execute(permission='rule',
username='******',
options=dict(actions=["modify"],
products=['a', "b"]),
data_version=1)
dbo.permissions.t.insert().execute(permission='release',
username='******',
options=dict(actions=["modify"],
products=['a']),
data_version=1)
dbo.permissions.t.insert().execute(permission="scheduled_change",
username="******",
options=dict(actions=["enact"]),
data_version=1)
dbo.permissions.t.insert().execute(permission='release_locale',
username='******',
options=dict(actions=["modify"],
products=['a']),
data_version=1)
dbo.permissions.t.insert().execute(permission='admin',
username='******',
options=dict(products=['a']),
data_version=1)
dbo.permissions.user_roles.t.insert().execute(username="******",
role="releng",
data_version=1)
dbo.permissions.user_roles.t.insert().execute(username="******",
role="qa",
data_version=1)
dbo.permissions.user_roles.t.insert().execute(username="******",
role="relman",
data_version=1)
dbo.permissions.user_roles.t.insert().execute(username="******",
role="releng",
data_version=1)
dbo.permissions.user_roles.t.insert().execute(username="******",
role="relman",
data_version=1)
dbo.productRequiredSignoffs.t.insert().execute(product="fake",
channel="a",
role="releng",
signoffs_required=1,
data_version=1)
dbo.productRequiredSignoffs.t.insert().execute(product="fake",
channel="e",
role="releng",
signoffs_required=1,
data_version=1)
dbo.productRequiredSignoffs.t.insert().execute(product="fake",
channel="j",
role="releng",
signoffs_required=1,
data_version=1)
dbo.productRequiredSignoffs.t.insert().execute(product="fake",
channel="k",
role="relman",
signoffs_required=1,
data_version=2)
dbo.productRequiredSignoffs.history.t.insert().execute(
change_id=1,
changed_by="bill",
timestamp=10,
product="fake",
channel="k",
role="relman")
dbo.productRequiredSignoffs.history.t.insert().execute(
change_id=2,
changed_by="bill",
timestamp=11,
product="fake",
channel="k",
role="relman",
signoffs_required=2,
data_version=1)
dbo.productRequiredSignoffs.history.t.insert().execute(
change_id=3,
changed_by="bill",
timestamp=25,
product="fake",
channel="k",
role="relman",
signoffs_required=1,
data_version=2)
dbo.permissionsRequiredSignoffs.t.insert().execute(product="fake",
role="releng",
signoffs_required=1,
data_version=1)
dbo.permissionsRequiredSignoffs.t.insert().execute(product="bar",
role="releng",
signoffs_required=1,
data_version=1)
dbo.permissionsRequiredSignoffs.t.insert().execute(product="blah",
role="releng",
signoffs_required=1,
data_version=1)
dbo.permissionsRequiredSignoffs.t.insert().execute(product="doop",
role="releng",
signoffs_required=1,
data_version=2)
dbo.permissionsRequiredSignoffs.t.insert().execute(product="superfake",
role="relman",
signoffs_required=1,
data_version=1)
dbo.permissionsRequiredSignoffs.history.t.insert().execute(
change_id=1,
changed_by="bill",
timestamp=10,
product="doop",
role="releng")
dbo.permissionsRequiredSignoffs.history.t.insert().execute(
change_id=2,
changed_by="bill",
timestamp=11,
product="doop",
role="releng",
signoffs_required=2,
data_version=1)
dbo.permissionsRequiredSignoffs.history.t.insert().execute(
change_id=3,
changed_by="bill",
timestamp=25,
product="doop",
role="releng",
signoffs_required=1,
data_version=2)
dbo.releases.t.insert().execute(name='a',
product='a',
data=createBlob(
dict(name='a',
hashFunction="sha512",
schema_version=1)),
data_version=1)
dbo.releases.t.insert().execute(name='ab',
product='a',
data=createBlob(
dict(name='ab',
hashFunction="sha512",
schema_version=1)),
data_version=1)
dbo.releases.t.insert().execute(name='b',
product='b',
data=createBlob(
dict(name='b',
hashFunction="sha512",
schema_version=1)),
data_version=1)
dbo.releases.t.insert().execute(name='c',
product='c',
data=createBlob(
dict(name='c',
hashFunction="sha512",
schema_version=1)),
data_version=1)
dbo.releases.t.insert().execute(name='d',
product='d',
data_version=1,
data=createBlob("""
{
"name": "d",
"schema_version": 1,
"hashFunction": "sha512",
"platforms": {
"p": {
"locales": {
"d": {
"complete": {
"filesize": 1234,
"from": "*",
"hashValue": "abc"
}
}
}
}
}
}
"""))
dbo.rules.t.insert().execute(rule_id=1,
priority=100,
version='3.5',
buildTarget='d',
backgroundRate=100,
mapping='c',
update_type='minor',
product="a",
channel="a",
data_version=1)
dbo.rules.t.insert().execute(rule_id=2,
alias="frodo",
priority=100,
version='3.3',
buildTarget='d',
backgroundRate=100,
mapping='b',
update_type='minor',
product="a",
channel="a",
data_version=1)
dbo.rules.t.insert().execute(rule_id=3,
product='a',
priority=100,
version='3.5',
buildTarget='a',
backgroundRate=100,
mapping='a',
update_type='minor',
channel="a",
data_version=1)
dbo.rules.t.insert().execute(rule_id=4,
product='fake',
priority=80,
buildTarget='d',
backgroundRate=100,
mapping='a',
update_type='minor',
channel="a",
data_version=1)
dbo.rules.t.insert().execute(rule_id=5,
priority=80,
buildTarget='d',
version='3.3',
backgroundRate=0,
mapping='c',
update_type='minor',
product="a",
channel="a",
data_version=1)
dbo.rules.t.insert().execute(rule_id=6,
product='fake',
priority=40,
backgroundRate=50,
mapping='a',
update_type='minor',
channel="e",
data_version=1)
dbo.rules.t.insert().execute(rule_id=7,
product='fake',
priority=30,
backgroundRate=85,
mapping='a',
update_type='minor',
channel="c",
data_version=1)
self.client = app.test_client()
def setup(self, insert_release, firefox_56_0_build1):
from auslib.web.admin.views import base as view_base
self.view_base = view_base
# Mock out verified_userinfo, because we don't want to talk to Auth0
# or need to provide real credentials in tests.
# We don't do this with "mock" because this is a base to all of other
# tests, and "mock" must be applied in the tests itself.
self.orig_base_verified_userinfo = view_base.verified_userinfo
self.mocked_user = None
def my_userinfo(*args, **kwargs):
return {"email": self.mocked_user}
view_base.verified_userinfo = my_userinfo
self.version_fd, self.version_file = mkstemp()
cache.reset()
cache.make_copies = True
app.config["SECRET_KEY"] = "abc123"
app.config["DEBUG"] = True
app.config["WTF_CSRF_ENABLED"] = False
app.config["WHITELISTED_DOMAINS"] = {"good.com": ("a", "b", "c", "d")}
app.config["VERSION_FILE"] = self.version_file
app.config["AUTH_DOMAIN"] = "balrog.test.dev"
app.config["AUTH_AUDIENCE"] = "balrog test"
app.config["M2M_ACCOUNT_MAPPING"] = {}
app.config["CORS_ORIGINS"] = "*"
with open(self.version_file, "w+") as f:
f.write("""
{
"source":"https://github.com/mozilla-releng/balrog",
"version":"1.0",
"commit":"abcdef123456"
}
""")
dbo.setDb("sqlite:///:memory:",
releases_history_buckets={"*": "fake"},
releases_history_class=FakeGCSHistory)
self.orig_releases_history = dbo.releases.history
# TODO: need a better mock, or maybe a Fake that stores history in memory?
# only the merge tests benefit from having history
# the rest are fine if we just let history be None
# maybe we should mock this only in the tests we care about?
dbo.releases.history = FakeGCSHistory()
dbo.setDomainWhitelist({"good.com": ("a", "b", "c", "d")})
self.metadata.create_all(dbo.engine)
dbo.permissions.t.insert().execute(permission="admin",
username="******",
data_version=1)
dbo.permissions.t.insert().execute(permission="permission",
username="******",
data_version=1)
dbo.permissions.t.insert().execute(permission="release",
username="******",
options=dict(
products=["fake", "a", "b"],
actions=["create", "modify"]),
data_version=1)
dbo.permissions.t.insert().execute(permission="release",
username="******",
options=dict(products=["a"],
actions=["modify"]),
data_version=1)
dbo.permissions.t.insert().execute(permission="rule",
username="******",
options=dict(products=["fake"],
actions=["create"]),
data_version=1)
dbo.permissions.t.insert().execute(permission="release_read_only",
username="******",
options=dict(actions=["set"],
products=["a", "b"]),
data_version=1)
dbo.permissions.t.insert().execute(permission="rule",
username="******",
options=dict(actions=["modify"],
products=["a", "b"]),
data_version=1)
dbo.permissions.t.insert().execute(permission="release",
username="******",
options=dict(actions=["modify"],
products=["a"]),
data_version=1)
dbo.permissions.t.insert().execute(permission="scheduled_change",
username="******",
options=dict(actions=["enact"]),
data_version=1)
dbo.permissions.t.insert().execute(permission="release_locale",
username="******",
options=dict(actions=["modify"],
products=["a"]),
data_version=1)
dbo.permissions.t.insert().execute(permission="admin",
username="******",
options=dict(products=["a"]),
data_version=1)
dbo.permissions.user_roles.t.insert().execute(username="******",
role="releng",
data_version=1)
dbo.permissions.user_roles.t.insert().execute(username="******",
role="qa",
data_version=1)
dbo.permissions.user_roles.t.insert().execute(username="******",
role="relman",
data_version=1)
dbo.permissions.user_roles.t.insert().execute(username="******",
role="releng",
data_version=1)
dbo.permissions.user_roles.t.insert().execute(username="******",
role="relman",
data_version=1)
dbo.productRequiredSignoffs.t.insert().execute(product="fake",
channel="a",
role="releng",
signoffs_required=1,
data_version=1)
dbo.productRequiredSignoffs.t.insert().execute(product="fake",
channel="e",
role="releng",
signoffs_required=1,
data_version=1)
dbo.productRequiredSignoffs.t.insert().execute(product="fake",
channel="j",
role="releng",
signoffs_required=1,
data_version=1)
dbo.productRequiredSignoffs.t.insert().execute(product="fake",
channel="k",
role="relman",
signoffs_required=1,
data_version=2)
dbo.productRequiredSignoffs.history.t.insert().execute(
change_id=1,
changed_by="bill",
timestamp=10,
product="fake",
channel="k",
role="relman")
dbo.productRequiredSignoffs.history.t.insert().execute(
change_id=2,
changed_by="bill",
timestamp=11,
product="fake",
channel="k",
role="relman",
signoffs_required=2,
data_version=1)
dbo.productRequiredSignoffs.history.t.insert().execute(
change_id=3,
changed_by="bill",
timestamp=25,
product="fake",
channel="k",
role="relman",
signoffs_required=1,
data_version=2)
dbo.permissionsRequiredSignoffs.t.insert().execute(product="fake",
role="releng",
signoffs_required=1,
data_version=1)
dbo.permissionsRequiredSignoffs.t.insert().execute(product="bar",
role="releng",
signoffs_required=1,
data_version=1)
dbo.permissionsRequiredSignoffs.t.insert().execute(product="blah",
role="releng",
signoffs_required=1,
data_version=1)
dbo.permissionsRequiredSignoffs.t.insert().execute(product="doop",
role="releng",
signoffs_required=1,
data_version=2)
dbo.permissionsRequiredSignoffs.t.insert().execute(product="superfake",
role="relman",
signoffs_required=1,
data_version=1)
dbo.permissionsRequiredSignoffs.history.t.insert().execute(
change_id=1,
changed_by="bill",
timestamp=10,
product="doop",
role="releng")
dbo.permissionsRequiredSignoffs.history.t.insert().execute(
change_id=2,
changed_by="bill",
timestamp=11,
product="doop",
role="releng",
signoffs_required=2,
data_version=1)
dbo.permissionsRequiredSignoffs.history.t.insert().execute(
change_id=3,
changed_by="bill",
timestamp=25,
product="doop",
role="releng",
signoffs_required=1,
data_version=2)
dbo.releases.t.insert().execute(name="a",
product="a",
data=createBlob(
dict(name="a",
hashFunction="sha512",
schema_version=1)),
data_version=1)
dbo.releases.t.insert().execute(name="ab",
product="a",
data=createBlob(
dict(name="ab",
hashFunction="sha512",
schema_version=1)),
data_version=1)
dbo.releases.history.bucket.blobs["ab/None-456-bob.json"] = FakeBlob(
"")
dbo.releases.history.bucket.blobs["ab/1-456-bob.json"] = FakeBlob("""
{
"name": "ab",
"hashFunction": "sha512",
"schema_version": 1
}
""")
dbo.releases.t.insert().execute(name="b",
product="b",
data=createBlob(
dict(name="b",
hashFunction="sha512",
schema_version=1)),
data_version=1)
dbo.releases.history.bucket.blobs["b/None-567-bob.json"] = FakeBlob("")
dbo.releases.history.bucket.blobs["b/1-567-bob.json"] = FakeBlob("""
{
"name": "b",
"hashFunction": "sha512",
"schema_version": 1
}
""")
dbo.releases.t.insert().execute(name="c",
product="c",
data=createBlob(
dict(name="c",
hashFunction="sha512",
schema_version=1)),
data_version=1)
dbo.releases.t.insert().execute(
name="d",
product="d",
data_version=1,
data=createBlob("""
{
"name": "d",
"schema_version": 1,
"hashFunction": "sha512",
"platforms": {
"p": {
"locales": {
"d": {
"complete": {
"filesize": 1234,
"from": "*",
"hashValue": "abc"
}
}
}
}
}
}
"""),
)
dbo.releases.history.bucket.blobs["d/None-678-bob.json"] = FakeBlob("")
dbo.releases.history.bucket.blobs["d/1-678-bob.json"] = FakeBlob("""
{
"name": "d",
"schema_version": 1,
"hashFunction": "sha512",
"platforms": {
"p": {
"locales": {
"d": {
"complete": {
"filesize": 1234,
"from": "*",
"hashValue": "abc"
}
}
}
}
}
}
""")
dbo.rules.t.insert().execute(
rule_id=1,
priority=100,
version="3.5",
buildTarget="d",
backgroundRate=100,
mapping="c",
update_type="minor",
product="a",
channel="a",
data_version=2,
)
dbo.rules.history.t.insert().execute(change_id=1,
timestamp=50,
changed_by="bill",
rule_id=1)
dbo.rules.history.t.insert().execute(
change_id=2,
timestamp=51,
changed_by="bill",
rule_id=1,
priority=100,
version="3.5",
buildTarget="d",
backgroundRate=50,
mapping="c",
update_type="minor",
product="a",
channel="a",
data_version=1,
)
dbo.rules.history.t.insert().execute(
change_id=3,
timestamp=65,
changed_by="bill",
rule_id=1,
priority=100,
version="3.5",
buildTarget="d",
backgroundRate=100,
mapping="c",
update_type="minor",
product="a",
channel="a",
data_version=2,
)
dbo.rules.t.insert().execute(
rule_id=2,
alias="frodo",
priority=100,
version="3.3",
buildTarget="d",
backgroundRate=100,
mapping="b",
update_type="minor",
product="a",
channel="a",
data_version=1,
)
dbo.rules.history.t.insert().execute(change_id=4,
timestamp=60,
changed_by="bill",
rule_id=2)
dbo.rules.history.t.insert().execute(
change_id=5,
timestamp=61,
changed_by="bill",
rule_id=2,
alias="frodo",
priority=100,
version="3.3",
buildTarget="d",
backgroundRate=100,
mapping="b",
update_type="minor",
product="a",
channel="a",
data_version=1,
)
dbo.rules.t.insert().execute(
rule_id=3,
product="a",
priority=100,
version="3.5",
buildTarget="a",
backgroundRate=100,
mapping="a",
update_type="minor",
channel="a",
data_version=2,
)
dbo.rules.history.t.insert().execute(change_id=6,
timestamp=72,
changed_by="bill",
rule_id=3)
dbo.rules.history.t.insert().execute(
change_id=7,
timestamp=73,
changed_by="bill",
rule_id=3,
product="a",
priority=50,
version="3.5",
buildTarget="a",
backgroundRate=100,
mapping="a",
update_type="minor",
channel="a",
data_version=1,
)
dbo.rules.history.t.insert().execute(
change_id=8,
timestamp=105,
changed_by="bill",
rule_id=3,
product="a",
priority=100,
version="3.5",
buildTarget="a",
backgroundRate=100,
mapping="a",
update_type="minor",
channel="a",
data_version=2,
)
dbo.rules.t.insert().execute(rule_id=4,
product="fake",
priority=80,
buildTarget="d",
backgroundRate=100,
mapping="a",
update_type="minor",
channel="a",
data_version=1)
dbo.rules.history.t.insert().execute(change_id=9,
timestamp=80,
changed_by="bill",
rule_id=4)
dbo.rules.history.t.insert().execute(
change_id=10,
timestamp=81,
changed_by="bill",
rule_id=4,
product="fake",
priority=80,
buildTarget="d",
backgroundRate=100,
mapping="a",
update_type="minor",
channel="a",
data_version=1,
)
dbo.rules.t.insert().execute(rule_id=5,
priority=80,
buildTarget="d",
version="3.3",
backgroundRate=0,
mapping="c",
update_type="minor",
product="a",
channel="a",
data_version=1)
dbo.rules.history.t.insert().execute(change_id=11,
timestamp=90,
changed_by="bill",
rule_id=5)
dbo.rules.history.t.insert().execute(
change_id=12,
timestamp=91,
changed_by="bill",
rule_id=5,
priority=80,
buildTarget="d",
version="3.3",
backgroundRate=0,
mapping="c",
update_type="minor",
product="a",
channel="a",
data_version=1,
)
dbo.rules.t.insert().execute(rule_id=6,
product="fake",
priority=40,
backgroundRate=50,
mapping="a",
update_type="minor",
channel="e",
data_version=1)
dbo.rules.history.t.insert().execute(change_id=13,
timestamp=110,
changed_by="bill",
rule_id=6)
dbo.rules.history.t.insert().execute(
change_id=14,
timestamp=111,
changed_by="bill",
rule_id=6,
product="fake",
priority=40,
backgroundRate=50,
mapping="a",
update_type="minor",
channel="e",
data_version=1,
)
dbo.rules.t.insert().execute(rule_id=7,
product="fake",
priority=30,
backgroundRate=85,
mapping="a",
update_type="minor",
channel="c",
data_version=1)
dbo.rules.history.t.insert().execute(change_id=15,
timestamp=115,
changed_by="bill",
rule_id=7)
dbo.rules.history.t.insert().execute(
change_id=16,
timestamp=116,
changed_by="bill",
rule_id=7,
product="fake",
priority=30,
backgroundRate=85,
mapping="a",
update_type="minor",
channel="c",
data_version=1,
)
dbo.rules.t.insert().execute(rule_id=8,
product="fake2",
priority=25,
backgroundRate=100,
mapping="a",
update_type="minor",
channel="c",
mig64=True,
data_version=1)
dbo.rules.history.t.insert().execute(change_id=17,
timestamp=150,
changed_by="bill",
rule_id=8)
dbo.rules.history.t.insert().execute(
change_id=18,
timestamp=151,
changed_by="bill",
rule_id=8,
product="fake2",
priority=25,
backgroundRate=100,
mapping="a",
update_type="minor",
channel="c",
mig64=True,
data_version=1,
)
dbo.rules.t.insert().execute(rule_id=9,
product="fake3",
priority=25,
backgroundRate=100,
mapping="a",
update_type="minor",
channel="c",
jaws=True,
data_version=1)
dbo.rules.history.t.insert().execute(change_id=19,
timestamp=160,
changed_by="bill",
rule_id=9)
dbo.rules.history.t.insert().execute(
change_id=20,
timestamp=161,
changed_by="bill",
rule_id=9,
product="fake3",
priority=25,
backgroundRate=100,
mapping="a",
update_type="minor",
channel="c",
jaws=True,
data_version=1,
)
self.client = app.test_client()
# Insert a release into the new tables to make sure it doesn't show up
# in the old API.
insert_release(firefox_56_0_build1, "Firefox", history=False)
yield
dbo.reset()
os.close(self.version_fd)
os.remove(self.version_file)
self.view_base.verified_userinfo = self.orig_base_verified_userinfo
dbo.releases.history = self.orig_releases_history
def setUp(self):
from auslib.web.admin.views import base as view_base
self.view_base = view_base
# Mock out verified_userinfo, because we don't want to talk to Auth0
# or need to provide real credentials in tests.
# We don't do this with "mock" because this is a base to all of other
# tests, and "mock" must be applied in the tests itself.
self.orig_base_verified_userinfo = view_base.verified_userinfo
self.mocked_user = None
def my_userinfo(*args, **kwargs):
return {"email": self.mocked_user}
view_base.verified_userinfo = my_userinfo
self.version_fd, self.version_file = mkstemp()
cache.reset()
cache.make_copies = True
app.config["SECRET_KEY"] = "abc123"
app.config["DEBUG"] = True
app.config["WTF_CSRF_ENABLED"] = False
app.config["WHITELISTED_DOMAINS"] = {"good.com": ("a", "b", "c", "d")}
app.config["VERSION_FILE"] = self.version_file
app.config["AUTH_DOMAIN"] = "balrog.test.dev"
app.config["AUTH_AUDIENCE"] = "balrog test"
app.config["M2M_ACCOUNT_MAPPING"] = {}
with open(self.version_file, "w+") as f:
f.write(
"""
{
"source":"https://github.com/mozilla/balrog",
"version":"1.0",
"commit":"abcdef123456"
}
"""
)
dbo.setDb("sqlite:///:memory:")
dbo.setDomainWhitelist({"good.com": ("a", "b", "c", "d")})
self.metadata.create_all(dbo.engine)
dbo.permissions.t.insert().execute(permission="admin", username="******", data_version=1)
dbo.permissions.t.insert().execute(permission="permission", username="******", data_version=1)
dbo.permissions.t.insert().execute(
permission="release", username="******", options=dict(products=["fake", "a", "b"], actions=["create", "modify"]), data_version=1
)
dbo.permissions.t.insert().execute(permission="release", username="******", options=dict(products=["a"], actions=["modify"]), data_version=1)
dbo.permissions.t.insert().execute(permission="rule", username="******", options=dict(products=["fake"], actions=["create"]), data_version=1)
dbo.permissions.t.insert().execute(permission="release_read_only", username="******", options=dict(actions=["set"], products=["a", "b"]), data_version=1)
dbo.permissions.t.insert().execute(permission="rule", username="******", options=dict(actions=["modify"], products=["a", "b"]), data_version=1)
dbo.permissions.t.insert().execute(permission="release", username="******", options=dict(actions=["modify"], products=["a"]), data_version=1)
dbo.permissions.t.insert().execute(permission="scheduled_change", username="******", options=dict(actions=["enact"]), data_version=1)
dbo.permissions.t.insert().execute(permission="release_locale", username="******", options=dict(actions=["modify"], products=["a"]), data_version=1)
dbo.permissions.t.insert().execute(permission="admin", username="******", options=dict(products=["a"]), data_version=1)
dbo.permissions.user_roles.t.insert().execute(username="******", role="releng", data_version=1)
dbo.permissions.user_roles.t.insert().execute(username="******", role="qa", data_version=1)
dbo.permissions.user_roles.t.insert().execute(username="******", role="relman", data_version=1)
dbo.permissions.user_roles.t.insert().execute(username="******", role="releng", data_version=1)
dbo.permissions.user_roles.t.insert().execute(username="******", role="relman", data_version=1)
dbo.productRequiredSignoffs.t.insert().execute(product="fake", channel="a", role="releng", signoffs_required=1, data_version=1)
dbo.productRequiredSignoffs.t.insert().execute(product="fake", channel="e", role="releng", signoffs_required=1, data_version=1)
dbo.productRequiredSignoffs.t.insert().execute(product="fake", channel="j", role="releng", signoffs_required=1, data_version=1)
dbo.productRequiredSignoffs.t.insert().execute(product="fake", channel="k", role="relman", signoffs_required=1, data_version=2)
dbo.productRequiredSignoffs.history.t.insert().execute(change_id=1, changed_by="bill", timestamp=10, product="fake", channel="k", role="relman")
dbo.productRequiredSignoffs.history.t.insert().execute(
change_id=2, changed_by="bill", timestamp=11, product="fake", channel="k", role="relman", signoffs_required=2, data_version=1
)
dbo.productRequiredSignoffs.history.t.insert().execute(
change_id=3, changed_by="bill", timestamp=25, product="fake", channel="k", role="relman", signoffs_required=1, data_version=2
)
dbo.permissionsRequiredSignoffs.t.insert().execute(product="fake", role="releng", signoffs_required=1, data_version=1)
dbo.permissionsRequiredSignoffs.t.insert().execute(product="bar", role="releng", signoffs_required=1, data_version=1)
dbo.permissionsRequiredSignoffs.t.insert().execute(product="blah", role="releng", signoffs_required=1, data_version=1)
dbo.permissionsRequiredSignoffs.t.insert().execute(product="doop", role="releng", signoffs_required=1, data_version=2)
dbo.permissionsRequiredSignoffs.t.insert().execute(product="superfake", role="relman", signoffs_required=1, data_version=1)
dbo.permissionsRequiredSignoffs.history.t.insert().execute(change_id=1, changed_by="bill", timestamp=10, product="doop", role="releng")
dbo.permissionsRequiredSignoffs.history.t.insert().execute(
change_id=2, changed_by="bill", timestamp=11, product="doop", role="releng", signoffs_required=2, data_version=1
)
dbo.permissionsRequiredSignoffs.history.t.insert().execute(
change_id=3, changed_by="bill", timestamp=25, product="doop", role="releng", signoffs_required=1, data_version=2
)
dbo.releases.t.insert().execute(name="a", product="a", data=createBlob(dict(name="a", hashFunction="sha512", schema_version=1)), data_version=1)
dbo.releases.t.insert().execute(name="ab", product="a", data=createBlob(dict(name="ab", hashFunction="sha512", schema_version=1)), data_version=1)
dbo.releases.history.t.insert().execute(change_id=1, timestamp=5, changed_by="bill", name="ab")
dbo.releases.history.t.insert().execute(
change_id=2,
timestamp=6,
changed_by="bill",
name="ab",
product="a",
data=createBlob(dict(name="ab", hashFunction="sha512", schema_version=1)),
data_version=1,
)
dbo.releases.t.insert().execute(name="b", product="b", data=createBlob(dict(name="b", hashFunction="sha512", schema_version=1)), data_version=1)
dbo.releases.history.t.insert().execute(change_id=5, timestamp=15, changed_by="bill", name="b")
dbo.releases.history.t.insert().execute(
change_id=6,
timestamp=16,
changed_by="bill",
name="b",
product="b",
data=createBlob(dict(name="b", hashFunction="sha512", schema_version=1)),
data_version=1,
)
dbo.releases.t.insert().execute(name="c", product="c", data=createBlob(dict(name="c", hashFunction="sha512", schema_version=1)), data_version=1)
dbo.releases.t.insert().execute(
name="d",
product="d",
data_version=1,
data=createBlob(
"""
{
"name": "d",
"schema_version": 1,
"hashFunction": "sha512",
"platforms": {
"p": {
"locales": {
"d": {
"complete": {
"filesize": 1234,
"from": "*",
"hashValue": "abc"
}
}
}
}
}
}
"""
),
)
dbo.releases.history.t.insert().execute(change_id=3, timestamp=9, changed_by="bill", name="d")
dbo.releases.history.t.insert().execute(
change_id=4,
timestamp=10,
changed_by="bill",
name="d",
product="d",
data_version=1,
data=createBlob(
"""
{
"name": "d",
"schema_version": 1,
"hashFunction": "sha512",
"platforms": {
"p": {
"locales": {
"d": {
"complete": {
"filesize": 1234,
"from": "*",
"hashValue": "abc"
}
}
}
}
}
}
"""
),
)
dbo.rules.t.insert().execute(
rule_id=1,
priority=100,
version="3.5",
buildTarget="d",
backgroundRate=100,
mapping="c",
update_type="minor",
product="a",
channel="a",
data_version=1,
)
dbo.rules.t.insert().execute(
rule_id=2,
alias="frodo",
priority=100,
version="3.3",
buildTarget="d",
backgroundRate=100,
mapping="b",
update_type="minor",
product="a",
channel="a",
data_version=1,
)
dbo.rules.t.insert().execute(
rule_id=3,
product="a",
priority=100,
version="3.5",
buildTarget="a",
backgroundRate=100,
mapping="a",
update_type="minor",
channel="a",
data_version=1,
)
dbo.rules.t.insert().execute(
rule_id=4, product="fake", priority=80, buildTarget="d", backgroundRate=100, mapping="a", update_type="minor", channel="a", data_version=1
)
dbo.rules.t.insert().execute(
rule_id=5, priority=80, buildTarget="d", version="3.3", backgroundRate=0, mapping="c", update_type="minor", product="a", channel="a", data_version=1
)
dbo.rules.t.insert().execute(rule_id=6, product="fake", priority=40, backgroundRate=50, mapping="a", update_type="minor", channel="e", data_version=1)
dbo.rules.t.insert().execute(rule_id=7, product="fake", priority=30, backgroundRate=85, mapping="a", update_type="minor", channel="c", data_version=1)
dbo.rules.t.insert().execute(
rule_id=8, product="fake2", priority=25, backgroundRate=100, mapping="a", update_type="minor", channel="c", mig64=True, data_version=1
)
dbo.rules.t.insert().execute(
rule_id=9, product="fake3", priority=25, backgroundRate=100, mapping="a", update_type="minor", channel="c", jaws=True, data_version=1
)
self.client = app.test_client()