mirror of
https://github.com/classilla/tenfourfox.git
synced 2024-10-22 02:25:05 +00:00
30 lines
837 B
Python
30 lines
837 B
Python
|
# This Source Code Form is subject to the terms of the Mozilla Public
|
||
|
# License, v. 2.0. If a copy of the MPL was not distributed with this
|
||
|
# file, You can obtain one at http://mozilla.org/MPL/2.0/.
|
||
|
|
||
|
import re
|
||
|
import socket
|
||
|
|
||
|
from dmunit import DeviceManagerTestCase
|
||
|
|
||
|
class PromptTestCase(DeviceManagerTestCase):
|
||
|
|
||
|
def tearDown(self):
|
||
|
if self.sock:
|
||
|
self.sock.close()
|
||
|
|
||
|
def runTest(self):
|
||
|
"""This tests getting a prompt from the device.
|
||
|
"""
|
||
|
self.sock = None
|
||
|
ip = self.dm.host
|
||
|
port = self.dm.port
|
||
|
|
||
|
promptre = re.compile('.*\$\>\x00')
|
||
|
data = ""
|
||
|
self.sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
|
||
|
self.sock.connect((ip, int(port)))
|
||
|
data = self.sock.recv(1024)
|
||
|
print data
|
||
|
self.assertTrue(promptre.match(data))
|