# -*- coding: utf-8 -*- # This Source Code Form is subject to the terms of the Mozilla Public # License, v. 2.0. If a copy of the MPL was not distributed with this file, # You can obtain one at http://mozilla.org/MPL/2.0/. from logging import INFO, basicConfig, getLogger from bugbug import model from bugbug.models import MODELS, get_model_class basicConfig(level=INFO) logger = getLogger(__name__) def test_import_all_models(): """Try loading all defined models to ensure that their full qualified names are still good """ for model_name in MODELS: logger.info("Try loading model %s", model_name) get_model_class(model_name) def test_component_is_bugmodel(): model_class = get_model_class("component") assert issubclass(model_class, model.BugModel) model_class = get_model_class("regression") assert issubclass(model_class, model.BugModel) def test_backout_is_commitmodel(): model_class = get_model_class("backout") assert issubclass(model_class, model.CommitModel)