Exemple #1
0
    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']
        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'])
        dbo.create()
        dbo.permissions.t.insert().execute(permission='admin', username='******', data_version=1)
        dbo.permissions.t.insert().execute(permission='/users/:id/permissions/:permission', username='******', data_version=1)
        dbo.permissions.t.insert().execute(permission='/releases/:name', username='******', options=json.dumps(dict(product=['fake'])), data_version=1)
        dbo.permissions.t.insert().execute(permission='/releases/:name/read_only', username='******', options=json.dumps(dict(method='PUT')), data_version=1)
        dbo.permissions.t.insert().execute(permission='/rules/:id', username='******', options=json.dumps(dict(product=['fake'])), data_version=1)
        dbo.releases.t.insert().execute(
            name='a', product='a', data=json.dumps(dict(name='a', hashFunction="sha512", schema_version=1)), data_version=1)
        dbo.releases.t.insert().execute(
            name='ab', product='a', data=json.dumps(dict(name='ab', hashFunction="sha512", schema_version=1)), data_version=1)
        dbo.releases.t.insert().execute(
            name='b', product='b', data=json.dumps(dict(name='b', hashFunction="sha512", schema_version=1)), data_version=1)
        dbo.releases.t.insert().execute(
            name='c', product='c', data=json.dumps(dict(name='c', hashFunction="sha512", schema_version=1)), data_version=1)
        dbo.releases.t.insert().execute(name='d', product='d', data_version=1, data="""
{
    "name": "d",
    "schema_version": 1,
    "hashFunction": "sha512",
    "platforms": {
        "p": {
            "locales": {
                "d": {
                    "complete": {
                        "filesize": 1234,
                        "from": "*",
                        "hashValue": "abc"
                    }
                }
            }
        }
    }
}
""")
        dbo.rules.t.insert().execute(id=1, priority=100, version='3.5', buildTarget='d', backgroundRate=100, mapping='c', update_type='minor', data_version=1)
        dbo.rules.t.insert().execute(id=2, alias="frodo", priority=100, version='3.3', buildTarget='d', backgroundRate=100, mapping='b', update_type='minor',
                                     data_version=1)
        dbo.rules.t.insert().execute(id=3, priority=100, version='3.5', buildTarget='a', backgroundRate=100, mapping='a', update_type='minor', data_version=1)
        dbo.rules.t.insert().execute(id=4, product='fake', priority=80, buildTarget='d', backgroundRate=100, mapping='a', update_type='minor', data_version=1)
        dbo.rules.t.insert().execute(id=5, priority=80, buildTarget='d', version='3.3', backgroundRate=0, mapping='c', update_type='minor', data_version=1)
        self.client = app.test_client()
Exemple #2
0
    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()
Exemple #3
0
    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=json.dumps(dict(products=["fake", "b"], actions=["create", "modify"])),
            data_version=1,
        )
        dbo.permissions.t.insert().execute(
            permission="release_read_only", username="******", options=json.dumps(dict(actions=["set"])), data_version=1
        )
        dbo.permissions.t.insert().execute(
            permission="rule",
            username="******",
            options=json.dumps(dict(actions=["modify"], products=["fake"])),
            data_version=1,
        )
        dbo.permissions.t.insert().execute(
            permission="build",
            username="******",
            options=json.dumps(dict(actions=["modify"], products=["a"])),
            data_version=1,
        )
        dbo.permissions.t.insert().execute(
            permission="scheduled_change", username="******", options=json.dumps(dict(actions=["enact"])), data_version=1
        )
        dbo.permissions.t.insert().execute(
            permission="release_locale",
            username="******",
            options=json.dumps(dict(actions=["modify"], products=["a"])),
            data_version=1,
        )
        dbo.permissions.t.insert().execute(
            permission="admin", username="******", options=json.dumps(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="relman", data_version=1)
        dbo.releases.t.insert().execute(
            name="a",
            product="a",
            data=json.dumps(dict(name="a", hashFunction="sha512", schema_version=1)),
            data_version=1,
        )
        dbo.releases.t.insert().execute(
            name="ab",
            product="a",
            data=json.dumps(dict(name="ab", hashFunction="sha512", schema_version=1)),
            data_version=1,
        )
        dbo.releases.t.insert().execute(
            name="b",
            product="b",
            data=json.dumps(dict(name="b", hashFunction="sha512", schema_version=1)),
            data_version=1,
        )
        dbo.releases.t.insert().execute(
            name="c",
            product="c",
            data=json.dumps(dict(name="c", hashFunction="sha512", schema_version=1)),
            data_version=1,
        )
        dbo.releases.t.insert().execute(
            name="d",
            product="d",
            data_version=1,
            data="""
{
    "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",
            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",
            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",
            data_version=1,
        )
        dbo.rules.t.insert().execute(
            rule_id=4,
            product="fake",
            priority=80,
            buildTarget="d",
            backgroundRate=100,
            mapping="a",
            update_type="minor",
            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",
            data_version=1,
        )
        self.client = app.test_client()
Exemple #4
0
    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()
Exemple #5
0
    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()
Exemple #6
0
 def setUp(self):
     cache.reset()
     cache.make_cache("blob_schema", 50, 10000)
Exemple #7
0
 def tearDown(self):
     cache.reset()
Exemple #8
0
 def setUp(self):
     cache.reset()
     cache.make_cache("blob_schema", 50, 10000)
Exemple #9
0
 def tearDown(self):
     cache.reset()
Exemple #10
0
    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
Exemple #11
0
    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='/users/:id/permissions/:permission',
            username='******',
            data_version=1)
        dbo.permissions.t.insert().execute(permission='/releases/:name',
                                           username='******',
                                           options=json.dumps(
                                               dict(product=['fake'])),
                                           data_version=1)
        dbo.permissions.t.insert().execute(
            permission='/releases/:name/read_only',
            username='******',
            options=json.dumps(dict(method='PUT')),
            data_version=1)
        dbo.permissions.t.insert().execute(permission='/rules/:id',
                                           username='******',
                                           options=json.dumps(
                                               dict(product=['fake'])),
                                           data_version=1)
        dbo.releases.t.insert().execute(name='a',
                                        product='a',
                                        data=json.dumps(
                                            dict(name='a',
                                                 hashFunction="sha512",
                                                 schema_version=1)),
                                        data_version=1)
        dbo.releases.t.insert().execute(name='ab',
                                        product='a',
                                        data=json.dumps(
                                            dict(name='ab',
                                                 hashFunction="sha512",
                                                 schema_version=1)),
                                        data_version=1)
        dbo.releases.t.insert().execute(name='b',
                                        product='b',
                                        data=json.dumps(
                                            dict(name='b',
                                                 hashFunction="sha512",
                                                 schema_version=1)),
                                        data_version=1)
        dbo.releases.t.insert().execute(name='c',
                                        product='c',
                                        data=json.dumps(
                                            dict(name='c',
                                                 hashFunction="sha512",
                                                 schema_version=1)),
                                        data_version=1)
        dbo.releases.t.insert().execute(name='d',
                                        product='d',
                                        data_version=1,
                                        data="""
{
    "name": "d",
    "schema_version": 1,
    "hashFunction": "sha512",
    "platforms": {
        "p": {
            "locales": {
                "d": {
                    "complete": {
                        "filesize": 1234,
                        "from": "*",
                        "hashValue": "abc"
                    }
                }
            }
        }
    }
}
""")
        dbo.rules.t.insert().execute(id=1,
                                     priority=100,
                                     version='3.5',
                                     buildTarget='d',
                                     backgroundRate=100,
                                     mapping='c',
                                     update_type='minor',
                                     data_version=1)
        dbo.rules.t.insert().execute(id=2,
                                     alias="frodo",
                                     priority=100,
                                     version='3.3',
                                     buildTarget='d',
                                     backgroundRate=100,
                                     mapping='b',
                                     update_type='minor',
                                     data_version=1)
        dbo.rules.t.insert().execute(id=3,
                                     priority=100,
                                     version='3.5',
                                     buildTarget='a',
                                     backgroundRate=100,
                                     mapping='a',
                                     update_type='minor',
                                     data_version=1)
        dbo.rules.t.insert().execute(id=4,
                                     product='fake',
                                     priority=80,
                                     buildTarget='d',
                                     backgroundRate=100,
                                     mapping='a',
                                     update_type='minor',
                                     data_version=1)
        dbo.rules.t.insert().execute(id=5,
                                     priority=80,
                                     buildTarget='d',
                                     version='3.3',
                                     backgroundRate=0,
                                     mapping='c',
                                     update_type='minor',
                                     data_version=1)
        self.client = app.test_client()
Exemple #12
0
    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()