1

#!/usr/bin/env python

     2

# -*- coding: utf-8 -*-

     3

     4

# This file is part of Beremiz, a Integrated Development Environment for

     5

# programming IEC 61131-3 automates supporting plcopen standard and CanFestival.

     6

#

     7

# Copyright (C) 2017: Andrey Skvortsov

     8

#

     9

# See COPYING file for copyrights details.

    10

#

    11

# This program is free software; you can redistribute it and/or

    12

# modify it under the terms of the GNU General Public License

    13

# as published by the Free Software Foundation; either version 2

    14

# of the License, or (at your option) any later version.

    15

#

    16

# This program is distributed in the hope that it will be useful,

    17

# but WITHOUT ANY WARRANTY; without even the implied warranty of

    18

# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

    19

# GNU General Public License for more details.

    20

#

    21

# You should have received a copy of the GNU General Public License

    22

# along with this program; if not, write to the Free Software

    23

# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.

    24

    25

import os

    26

import sys

    27

import unittest

    28

import pytest

    29

import wx

    30

import time

    31

import traceback

    32

import ddt

    33

    34

import conftest

    35

import Beremiz

    36

    37

    38

class UserApplicationTest(unittest.TestCase):

    39

    def InstallExceptionHandler(self):

    40

        def handle_exception(e_type, e_value, e_traceback):

    41

            # traceback.print_exception(e_type, e_value, e_traceback)

    42

            self.exc_info = [e_type, e_value, e_traceback]

    43

        self.exc_info = None

    44

        self.old_excepthook = sys.excepthook

    45

        sys.excepthook = handle_exception

    46

    47

    def StartApp(self):

    48

        self.app = None

    49

    50

    def FinishApp(self):

    51

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

    52

        self.app.MainLoop()

    53

        self.app = None

    54

    55

    def setUp(self):

    56

        self.app = None

    57

    58

    def tearDown(self):

    59

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

    60

            self.FinishApp()

    61

    62

    def RunUIActions(self, actions):

    63

        for act in actions:

    64

            wx.CallAfter(*act)

    65

            self.ProcessEvents()

    66

    67

    def CheckForErrors(self):

    68

        if self.exc_info is not None:

    69

            # reraise catched previously exception

    70

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

    71

    72

    def ProcessEvents(self):

    73

        for i in range(0, 30):

    74

            self.CheckForErrors()

    75

            wx.Yield()

    76

            time.sleep(0.01)

    77

    78

    79

@ddt.ddt

    80

class BeremizApplicationTest(UserApplicationTest):

    81

    """Test Beremiz as whole application"""

    82

    def StartApp(self):

    83

        self.app = Beremiz.BeremizIDELauncher()

    84

        # disable default exception handler in Beremiz

    85

        self.app.InstallExceptionHandler = lambda: None

    86

        self.InstallExceptionHandler()

    87

        self.app.PreStart()

    88

    89

    def FinishApp(self):

    90

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

    91

        self.app.MainLoop()

    92

        time.sleep(1)

    93

        self.app = None

    94

    95

    def OpenAllProjectElements(self):

    96

        """Open editor for every object in the project tree"""

    97

        self.app.frame.ProjectTree.ExpandAll()

    98

        self.ProcessEvents()

    99

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

   100

        while item is not None:

   101

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

   102

            self.ProcessEvents()

   103

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

   104

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

   105

                wx.lib.agw.customtreectrl.wxEVT_TREE_ITEM_ACTIVATED,

   106

                id, item)

   107

            self.app.frame.OnProjectTreeItemActivated(event)

   108

            self.ProcessEvents()

   109

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

   110

   111

    def CheckTestProject(self, name):

   112

        project = self.GetProjectPath(name)

   113

        print "Testing example " + name

   114

        sys.argv = ["", project]

   115

        self.StartApp()

   116

        self.OpenAllProjectElements()

   117

   118

        user_actions = [

   119

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

   120

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

   121

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

   122

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

   123

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

   124

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

   125

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

   126

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

   127

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

   128

        ]

   129

   130

        self.RunUIActions(user_actions)

   131

        self.FinishApp()

   132

   133

    def GetProjectPath(self, project):

   134

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

   135

   136

    def testStartUp(self):

   137

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

   138

        self.StartApp()

   139

        self.FinishApp()

   140

   141

    @ddt.data(

   142

        "first_steps",

   143

        "logging",

   144

        "svgui",

   145

        "traffic_lights",

   146

        "wxGlade",

   147

        "python",

   148

        "wiimote",

   149

        "wxHMI",

   150

    )

   151

    @pytest.mark.timeout(30)

   152

    def testCheckProject(self, name):

   153

        self.CheckTestProject(name)

   154

   155

   156

if __name__ == '__main__':

   157

    unittest.main()