contentdb/app/tests/unit/logic/test_package_approval.py

186 lines
7.5 KiB
Python
Raw Normal View History

# ContentDB
# Copyright (C) rubenwardy
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU Affero General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU Affero General Public License for more details.
#
# You should have received a copy of the GNU Affero General Public License
# along with this program. If not, see <https://www.gnu.org/licenses/>.
from unittest.mock import MagicMock, patch
from app.logic import package_approval
from app.models import Package, PackageType
class MockPackageHelper:
package: Package
def __init__(self, type_: PackageType = PackageType.MOD):
self.package = MagicMock()
self.package.type = type_
self.package.name = "foobar"
self.package.normalised_name.name = "foobar"
self.package.author.username = "username"
self.package.author.id = 3
self.package.releases.filter.return_value.count.return_value = 0
self.package.releases.count.return_value = 0
self.package.get_url.return_value = "hi"
self.package.screenshots.count.return_value = 0
def add_release(self):
self.package.releases.filter.return_value.count.return_value = 1
self.package.releases.count.return_value = 1
def add_pending_release(self):
self.package.releases.filter.return_value.count.return_value = 0
self.package.releases.count.return_value = 1
def add_screenshot(self):
self.package.screenshots.count.return_value = 1
def add_missing_hard_deps(self):
mod_name = MagicMock()
mod_name.name = "missing"
self.package.get_missing_hard_dependencies_query.return_value.all.return_value = [mod_name]
def set_license(self, code_license: str, media_license: str):
self.package.license.name = code_license
self.package.media_license.name = media_license
def set_no_game_support(self):
assert self.package.type != PackageType.GAME
self.package.supports_all_games = False
self.package.supported_games.count.return_value = 0
def test_requires_release():
mock_package = MockPackageHelper()
notes = package_approval.validate_package_for_approval(mock_package.package)
assert len(notes) == 1
assert notes[0].message == "You need to create a release before this package can be approved."
mock_package.add_pending_release()
notes = package_approval.validate_package_for_approval(mock_package.package)
assert len(notes) == 1
assert notes[0].message == "Release is still importing, or has an error."
@patch("app.logic.package_approval.is_package_name_taken", MagicMock(return_value=False))
@patch("app.logic.package_approval.get_conflicting_mod_names", MagicMock(return_value=set()))
@patch("app.logic.package_approval.count_packages_with_forum_topic", MagicMock(return_value=1))
@patch("app.logic.package_approval.get_forum_topic")
def test_missing_hard_deps(get_forum_topic):
mock_package = MockPackageHelper(PackageType.MOD)
mock_package.add_release()
mock_package.add_missing_hard_deps()
topic = MagicMock()
topic.author = mock_package.package.author
get_forum_topic.return_value = topic
notes = package_approval.validate_package_for_approval(mock_package.package)
assert len(notes) == 1
assert notes[0].message == "The following hard dependencies need to be added to ContentDB first: missing"
@patch("app.logic.package_approval.is_package_name_taken", MagicMock(return_value=True))
@patch("app.logic.package_approval.get_conflicting_mod_names", MagicMock(return_value={"one", "two"}))
@patch("app.logic.package_approval.count_packages_with_forum_topic", MagicMock(return_value=2))
@patch("app.logic.package_approval.get_forum_topic", MagicMock(return_value=None))
def test_requires_multiple_issues():
mock_package = MockPackageHelper()
mock_package.add_release()
mock_package.set_license("Other", "Other")
mock_package.set_no_game_support()
notes = package_approval.validate_package_for_approval(mock_package.package)
assert len(notes) == 5
assert notes[0].message == "What games does your package support? Please specify on the supported games page"
assert notes[1].message == "Please wait for the license to be added to CDB."
assert notes[2].message == "Please make sure that this package has the right to the names one, two"
assert notes[3].message == "<b>Error: Another package already uses this forum topic!</b>"
assert notes[4].message == "Warning: Forum topic not found. The topic may have been created since the last forum crawl."
@patch("app.logic.package_approval.is_package_name_taken", MagicMock(return_value=False))
@patch("app.logic.package_approval.get_conflicting_mod_names", MagicMock(return_value=set()))
@patch("app.logic.package_approval.count_packages_with_forum_topic", MagicMock(return_value=1))
@patch("app.logic.package_approval.get_forum_topic")
def test_forum_topic_author_mismatch(get_forum_topic):
mock_package = MockPackageHelper()
mock_package.add_release()
topic = MagicMock()
get_forum_topic.return_value = topic
notes = package_approval.validate_package_for_approval(mock_package.package)
assert len(notes) == 1
assert notes[0].message == "<b>Error: Forum topic author doesn't match package author.</b>"
@patch("app.logic.package_approval.is_package_name_taken", MagicMock(return_value=False))
@patch("app.logic.package_approval.get_conflicting_mod_names", MagicMock(return_value=set()))
@patch("app.logic.package_approval.count_packages_with_forum_topic", MagicMock(return_value=1))
@patch("app.logic.package_approval.get_forum_topic")
def test_passes(get_forum_topic):
mock_package = MockPackageHelper()
mock_package.add_release()
topic = MagicMock()
topic.author = mock_package.package.author
get_forum_topic.return_value = topic
notes = package_approval.validate_package_for_approval(mock_package.package)
assert len(notes) == 0
@patch("app.logic.package_approval.is_package_name_taken", MagicMock(return_value=True))
@patch("app.logic.package_approval.get_conflicting_mod_names", MagicMock(return_value=set()))
@patch("app.logic.package_approval.count_packages_with_forum_topic", MagicMock(return_value=1))
@patch("app.logic.package_approval.get_forum_topic")
def test_games_txp_must_have_unique_name(get_forum_topic):
mock_package = MockPackageHelper(PackageType.GAME)
mock_package.add_release()
mock_package.add_screenshot()
topic = MagicMock()
topic.author = mock_package.package.author
get_forum_topic.return_value = topic
notes = package_approval.validate_package_for_approval(mock_package.package)
assert len(notes) == 1
assert notes[0].message == "A package already exists with this name. Please see Policy and Guidance 3"
@patch("app.logic.package_approval.is_package_name_taken", MagicMock(return_value=False))
@patch("app.logic.package_approval.get_conflicting_mod_names", MagicMock(return_value=set()))
@patch("app.logic.package_approval.count_packages_with_forum_topic", MagicMock(return_value=1))
@patch("app.logic.package_approval.get_forum_topic")
def test_games_txp_require_screenshots(get_forum_topic):
mock_package = MockPackageHelper(PackageType.GAME)
mock_package.add_release()
topic = MagicMock()
topic.author = mock_package.package.author
get_forum_topic.return_value = topic
notes = package_approval.validate_package_for_approval(mock_package.package)
assert len(notes) == 1
assert notes[0].message == "You need to add at least one screenshot."
mock_package.add_screenshot()
notes = package_approval.validate_package_for_approval(mock_package.package)
assert len(notes) == 0