andrej@1808: #!/usr/bin/env python andrej@1808: # -*- coding: utf-8 -*- andrej@1808: andrej@1808: # This file is part of Beremiz, a Integrated Development Environment for andrej@1808: # programming IEC 61131-3 automates supporting plcopen standard and CanFestival. andrej@1808: # andrej@1808: # Copyright (C) 2017: Andrey Skvortsov andrej@1808: # andrej@1808: # See COPYING file for copyrights details. andrej@1808: # andrej@1808: # This program is free software; you can redistribute it and/or andrej@1808: # modify it under the terms of the GNU General Public License andrej@1808: # as published by the Free Software Foundation; either version 2 andrej@1808: # of the License, or (at your option) any later version. andrej@1808: # andrej@1808: # This program is distributed in the hope that it will be useful, andrej@1808: # but WITHOUT ANY WARRANTY; without even the implied warranty of andrej@1808: # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the andrej@1808: # GNU General Public License for more details. andrej@1808: # andrej@1808: # You should have received a copy of the GNU General Public License andrej@1808: # along with this program; if not, write to the Free Software andrej@1808: # Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA. andrej@1808: andrej@1826: andrej@1826: from __future__ import print_function andrej@1796: import os andrej@1796: import sys andrej@1796: import unittest andrej@1832: import time andrej@1832: import traceback andrej@1832: andrej@1809: import pytest andrej@1796: import wx andrej@1809: import ddt andrej@1796: andrej@1808: import conftest andrej@1808: import Beremiz andrej@1819: import PLCOpenEditor andrej@1796: andrej@1796: andrej@1796: class UserApplicationTest(unittest.TestCase): andrej@1796: def InstallExceptionHandler(self): andrej@1796: def handle_exception(e_type, e_value, e_traceback): andrej@1796: # traceback.print_exception(e_type, e_value, e_traceback) andrej@1796: self.exc_info = [e_type, e_value, e_traceback] andrej@1796: self.exc_info = None andrej@1796: self.old_excepthook = sys.excepthook andrej@1796: sys.excepthook = handle_exception andrej@1796: andrej@1796: def StartApp(self): andrej@1796: self.app = None andrej@1796: andrej@1796: def FinishApp(self): andrej@1796: wx.CallAfter(self.app.frame.Close) andrej@1796: self.app.MainLoop() andrej@1796: self.app = None andrej@1796: andrej@1809: def setUp(self): andrej@1809: self.app = None andrej@1809: andrej@1796: def tearDown(self): andrej@1796: if self.app is not None and self.app.frame is not None: andrej@1796: self.FinishApp() andrej@1796: andrej@1796: def RunUIActions(self, actions): andrej@1796: for act in actions: andrej@1796: wx.CallAfter(*act) andrej@1796: self.ProcessEvents() andrej@1796: andrej@1796: def CheckForErrors(self): andrej@1796: if self.exc_info is not None: andrej@1796: # reraise catched previously exception andrej@1796: raise self.exc_info[0], self.exc_info[1], self.exc_info[2] andrej@1796: andrej@1796: def ProcessEvents(self): andrej@1847: for dummy in range(0, 30): andrej@1796: self.CheckForErrors() andrej@1796: wx.Yield() andrej@1796: time.sleep(0.01) andrej@1796: andrej@1796: andrej@1809: @ddt.ddt andrej@1796: class BeremizApplicationTest(UserApplicationTest): andrej@1796: """Test Beremiz as whole application""" andrej@1816: andrej@1796: def StartApp(self): andrej@1796: self.app = Beremiz.BeremizIDELauncher() andrej@1796: # disable default exception handler in Beremiz andrej@1796: self.app.InstallExceptionHandler = lambda: None andrej@1796: self.InstallExceptionHandler() andrej@1796: self.app.PreStart() andrej@1818: self.app.frame.Show() andrej@1818: self.ProcessEvents() andrej@1818: self.app.frame.ShowFullScreen(True) andrej@1818: self.ProcessEvents() andrej@1796: andrej@1796: def FinishApp(self): andrej@1796: wx.CallAfter(self.app.frame.Close) andrej@1796: self.app.MainLoop() andrej@1796: time.sleep(1) andrej@1796: self.app = None andrej@1796: andrej@1816: def GetSkippedProjectTreeItems(self): andrej@1816: """ andrej@1816: Returns the list of skipped items in the project tree. andrej@1816: andrej@1816: Beremiz test don't need to skip any elemnts in the project tree. andrej@1816: """ andrej@1816: return [] andrej@1816: andrej@1796: def OpenAllProjectElements(self): andrej@1808: """Open editor for every object in the project tree""" andrej@1796: self.app.frame.ProjectTree.ExpandAll() andrej@1796: self.ProcessEvents() andrej@1796: item = self.app.frame.ProjectTree.GetRootItem() andrej@1819: skip = self.GetSkippedProjectTreeItems() andrej@1819: tree_id = self.app.frame.ProjectTree.GetId() andrej@1796: while item is not None: andrej@1796: self.app.frame.ProjectTree.SelectItem(item, True) andrej@1796: self.ProcessEvents() andrej@1819: if item not in skip: andrej@1819: event = wx.lib.agw.customtreectrl.TreeEvent( andrej@1819: wx.lib.agw.customtreectrl.wxEVT_TREE_ITEM_ACTIVATED, andrej@1819: tree_id, item) andrej@1819: self.app.frame.OnProjectTreeItemActivated(event) andrej@1796: self.ProcessEvents() andrej@1796: item = self.app.frame.ProjectTree.GetNextVisible(item) andrej@1796: andrej@1816: def CheckTestProject(self, project): andrej@1796: sys.argv = ["", project] andrej@1796: self.StartApp() andrej@1796: self.OpenAllProjectElements() andrej@1816: user_actions = self.GetUserActions() andrej@1816: self.RunUIActions(user_actions) andrej@1816: self.FinishApp() andrej@1796: andrej@1816: def GetProjectPath(self, project): andrej@1816: return os.path.abspath(os.path.join(os.path.dirname(__file__), "..", project)) andrej@1816: andrej@1816: def GetUserActions(self): andrej@1816: """ andrej@1816: Returns list of user actions that will be executed andrej@1816: on every test project by testCheckProject test. andrej@1816: """ andrej@1796: user_actions = [ andrej@1796: [self.app.frame.SwitchFullScrMode, None], andrej@1796: [self.app.frame.SwitchFullScrMode, None], andrej@1796: [self.app.frame.CTR._Clean], andrej@1796: [self.app.frame.CTR._Build], andrej@1796: [self.app.frame.CTR._Connect], andrej@1796: [self.app.frame.CTR._Transfer], andrej@1796: [self.app.frame.CTR._Run], andrej@1796: [self.app.frame.CTR._Stop], andrej@1796: [self.app.frame.CTR._Disconnect], andrej@1822: [self.app.frame.CTR._Clean], andrej@1796: ] andrej@1816: return user_actions andrej@1796: andrej@1796: def testStartUp(self): andrej@1796: """Checks whether the app starts and finishes correctly""" andrej@1817: sys.argv = [""] andrej@1796: self.StartApp() andrej@1796: self.FinishApp() andrej@1796: andrej@1809: @ddt.data( andrej@1809: "first_steps", andrej@1809: "logging", andrej@1809: "svgui", andrej@1809: "traffic_lights", andrej@1809: "wxGlade", andrej@1809: "python", andrej@1809: "wiimote", andrej@1809: "wxHMI", andrej@1809: ) andrej@1809: @pytest.mark.timeout(30) andrej@1809: def testCheckProject(self, name): andrej@1816: """ andrej@1816: Checks that test PLC project can be open, andrej@1816: compiled and run on SoftPLC. andrej@1816: """ andrej@1816: project = self.GetProjectPath(name) andrej@1826: print("Testing example " + name) andrej@1816: self.CheckTestProject(project) andrej@1796: andrej@1796: andrej@1819: class PLCOpenEditorApplicationTest(BeremizApplicationTest): andrej@1819: """Test PLCOpenEditor as whole application""" andrej@1819: andrej@1819: def StartApp(self): andrej@1819: self.app = PLCOpenEditor.PLCOpenEditorApp() andrej@1819: # disable default exception handler in application andrej@1819: self.app.InstallExceptionHandler = lambda: None andrej@1819: self.InstallExceptionHandler() andrej@1819: self.app.Show() andrej@1819: self.ProcessEvents() andrej@1819: self.app.frame.ShowFullScreen(True) andrej@1819: self.ProcessEvents() andrej@1819: andrej@1819: def FinishApp(self): andrej@1819: wx.CallAfter(self.app.frame.Close) andrej@1819: self.app.MainLoop() andrej@1819: time.sleep(1) andrej@1819: self.app = None andrej@1819: andrej@1819: def GetSkippedProjectTreeItems(self): andrej@1819: """ andrej@1819: Returns the list of skipped items in the project tree. andrej@1819: andrej@1819: Root item opens dialog window for project settings. andrej@1819: To avoid code that handles closing dialog windows just skip this item. andrej@1819: """ andrej@1819: return [self.app.frame.ProjectTree.GetRootItem()] andrej@1819: andrej@1819: def GetUserActions(self): andrej@1819: return [] andrej@1819: andrej@1819: def GetProjectPath(self, project): andrej@1819: """Open PLC program in every Beremiz test project""" andrej@1819: project_dir = BeremizApplicationTest.GetProjectPath(self, project) andrej@1819: return os.path.join(project_dir, "plc.xml") andrej@1819: andrej@1819: andrej@1796: if __name__ == '__main__': andrej@1796: unittest.main()