You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
235 lines
10 KiB
Python
235 lines
10 KiB
Python
#
|
|
# All or portions of this file Copyright (c) Amazon.com, Inc. or its affiliates or
|
|
# its licensors.
|
|
#
|
|
# For complete copyright and license terms please see the LICENSE at the root of this
|
|
# distribution (the "License"). All use of this software is governed by the License,
|
|
# or, if provided, by the license below or the license accompanying this file. Do not
|
|
# remove or modify any license notices. This file is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
#
|
|
|
|
import pytest
|
|
import os
|
|
import sys
|
|
import tempfile
|
|
import logging
|
|
from unittest.mock import MagicMock
|
|
|
|
logger = logging.getLogger()
|
|
|
|
# Code lives one folder above
|
|
projects_path = os.path.realpath(os.path.join(os.path.dirname(__file__), '..'))
|
|
sys.path.append(projects_path)
|
|
|
|
from pyside import add_pyside_environment, is_configuration_valid
|
|
from ly_test_tools import WINDOWS
|
|
|
|
project_marker_file = "project.json"
|
|
|
|
class ProjectHelper:
|
|
def __init__(self):
|
|
self._temp_directory = tempfile.TemporaryDirectory()
|
|
self.temp_project_root = self._temp_directory.name
|
|
self.temp_file_dir = os.path.join(self.temp_project_root, "o3de")
|
|
|
|
self.application = None
|
|
self.dialog = None
|
|
|
|
if not os.path.exists(self.temp_file_dir):
|
|
os.makedirs(self.temp_file_dir)
|
|
|
|
def create_empty_projects(self):
|
|
self.project_1_dir = os.path.join(self.temp_project_root, "Project1")
|
|
if not os.path.exists(self.project_1_dir):
|
|
os.makedirs(self.project_1_dir)
|
|
with open(os.path.join(self.project_1_dir,project_marker_file), 'w') as marker_file:
|
|
marker_file.write("{}")
|
|
self.project_2_dir = os.path.join(self.temp_project_root, "Project2")
|
|
if not os.path.exists(self.project_2_dir):
|
|
os.makedirs(self.project_2_dir)
|
|
with open(os.path.join(self.project_2_dir, project_marker_file), 'w') as marker_file:
|
|
marker_file.write("{}")
|
|
self.project_3_dir = os.path.join(self.temp_project_root, "Project3")
|
|
if not os.path.exists(self.project_3_dir):
|
|
os.makedirs(self.project_3_dir)
|
|
with open(os.path.join(self.project_3_dir, project_marker_file), 'w') as marker_file:
|
|
marker_file.write("{}")
|
|
self.invalid_project_dir = os.path.join(self.temp_project_root, "InvalidProject")
|
|
|
|
|
|
def setup_dialog_test(self, workspace):
|
|
add_pyside_environment(workspace.paths.build_directory())
|
|
|
|
if not is_configuration_valid(workspace):
|
|
# This is essentially skipif debug. Our debug tests use our profile version of python, but that means we'd
|
|
# need to use the profile version of PySide which works with the profile QT libs which aren't in the debug
|
|
# folder we've built.
|
|
return None
|
|
from PySide2.QtWidgets import QApplication, QMessageBox
|
|
|
|
if QApplication.instance():
|
|
self.application = QApplication.instance()
|
|
else:
|
|
self.application = QApplication(sys.argv)
|
|
assert self.application
|
|
|
|
from projects import ProjectDialog
|
|
|
|
try:
|
|
self.dialog = ProjectDialog(settings_folder=self.temp_file_dir)
|
|
return self.dialog
|
|
except Exception as e:
|
|
logger.error(f'Failed to create ProjectDialog with error {e}')
|
|
return None
|
|
|
|
def create_project_from_template(self, project_name) -> bool:
|
|
"""
|
|
Uses the dialog to create a temporary project based on the first template found
|
|
:param project_name: Name of project to create. Will be created under temp_project_root
|
|
:return: True for Success, False for failure
|
|
"""
|
|
from PySide2.QtWidgets import QWidget, QFileDialog
|
|
from projects import ProjectDialog
|
|
|
|
QWidget.exec = MagicMock()
|
|
self.dialog.create_project_handler()
|
|
QWidget.exec.assert_called_once()
|
|
|
|
assert len(self.dialog.project_templates), 'Failed to find any project templates'
|
|
ProjectDialog.get_selected_project_template = MagicMock(return_value=self.dialog.project_templates[0])
|
|
|
|
QFileDialog.exec = MagicMock()
|
|
create_project_dir = os.path.join(self.temp_project_root, project_name)
|
|
QFileDialog.selectedFiles = MagicMock(return_value=[create_project_dir])
|
|
self.dialog.create_project_accepted_handler()
|
|
assert os.path.isdir(create_project_dir), f"Expected project folder not found at {create_project_dir}"
|
|
assert QWidget.exec.call_count == 2, "Message box confirming project creation failed to show"
|
|
|
|
|
|
@pytest.fixture
|
|
def project_helper():
|
|
return ProjectHelper()
|
|
|
|
|
|
@pytest.mark.skipif(not WINDOWS, reason="PySide2 only works on windows currently")
|
|
@pytest.mark.parametrize('project', ['']) # Workspace wants a project, but this test is not project dependent
|
|
def test_logger_handler(workspace, project_helper):
|
|
my_dialog = project_helper.setup_dialog_test(workspace)
|
|
if not my_dialog:
|
|
return
|
|
|
|
from PySide2.QtWidgets import QMessageBox
|
|
QMessageBox.warning = MagicMock()
|
|
logger.error(f'Testing logger')
|
|
QMessageBox.warning.assert_called_once()
|
|
|
|
|
|
@pytest.mark.skipif(not WINDOWS, reason="PySide2 only works on windows currently")
|
|
@pytest.mark.parametrize('project', ['']) # Workspace wants a project, but this test is not project dependent
|
|
def test_mru_list(workspace, project_helper):
|
|
my_dialog = project_helper.setup_dialog_test(workspace)
|
|
if not my_dialog:
|
|
return
|
|
|
|
project_helper.create_empty_projects()
|
|
|
|
from PySide2.QtWidgets import QMessageBox
|
|
|
|
mru_list = my_dialog.get_mru_list()
|
|
assert len(mru_list) == 0, f'MRU list unexpectedly had entries: {mru_list}'
|
|
|
|
QMessageBox.warning = MagicMock()
|
|
my_dialog.add_new_project('TestProjectInvalid')
|
|
mru_list = my_dialog.get_mru_list()
|
|
assert len(mru_list) == 0, f'MRU list unexpectedly added an invalid project : {mru_list}'
|
|
QMessageBox.warning.assert_called_once()
|
|
|
|
my_dialog.add_new_project(project_helper.project_1_dir)
|
|
mru_list = my_dialog.get_mru_list()
|
|
assert len(mru_list) == 1, f'MRU list failed to add project at {project_helper.project_1_dir}'
|
|
|
|
my_dialog.add_new_project(project_helper.project_1_dir)
|
|
mru_list = my_dialog.get_mru_list()
|
|
assert len(mru_list) == 1, f'MRU list added project at {project_helper.project_1_dir} a second time : {mru_list}'
|
|
|
|
my_dialog.update_mru_list(project_helper.project_1_dir)
|
|
mru_list = my_dialog.get_mru_list()
|
|
assert len(mru_list) == 1, f'MRU list added project at {project_helper.project_1_dir} a second time : {mru_list}'
|
|
|
|
my_dialog.add_new_project(project_helper.project_2_dir)
|
|
mru_list = my_dialog.get_mru_list()
|
|
assert len(mru_list) == 2, f'MRU list failed to add project at {project_helper.project_2_dir}'
|
|
|
|
assert mru_list[0] == project_helper.project_2_dir, f"{project_helper.project_2_dir} wasn't first item"
|
|
assert mru_list[1] == project_helper.project_1_dir, f"{project_helper.project_1_dir} wasn't second item"
|
|
|
|
my_dialog.update_mru_list(project_helper.project_1_dir)
|
|
mru_list = my_dialog.get_mru_list()
|
|
assert len(mru_list) == 2, f'MRU list added wrong items {mru_list}'
|
|
assert mru_list[0] == project_helper.project_1_dir, f"{project_helper.project_1_dir} wasn't first item"
|
|
assert mru_list[1] == project_helper.project_2_dir, f"{project_helper.project_2_dir} wasn't second item"
|
|
|
|
my_dialog.add_new_project(project_helper.invalid_project_dir)
|
|
mru_list = my_dialog.get_mru_list()
|
|
assert len(mru_list) == 2, f'MRU list added invalid item {mru_list}'
|
|
assert mru_list[0] == project_helper.project_1_dir, f"{project_helper.project_1_dir} wasn't first item"
|
|
assert mru_list[1] == project_helper.project_2_dir, f"{project_helper.project_2_dir} wasn't second item"
|
|
|
|
my_dialog.add_new_project(project_helper.project_3_dir)
|
|
mru_list = my_dialog.get_mru_list()
|
|
assert len(mru_list) == 3, f'MRU list failed to add {project_helper.project_3_dir} : {mru_list}'
|
|
assert mru_list[0] == project_helper.project_3_dir, f"{project_helper.project_3_dir} wasn't first item"
|
|
assert mru_list[1] == project_helper.project_1_dir, f"{project_helper.project_1_dir} wasn't second item"
|
|
assert mru_list[2] == project_helper.project_2_dir, f"{project_helper.project_2_dir} wasn't third item"
|
|
|
|
|
|
@pytest.mark.skipif(not WINDOWS, reason="PySide2 only works on windows currently")
|
|
@pytest.mark.parametrize('project', ['']) # Workspace wants a project, but this test is not project dependent
|
|
def test_create_project(workspace, project_helper):
|
|
my_dialog = project_helper.setup_dialog_test(workspace)
|
|
if not my_dialog:
|
|
return
|
|
|
|
project_helper.create_project_from_template("TestCreateProject")
|
|
|
|
|
|
@pytest.mark.skipif(not WINDOWS, reason="PySide2 only works on windows currently")
|
|
@pytest.mark.parametrize('project', ['']) # Workspace wants a project, but this test is not project dependent
|
|
def test_add_remove_gems(workspace, project_helper):
|
|
my_dialog = project_helper.setup_dialog_test(workspace)
|
|
if not my_dialog:
|
|
return
|
|
|
|
my_project_name = "TestAddRemoveGems"
|
|
|
|
project_helper.create_project_from_template(my_project_name)
|
|
my_project_path = os.path.join(project_helper.temp_project_root, my_project_name)
|
|
|
|
from PySide2.QtWidgets import QWidget, QFileDialog
|
|
from projects import ProjectDialog
|
|
|
|
assert my_dialog.path_for_selection() == my_project_path, "Gems project not selected"
|
|
QWidget.exec = MagicMock()
|
|
my_dialog.manage_gems_handler()
|
|
assert my_dialog.manage_gems_dialog, "No gem management dialog created"
|
|
QWidget.exec.assert_called_once()
|
|
|
|
assert len(my_dialog.gems_list), 'Failed to find any gems'
|
|
my_test_gem = my_dialog.gems_list[0]
|
|
my_test_gem_name = my_test_gem.get("Name")
|
|
my_test_gem_path = my_test_gem.get("Path")
|
|
my_test_gem_selection = (my_test_gem_name, my_test_gem_path)
|
|
ProjectDialog.get_selected_add_gems = MagicMock(return_value=[my_test_gem_selection])
|
|
|
|
|
|
assert my_test_gem_name, "No Name set in test gem"
|
|
assert my_test_gem_name not in my_dialog.project_gem_list, f'Gem {my_test_gem_name} already in project gem list'
|
|
my_dialog.add_gems_handler()
|
|
assert my_test_gem_name in my_dialog.project_gem_list, f'Gem {my_test_gem_name} failed to add to gem list'
|
|
ProjectDialog.get_selected_project_gems = MagicMock(return_value=[my_test_gem_name])
|
|
my_dialog.remove_gems_handler()
|
|
assert my_test_gem_name not in my_dialog.project_gem_list, f'Gem {my_test_gem_name} still in project gem list'
|
|
|
|
|