|
1 # This Source Code Form is subject to the terms of the Mozilla Public |
|
2 # License, v. 2.0. If a copy of the MPL was not distributed with this |
|
3 # file, You can obtain one at http://mozilla.org/MPL/2.0/. |
|
4 |
|
5 import re |
|
6 import socket |
|
7 |
|
8 from dmunit import DeviceManagerTestCase |
|
9 |
|
10 class PromptTestCase(DeviceManagerTestCase): |
|
11 |
|
12 def tearDown(self): |
|
13 if self.sock: |
|
14 self.sock.close() |
|
15 |
|
16 def runTest(self): |
|
17 """This tests getting a prompt from the device. |
|
18 """ |
|
19 self.sock = None |
|
20 ip = self.dm.host |
|
21 port = self.dm.port |
|
22 |
|
23 promptre = re.compile('.*\$\>\x00') |
|
24 data = "" |
|
25 self.sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM) |
|
26 self.sock.connect((ip, int(port))) |
|
27 data = self.sock.recv(1024) |
|
28 print data |
|
29 self.assertTrue(promptre.match(data)) |