diff options
Diffstat (limited to 'src/testdir/test_netbeans.py')
| -rw-r--r-- | src/testdir/test_netbeans.py | 210 |
1 files changed, 210 insertions, 0 deletions
diff --git a/src/testdir/test_netbeans.py b/src/testdir/test_netbeans.py new file mode 100644 index 0000000..0d6b096 --- /dev/null +++ b/src/testdir/test_netbeans.py @@ -0,0 +1,210 @@ +#!/usr/bin/python +# +# Server that will communicate with Vim through the netbeans interface. +# Used by test_netbeans.vim. +# +# This requires Python 2.6 or later. + +from __future__ import print_function +import socket +import sys +import time +import threading +import re + +try: + # Python 3 + import socketserver +except ImportError: + # Python 2 + import SocketServer as socketserver + +class ThreadedTCPRequestHandler(socketserver.BaseRequestHandler): + + def process_msgs(self, msgbuf): + # Process all the received netbeans commands/responses/events from Vim. + # Each one is separated by a newline character. If a partial command + # is received, process it later after the rest of it is received. + while True: + (line, sep, rest) = msgbuf.partition('\n') + if sep == '': + # received partial line + return line + msgbuf = rest + + # Process a command only after receiving a newline. + response = '' + if line.find('Xcmdbuf') > 0: + name = line.split('"')[1] + response = '1:putBufferNumber!15 "' + name + '"\n' + response += '1:startDocumentListen!16\n' + elif re.match('1:insert=.* "\\\\n"', line): + # extract the command from the previous line + cmd = re.search('.*"(.*)"', self.prev_line).group(1) + + # map of test names and the netbeans commands/functions + testmap = { + 'getCursor_Test' : '0:getCursor/30\n', + 'E627_Test' : '0 setReadOnly!31\n', + 'E628_Test' : '0:setReadOnly 32\n', + 'E632_Test' : '0:getLength/33\n', + 'E633_Test' : '0:getText/34\n', + 'E634_Test' : '0:remove/35 1 1\n', + 'E635_Test' : '0:insert/36 0 "line1\\n"\n', + 'E636_Test' : '0:create!37\n', + 'E637_Test' : '0:startDocumentListen!38\n', + 'E638_Test' : '0:stopDocumentListen!39\n', + 'E639_Test' : '0:setTitle!40 "Title"\n', + 'E640_Test' : '0:initDone!41\n', + 'E641_Test' : '0:putBufferNumber!42 "XSomeBuf"\n', + 'E642_Test' : '9:putBufferNumber!43 "XInvalidBuf"\n', + 'E643_Test' : '0:setFullName!44 "XSomeBuf"\n', + 'E644_Test' : '0:editFile!45 "Xfile3"\n', + 'E645_Test' : '0:setVisible!46 T\n', + 'E646_Test' : '0:setModified!47 T\n', + 'E647_Test' : '0:setDot!48 1/1\n', + 'E648_Test' : '0:close!49\n', + 'E650_Test' : '0:defineAnnoType!50 1 "abc" "a" "a" 1 1\n', + 'E651_Test' : '0:addAnno!51 1 1 1 1\n', + 'E652_Test' : '0:getAnno/52 8\n', + 'editFile_Test' : '2:editFile!53 "Xfile3"\n', + 'getLength_Test' : '2:getLength/54\n', + 'getModified_Test' : '2:getModified/55\n', + 'getText_Test' : '2:getText/56\n', + 'setDot_Test' : '2:setDot!57 3/6\n', + 'setDot2_Test' : '2:setDot!57 9\n', + 'startDocumentListen_Test' : '2:startDocumentListen!58\n', + 'stopDocumentListen_Test' : '2:stopDocumentListen!59\n', + 'define_anno_Test' : '2:defineAnnoType!60 1 "s1" "x" "=>" blue none\n', + 'E532_Test' : '2:defineAnnoType!61 1 "s1" "x" "=>" aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa none\n', + 'add_anno_Test' : '2:addAnno!62 1 1 2/1 0\n', + 'get_anno_Test' : '2:getAnno/63 1\n', + 'remove_anno_Test' : '2:removeAnno!64 1\n', + 'getModifiedAll_Test' : '0:getModified/65\n', + 'create_Test' : '3:create!66\n', + 'setTitle_Test' : '3:setTitle!67 "Xfile4"\n', + 'setFullName_Test' : '3:setFullName!68 "Xfile4"\n', + 'initDone_Test' : '3:initDone!69\n', + 'setVisible_Test' : '3:setVisible!70 T\n', + 'setModtime_Test' : '3:setModtime!71 6\n', + 'insert_Test' : '3:insert/72 0 "line1\\nline2\\n"\n', + 'remove_Test' : '3:remove/73 3 4\n', + 'remove_invalid_offset_Test' : '3:remove/74 900 4\n', + 'remove_invalid_count_Test' : '3:remove/75 1 800\n', + 'guard_Test' : '3:guard!76 8 7\n', + 'setModified_Test' : '3:setModified!77 T\n', + 'setModifiedClear_Test' : '3:setModified!77 F\n', + 'insertDone_Test' : '3:insertDone!78 T F\n', + 'saveDone_Test' : '3:saveDone!79\n', + 'invalidcmd_Test' : '3:invalidcmd!80\n', + 'invalidfunc_Test' : '3:invalidfunc/81\n', + 'removeAnno_fail_Test' : '0:removeAnno/82 1\n', + 'guard_fail_Test' : '0:guard/83 1 1\n', + 'save_fail_Test' : '0:save/84\n', + 'netbeansBuffer_fail_Test' : '0:netbeansBuffer/85 T\n', + 'setExitDelay_Test' : '0:setExitDelay!86 2\n', + 'setReadOnly_Test' : '3:setReadOnly!87 T\n', + 'setReadOnlyClear_Test' : '3:setReadOnly!88 F\n', + 'save_Test' : '3:save!89\n', + 'close_Test' : '3:close!90\n', + 'specialKeys_Test' : '0:specialKeys!91 "F12 F13 C-F13"\n', + 'nbbufwrite_Test' : '4:editFile!92 "XnbBuffer"\n4:netbeansBuffer!93 T\n', + 'startAtomic_Test' : '0:startAtomic!94\n', + 'endAtomic_Test' : '0:endAtomic!95\n', + 'AnnoScale_Test' : "".join(['2:defineAnnoType!60 ' + str(i) + ' "s' + str(i) + '" "x" "=>" blue none\n' for i in range(2, 26)]), + 'detach_Test' : '2:close!96\n1:close!97\nDETACH\n' + } + # execute the specified test + if cmd not in testmap: + print("=== invalid command %s ===" % (cmd)) + else: + response = testmap[cmd] + elif line.find('disconnect') > 0: + # we're done + self.server.shutdown() + return + + # save the current line, this is used as the test to run after + # receiving a newline only line. + self.prev_line = line + + if len(response) > 0: + self.request.sendall(response.encode('utf-8')) + # Write the response into the file, so that the test can knows + # the command was sent. + with open("Xnetbeans", "a") as myfile: + myfile.write('send: ' + response) + if self.debug: + with open("save_Xnetbeans", "a") as myfile: + myfile.write('send: ' + response) + + def handle(self): + print("=== socket opened ===") + # To preserve the Xnetbeans file as save_Xnetbeans, set debug to 1 + self.debug = 0 + self.prev_line = '' + msgbuf = '' + while True: + try: + received = self.request.recv(4096).decode('utf-8') + except socket.error: + print("=== socket error ===") + break + except IOError: + print("=== socket closed ===") + break + if received == '': + print("=== socket closed ===") + break + print("received: {0}".format(received)) + + # Write the received line into the file, so that the test can check + # what happened. + with open("Xnetbeans", "a") as myfile: + myfile.write(received) + if self.debug: + with open("save_Xnetbeans", "a") as myfile: + myfile.write(received) + + # Can receive more than one line in a response or a partial line. + # Accumulate all the received characters and process one line at + # a time. + msgbuf += received + msgbuf = self.process_msgs(msgbuf) + +class ThreadedTCPServer(socketserver.ThreadingMixIn, socketserver.TCPServer): + pass + +def writePortInFile(port): + # Write the port number in Xportnr, so that the test knows it. + f = open("Xportnr", "w") + f.write("{0}".format(port)) + f.close() + +if __name__ == "__main__": + HOST, PORT = "localhost", 0 + + addrs = socket.getaddrinfo(HOST, PORT, 0, 0, socket.IPPROTO_TCP) + # Each addr is a (family, type, proto, canonname, sockaddr) tuple + sockaddr = addrs[0][4] + ThreadedTCPServer.address_family = addrs[0][0] + + server = ThreadedTCPServer(sockaddr[0:2], ThreadedTCPRequestHandler) + ip, port = server.server_address[0:2] + + # Start a thread with the server. That thread will then start a new thread + # for each connection. + server_thread = threading.Thread(target=server.serve_forever) + server_thread.start() + + writePortInFile(port) + + print("Listening on port {0}".format(port)) + + # Main thread terminates, but the server continues running + # until server.shutdown() is called. + try: + while server_thread.is_alive(): + server_thread.join(1) + except (KeyboardInterrupt, SystemExit): + server.shutdown() |