25

import os

    26

import sys

    27

import unittest

    28

import wx

    29

import time

    30

import traceback

    31

    34

    35

    36

class UserApplicationTest(unittest.TestCase):

    37

    def InstallExceptionHandler(self):

    38

        def handle_exception(e_type, e_value, e_traceback):

    39

            # traceback.print_exception(e_type, e_value, e_traceback)

    40

            self.exc_info = [e_type, e_value, e_traceback]

    41

        self.exc_info = None

    42

        self.old_excepthook = sys.excepthook

    43

        sys.excepthook = handle_exception

    44

    45

    def StartApp(self):

    46

        self.app = None

    47

    48

    def FinishApp(self):

    49

        wx.CallAfter(self.app.frame.Close)

    50

        self.app.MainLoop()

    51

        self.app = None

    52

    53

    def tearDown(self):

    54

        if self.app is not None and self.app.frame is not None:

    55

            self.FinishApp()

    56

    57

    def RunUIActions(self, actions):

    58

        for act in actions:

    59

            wx.CallAfter(*act)

    60

            self.ProcessEvents()

    61

    62

    def CheckForErrors(self):

    63

        if self.exc_info is not None:

    64

            # reraise catched previously exception

    65

            raise self.exc_info[0], self.exc_info[1], self.exc_info[2]

    66

    67

    def ProcessEvents(self):

    68

        for i in range(0, 30):

    69

            self.CheckForErrors()

    70

            wx.Yield()

    71

            time.sleep(0.01)

    72

    73

    74

class BeremizApplicationTest(UserApplicationTest):

    75

    """Test Beremiz as whole application"""

    76

    def StartApp(self):

    77

        self.app = Beremiz.BeremizIDELauncher()

    78

        # disable default exception handler in Beremiz

    79

        self.app.InstallExceptionHandler = lambda: None

    80

        self.InstallExceptionHandler()

    81

        self.app.PreStart()

    82

    83

    def FinishApp(self):

    84

        wx.CallAfter(self.app.frame.Close)

    85

        self.app.MainLoop()

    86

        time.sleep(1)

    87

        self.app = None

    88

    89

    def OpenAllProjectElements(self):

    91

        self.app.frame.ProjectTree.ExpandAll()

    92

        self.ProcessEvents()

    93

        item = self.app.frame.ProjectTree.GetRootItem()

    94

        while item is not None:

    95

            self.app.frame.ProjectTree.SelectItem(item, True)

    96

            self.ProcessEvents()

    97

            id = self.app.frame.ProjectTree.GetId()

    98

            event = wx.lib.agw.customtreectrl.TreeEvent(

    99

                wx.lib.agw.customtreectrl.wxEVT_TREE_ITEM_ACTIVATED,

   100

                id, item)

   101

            self.app.frame.OnProjectTreeItemActivated(event)

   102

            self.ProcessEvents()

   103

            item = self.app.frame.ProjectTree.GetNextVisible(item)

   104

   105

    def CheckTestProject(self, project):

   106

        sys.argv = ["", project]

   107

        self.StartApp()

   108

        self.OpenAllProjectElements()

   109

   110

        user_actions = [

   111

            [self.app.frame.SwitchFullScrMode, None],

   112

            [self.app.frame.SwitchFullScrMode, None],

   113

            [self.app.frame.CTR._Clean],

   114

            [self.app.frame.CTR._Build],

   115

            [self.app.frame.CTR._Connect],

   116

            [self.app.frame.CTR._Transfer],

   117

            [self.app.frame.CTR._Run],

   118

            [self.app.frame.CTR._Stop],

   119

            [self.app.frame.CTR._Disconnect],

   120

        ]

   121

   122

        self.RunUIActions(user_actions)

   123

        self.FinishApp()

   124

   125

    def GetProjectPath(self, project):

   126

        return os.path.abspath(os.path.join(os.path.dirname(__file__), "..", project))

   127

   128

    def testStartUp(self):

   129

        """Checks whether the app starts and finishes correctly"""

   130

        self.StartApp()

   131

        self.FinishApp()

   132

   133

    # @unittest.skip("")

   134

    def testOpenExampleProjects(self):

   135

        """Opens, builds and runs user PLC examples from tests directory"""

   136

        prj = [

   137

            "first_steps",

   138

            "logging",

   139

            "svgui",

   140

            "traffic_lights",

   141

            "wxGlade",

   142

            "python",

   143

            "wiimote",

   144

            "wxHMI",

   145

        ]

   146

        for name in prj:

   147

            project = self.GetProjectPath(name)

   148

            print "Testing example " + name

   149

            self.CheckTestProject(project)

   150

   151

   152

if __name__ == '__main__':

   153

    unittest.main()