Exemplo n.º 1
0
class x3270(object):
    def __init__(self,
                 visible=True,
                 timeout='30',
                 wait_time='0.5',
                 img_folder='.'):
        """You can change to hide the emulator screen set the argument visible=${False}
           
           For change the wait_time see `Change Wait Time`, to change the img_folder
           see the `Set Screenshot Folder` and to change the timeout see the `Change Timeout` keyword.
        """
        self.lu = None
        self.host = None
        self.port = None
        self.credential = None
        self.imgfolder = img_folder
        self.output_folder = BuiltIn().get_variable_value('${OUTPUT DIR}')
        self.wait = float(wait_time)
        self.timeout = int(timeout)
        self.mf = Emulator(visible=bool(visible), timeout=int(timeout))

    def change_timeout(self, seconds):
        """Change the timeout for connection in seconds.
        """
        self.timeout = int(seconds)

    def open_connection(self, host, LU=None, port=23):
        """Create a connection with IBM3270 mainframe with the default port 23. To make a connection with the mainframe
        you only must inform the Host. You can pass the Logical Unit Name and the Port as opcional.

        Example:
            | Open Connection | Hostname |
            | Open Connection | Hostname | LU=LUname |
            | Open Connection | Hostname | port=992 |
        """
        self.host = host
        self.lu = LU
        self.port = port
        if self.lu:
            self.credential = "%s@%s:%s" % (self.lu, self.host, self.port)
        else:
            self.credential = self.host
        self.mf.connect(self.credential)

    def close_connection(self):
        """Disconnect from the host.
        """
        try:
            self.mf.terminate()
        except socket.error:
            pass

    def change_wait_time(self, wait_time):
        """To give time for the mainframe screen to be "drawn" and receive the 
        next commands, a "wait time" has been created which by default is set to
        0.5 seconds. This is a sleep applied after the follow keywords:

        `Execute Command`
        `Send Enter`
        `Send PF`
        `Write`
        `Write in position`

        If you want to change this value just use this keyword passing the time in seconds.

        Examples:
            | Change Wait Time | 0.5 |
            | Change Wait Time | 2 |
        """
        self.wait = float(wait_time)

    def read(self, ypos, xpos, length):
        """Get a string of "length" at screen co-ordinates "ypos"/"xpos".

           Co-ordinates are 1 based, as listed in the status area of the terminal.

           Example for read a string in the position y=8 / x=10 of a length 15:
               | ${value} | Read | 8 | 10 | 15 |
        """
        self._check_limits(ypos, xpos)
        # Checks if the user has passed a length that will be larger than the x limit of the screen.
        if int(xpos) + int(length) > (80 + 1):
            raise Exception(
                'You have exceeded the x-axis limit of the mainframe screen')
        string = self.mf.string_get(int(ypos), int(xpos), int(length))
        return str(string)

    def execute_command(self, cmd):
        """Execute an [http://x3270.bgp.nu/wc3270-man.html#Actions|x3270 command].

           Examples:
               | Execute Command | Enter |
               | Execute Command | Home |
               | Execute Command | Tab |
               | Execute Command | PF(1) |
        """
        self.mf.exec_command(str(cmd))
        time.sleep(self.wait)

    def set_screenshot_folder(self, path):
        """Set a folder to keep the html files generated by the `Take Screenshot` keyword.

           Example:
               | Set Screenshot Folder | C:\\\Temp\\\Images |
        """
        if os.path.exists(
                os.path.normpath(os.path.join(self.output_folder, path))):
            self.imgfolder = path
        else:
            logger.error('Given screenshots path "%s" does not exist' % path)
            logger.warn('Screenshots will be saved in "%s"' % self.imgfolder)

    def take_screenshot(self, height='410', width='670'):
        """Generate a screenshot of the IBM 3270 Mainframe in a html format. The
           default folder is the log folder of RobotFramework, if you want change see the `Set Screenshot Folder`.

           The Screenshot is printed in a iframe log, with the values of height=410 and width=670, you
           can change this values passing them from the keyword. 

           Examples:
               | Take Screenshot |
               | Take Screenshot | height=500 | width=700 |
        """
        filename_prefix = 'screenshot'
        extension = 'html'
        filename_sufix = str(int(round(time.time() * 1000)))
        filepath = os.path.join(
            self.imgfolder,
            '%s_%s.%s' % (filename_prefix, filename_sufix, extension))
        self.mf.save_screen(os.path.join(self.output_folder, filepath))
        logger.write('<iframe src="%s" height="%s" width="%s"></iframe>' %
                     (filepath.replace("\\", "/"), height, width),
                     level='INFO',
                     html=True)

    def wait_field_detected(self):
        """Wait until the screen is ready, the cursor has been positioned
        on a modifiable field, and the keyboard is unlocked.

        Sometimes the server will "unlock" the keyboard but the screen
        will not yet be ready.  In that case, an attempt to read or write to the
        screen will result in a 'E' keyboard status because we tried to read from
        a screen that is not yet ready.

        Using this method tells the client to wait until a field is
        detected and the cursor has been positioned on it.
        """
        self.mf.wait_for_field()

    def delete_char(self, ypos=None, xpos=None):
        """Delete character under cursor. If you want to delete a character that is in
           another position, simply pass the coordinates "ypos"/"xpos".

           Co-ordinates are 1 based, as listed in the status area of the
           terminal.

           Examples:
               | Delete Char |
               | Delete Char | ypos=9 | xpos=25 |
        """
        if ypos is not None and xpos is not None:
            self.mf.move_to(int(ypos), int(xpos))
        self.mf.exec_command(b'Delete')

    def delete_field(self, ypos=None, xpos=None):
        """Delete a entire contents in field at current cursor location and positions
           cursor at beginning of field. If you want to delete a field that is in
           another position, simply pass the coordinates "ypos"/"xpos" of any part of the field.

           Co-ordinates are 1 based, as listed in the status area of the
           terminal.

           Examples:
               | Delete Field |
               | Delete Field | ypos=12 | xpos=6 |
        """
        if ypos is not None and xpos is not None:
            self.mf.move_to(int(ypos), int(xpos))
        self.mf.exec_command(b'DeleteField')

    def send_enter(self):
        """Send a Enter to the screen.
        """
        self.mf.send_enter()
        time.sleep(self.wait)

    def move_next_field(self):
        """Move the cursor to the next input field. Equivalent to pressing the Tab key.
        """
        self.mf.exec_command(b'Tab')

    def move_previous_field(self):
        """Move the cursor to the previous input field. Equivalent to pressing the Shift+Tab keys.
        """
        self.mf.exec_command(b'BackTab')

    def send_PF(self, PF):
        """Send a Program Function to the screen.

        Example:
               | Send PF | 3 |
        """
        self.mf.exec_command(b'PF(' + str(PF) + ')')
        time.sleep(self.wait)

    def write(self, txt):
        """Send a string to the screen at the current cursor location *and a Enter.*

           Example:
               | Write | something |
        """
        self._write(txt, enter='1')

    def write_bare(self, txt):
        """Send only the string to the screen at the current cursor location.

           Example:
               | Write Bare | something |
        """
        self._write(txt)

    def write_in_position(self, txt, ypos, xpos):
        """Send a string to the screen at screen co-ordinates "ypos"/"xpos" and a Enter.

           Co-ordinates are 1 based, as listed in the status area of the
           terminal.

           Example:
               | Write in Position | something | 9 | 11 |
        """
        self._check_limits(ypos, xpos)
        self._write(txt, ypos=ypos, xpos=xpos, enter='1')

    def write_bare_in_position(self, txt, ypos, xpos):
        """Send only the string to the screen at screen co-ordinates "ypos"/"xpos".

           Co-ordinates are 1 based, as listed in the status area of the
           terminal.

           Example:
               | Write Bare in Position | something | 9 | 11 |
        """
        self._check_limits(ypos, xpos)
        self._write(txt, ypos=ypos, xpos=xpos)

    def _write(self, txt, ypos=None, xpos=None, enter='0'):
        txt = txt.encode('utf-8')
        if ypos is not None and xpos is not None:
            self.mf.move_to(int(ypos), int(xpos))
        if not isinstance(txt, (list, tuple)): txt = [txt]
        [self.mf.send_string(el) for el in txt if el != '']
        for i in range(int(enter)):
            self.mf.send_enter()
            time.sleep(self.wait)

    def wait_until_string(self, txt, timeout=5):
        """Wait until a string exists on the mainframe screen to perform the next step. If the string not appear on
           5 seconds the keyword will raise a exception. You can define a different timeout.

           Example:
               | Wait Until String | something |
               | Wait Until String | something | timeout=10 |
        """
        max_time = time.ctime(int(time.time()) + int(timeout))
        while time.ctime(int(time.time())) < max_time:
            result = self._search_string(str(txt))
            if result:
                return txt
        raise Exception('String "' + txt + '" not found in ' + str(timeout) +
                        ' seconds')

    def _search_string(self, string, ignore_case=False):
        """Search if a string exists on the mainframe screen and return True or False.
        """
        def __read_screen(string, ignore_case):
            for ypos in range(24):
                line = self.mf.string_get(ypos + 1, 1, 80)
                if ignore_case: line = line.lower()
                if string in line:
                    return True
            return False

        status = __read_screen(string, ignore_case)
        return status

    def page_should_contain_string(self,
                                   txt,
                                   ignore_case=False,
                                   error_message=None):
        """Search if a given string exists on the mainframe screen.

           The search is case sensitive, if you want ignore this you can pass the argument: ignore_case=${True}
           and you can edit the raise exception message with error_message.

           Example:
               | Page Should Contain String | something |
               | Page Should Contain String | someTHING | ignore_case=${True} |
               | Page Should Contain String | something | error_message=New error message |
        """
        message = 'The string "' + txt + '" was not found'
        if error_message: message = error_message
        if ignore_case: txt = str(txt).lower()
        result = self._search_string(txt, ignore_case)
        if not result: raise Exception(message)
        logger.info('The string "' + txt + '" was found')

    def page_should_not_contain_string(self,
                                       txt,
                                       ignore_case=False,
                                       error_message=None):
        """Search if a given string NOT exists on the mainframe screen.

           The search is case sensitive, if you want ignore this you can pass the argument: ignore_case=${True}
           and you can edit the raise exception message with error_message.

           Example:
               | Page Should Not Contain String | something |
               | Page Should Not Contain String | someTHING | ignore_case=${True} |
               | Page Should Not Contain String | something | error_message=New error message |
        """
        message = 'The string "' + txt + '" was found'
        if error_message: message = error_message
        if ignore_case: txt = str(txt).lower()
        result = self._search_string(txt, ignore_case)
        if result: raise Exception(message)

    def page_should_contain_any_string(self,
                                       list_string,
                                       ignore_case=False,
                                       error_message=None):
        """Search if one of the strings in a given list exists on the mainframe screen.

           The search is case sensitive, if you want ignore this you can pass the argument: ignore_case=${True}
           and you can edit the raise exception message with error_message.

           Example:
               | Page Should Contain Any String | ${list_of_string} |
               | Page Should Contain Any String | ${list_of_string} | ignore_case=${True} |
               | Page Should Contain Any String | ${list_of_string} | error_message=New error message |
        """
        message = 'The strings "' + str(list_string) + '" was not found'
        if error_message: message = error_message
        if ignore_case: list_string = [item.lower() for item in list_string]
        for string in list_string:
            result = self._search_string(string, ignore_case)
            if result: break
        if not result: raise Exception(message)

    def page_should_not_contain_any_string(self,
                                           list_string,
                                           ignore_case=False,
                                           error_message=None):
        """Fails if one or more of the strings in a given list exists on the mainframe screen. if one or more of the
        string are found, the keyword will raise a exception.

        The search is case sensitive, if you want ignore this you can pass the argument: ignore_case=${True}
        and you can edit the raise exception message with error_message.

        Example:
            | Page Should Not Contain Any Strings | ${list_of_string} |
            | Page Should Not Contain Any Strings | ${list_of_string} | ignore_case=${True} |
            | Page Should Not Contain Any Strings | ${list_of_string} | error_message=New error message |
        """
        self._compare_all_list_with_screen_text(list_string,
                                                ignore_case,
                                                error_message,
                                                should_match=False)

    def page_should_contain_all_strings(self,
                                        list_string,
                                        ignore_case=False,
                                        error_message=None):
        """Search if all of the strings in a given list exists on the mainframe screen.

        The search is case sensitive, if you want ignore this you can pass the argument: ignore_case=${True}
        and you can edit the raise exception message with error_message.

        Example:
            | Page Should Contain All Strings | ${list_of_string} |
            | Page Should Contain All Strings | ${list_of_string} | ignore_case=${True} |
            | Page Should Contain All Strings | ${list_of_string} | error_message=New error message |
        """
        self._compare_all_list_with_screen_text(list_string,
                                                ignore_case,
                                                error_message,
                                                should_match=True)

    def page_should_not_contain_all_strings(self,
                                            list_string,
                                            ignore_case=False,
                                            error_message=None):
        """Fails if one of the strings in a given list exists on the mainframe screen. if one of the string
        are found, the keyword will raise a exception.

        The search is case sensitive, if you want ignore this you can pass the argument: ignore_case=${True}
        and you can edit the raise exception message with error_message.

        Example:
            | Page Should Not Contain All Strings | ${list_of_string} |
            | Page Should Not Contain All Strings | ${list_of_string} | ignore_case=${True} |
            | Page Should Not Contain All Strings | ${list_of_string} | error_message=New error message |
        """
        message = error_message
        if ignore_case: list_string = [item.lower() for item in list_string]
        for string in list_string:
            result = self._search_string(string, ignore_case)
            if result:
                if message is None:
                    message = 'The string "' + string + '" was found'
                raise Exception(message)

    def page_should_contain_string_x_times(self,
                                           txt,
                                           number,
                                           ignore_case=False,
                                           error_message=None):
        """Search if the entered string appears the desired number of times on the mainframe screen.

        The search is case sensitive, if you want ignore this you can pass the argument: ignore_case=${True} and you
        can edit the raise exception message with error_message.

        Example:
               | Page Should Contain String X Times | something | 3 |
               | Page Should Contain String X Times | someTHING | 3 | ignore_case=${True} |
               | Page Should Contain String X Times | something | 3 | error_message=New error message |
        """
        message = error_message
        number = int(number)
        all_screen = self._read_all_screen()
        if ignore_case:
            txt = str(txt).lower()
            all_screen = str(all_screen).lower()
        number_of_times = all_screen.count(txt)
        if number_of_times != number:
            if message is None:
                message = 'The string "' + txt + '" was not found "' + str(number) + '" times, it appears "' \
                          + str(number_of_times) + '" times'
            raise Exception(message)
        logger.info('The string "' + txt + '" was found "' + str(number) +
                    '" times')

    def page_should_match_regex(self, regex_pattern):
        """Fails if string does not match pattern as a regular expression. Regular expression check is
        implemented using the Python [https://docs.python.org/2/library/re.html|re module]. Python's
        regular expression syntax is derived from Perl, and it is thus also very similar to the syntax used,
        for example, in Java, Ruby and .NET.

        Backslash is an escape character in the test data, and possible backslashes in the pattern must
        thus be escaped with another backslash (e.g. \\\d\\\w+).
        """
        page_text = self._read_all_screen()
        if not re.findall(regex_pattern, page_text, re.MULTILINE):
            raise Exception('No matches found for "' + regex_pattern +
                            '" pattern')

    def page_should_not_match_regex(self, regex_pattern):
        """Fails if string does match pattern as a regular expression. Regular expression check is
        implemented using the Python [https://docs.python.org/2/library/re.html|re module]. Python's
        regular expression syntax is derived from Perl, and it is thus also very similar to the syntax used,
        for example, in Java, Ruby and .NET.

        Backslash is an escape character in the test data, and possible backslashes in the pattern must
        thus be escaped with another backslash (e.g. \\\d\\\w+).
        """
        page_text = self._read_all_screen()
        if re.findall(regex_pattern, page_text, re.MULTILINE):
            raise Exception('There are matches found for "' + regex_pattern +
                            '" pattern')

    def page_should_contain_match(self,
                                  txt,
                                  ignore_case=False,
                                  error_message=None):
        """Fails unless the given string matches the given pattern.

        Pattern matching is similar as matching files in a shell, and it is always case-sensitive.
        In the pattern, * matches to anything and ? matches to any single character.

        Note that the entire screen is only considered a string for this keyword, so if you want to search
        for the string "something" and it is somewhere other than at the beginning or end of the screen it
        should be reported as follows: **something**

        The search is case sensitive, if you want ignore this you can pass the argument: ignore_case=${True} and you
        can edit the raise exception message with error_message.

        Example:
            | Page Should Contain Match | **something** |
            | Page Should Contain Match | **so???hing** |
            | Page Should Contain Match | **someTHING** | ignore_case=${True} |
            | Page Should Contain Match | **something** | error_message=New error message |
        """
        message = error_message
        all_screen = self._read_all_screen()
        if ignore_case:
            txt = txt.lower()
            all_screen = all_screen.lower()
        matcher = Matcher(txt, caseless=False, spaceless=False)
        result = matcher.match(all_screen)
        if not result:
            if message is None:
                message = 'No matches found for "' + txt + '" pattern'
            raise Exception(message)

    def page_should_not_contain_match(self,
                                      txt,
                                      ignore_case=False,
                                      error_message=None):
        """Fails if the given string matches the given pattern.

        Pattern matching is similar as matching files in a shell, and it is always case-sensitive.
        In the pattern, * matches to anything and ? matches to any single character.

        Note that the entire screen is only considered a string for this keyword, so if you want to search
        for the string "something" and it is somewhere other than at the beginning or end of the screen it
        should be reported as follows: **something**

        The search is case sensitive, if you want ignore this you can pass the argument: ignore_case=${True} and you
        can edit the raise exception message with error_message.

        Example:
            | Page Should Not Contain Match | **something** |
            | Page Should Not Contain Match | **so???hing** |
            | Page Should Not Contain Match | **someTHING** | ignore_case=${True} |
            | Page Should Not Contain Match | **something** | error_message=New error message |
        """
        message = error_message
        all_screen = self._read_all_screen()
        if ignore_case:
            txt = txt.lower()
            all_screen = all_screen.lower()
        matcher = Matcher(txt, caseless=False, spaceless=False)
        result = matcher.match(all_screen)
        if result:
            if message is None:
                message = 'There are matches found for "' + txt + '" pattern'
            raise Exception(message)

    def _read_all_screen(self):
        """Read all the mainframe screen and return in a single string.
        """
        full_text = ''
        for ypos in range(24):
            line = self.mf.string_get(ypos + 1, 1, 80)
            for char in line:
                if char:
                    full_text += char
        return full_text

    def _compare_all_list_with_screen_text(self, list_string, ignore_case,
                                           message, should_match):
        if ignore_case: list_string = [item.lower() for item in list_string]
        for string in list_string:
            result = self._search_string(string, ignore_case)
            if not should_match and result:
                if message is None:
                    message = 'The string "' + string + '" was found'
                raise Exception(message)
            elif should_match and not result:
                if message is None:
                    message = 'The string "' + string + '" was not found'
                raise Exception(message)

    @staticmethod
    def _check_limits(ypos, xpos):
        """Checks if the user has passed some coordinate y / x greater than that existing in the mainframe
        """
        if int(ypos) > 24:
            raise Exception(
                'You have exceeded the y-axis limit of the mainframe screen')
        if int(xpos) > 80:
            raise Exception(
                'You have exceeded the x-axis limit of the mainframe screen')
Exemplo n.º 2
0
class Automation3270(object):
    '''Library for x3270 emulator automation. More information at http://x3270.bgp.nu/'''
    ROBOT_LIBRARY_SCOPE = 'GLOBAL'

    def Open_3270(self, hostname):
        '''Opens emulator. Provide the hostname as argument'''
        self.em = Emulator(visible=True)
        self.em.connect(str(hostname))

    def Open_New_3270(self):
        '''Closes all previous instances before starting'''
        self.process_name = 'wc3270.exe'
        self.get_pid = [
            self.item.split()[1]
            for self.item in os.popen('tasklist').read().splitlines()[4:]
            if self.process_name in self.item.split()
        ]
        if self.get_pid[0] == "":
            self.em = Emulator(visible=True)
            self.em.connect('prda.bcpcorp.net')
        else:
            print 'Program already open'

    def Close_3270(self):
        '''Closes emulator'''
        self.pid_proc = 'wc3270.exe'
        self.pid = [
            self.item.split()[1]
            for self.item in os.popen('tasklist').read().splitlines()[4:]
            if self.pid_proc in self.item.split()
        ]
        print "PID number is " + self.pid[0]
        os.kill(int(self.pid[0]), signal.SIGTERM)

    def Input_Text_on_Field(self, xpos, ypos, text):
        '''Writes text on field. Last parameters requires the lenght of the word'''
        self.em.fill_field(int(xpos), int(ypos), str(text), len(text) + 2)

    def Send_String(self, text, xpos, ypos):
        '''Verifies if a given string exists'''
        self.em.string_found(text, int(xpos), int(ypos))

    def Keyboard_State(self):
        '''If the keyboard is unlocked, the letter U. If the keyboard is locked waiting for a response from the host, or if not connected to a host, the letter L'''
        print self.em.status.keyboard.capitalize()

    def wait_For_Field(self):
        '''Waits for execution'''
        self.em.wait_for_field()

    def Execute_Command(self, command):
        '''Executes custom commands. See more a thttp://x3270.bgp.nu/x3270-script.html'''
        self.em.exec_command(self, command)

    def Connection_State(self):
        '''If connected to a host, the string C(hostname). Otherwise, the letter N'''
        print self.em.status.connection_state.capitalize()

    def Get_String(self, xpos, ypos, lenght):
        '''Get String based on coordinates and lenght'''
        get_text = self.em.string_get(int(xpos), int(ypos), int(lenght))
        return get_text

    def Press_Enter(self):
        '''Presses Enter key'''
        self.em.send_enter()

    def Move_cursor(self, xpos, ypos):
        '''Moves cursor to specified coordinates'''
        self.em.move_to(xpos, ypos)

    def Take_Screenshot(self):
        '''Takes printscreen'''
        self.em.save_screen()

    def Pause_Test(self):
        '''Pauses Test at a given moment'''
        self.em.app.readline()

    def Send_F3(self):
        '''Simulates F3 key'''
        self.em.send_pf3()

    def Send_F4(self):
        '''Simulates F4 key'''
        self.em.send_pf4()

    def Send_F5(self):
        '''Simulates F5 key'''
        self.em.send_pf5()

    def Send_F6(self):
        '''Simulates F6 key'''
        self.em.send_pf6()

    def Send_F7(self):
        '''Simulates F7 key'''
        self.em.send_pf7()

    def Send_F8(self):
        '''Simulates F8 key'''
        self.em.send_pf8()