# 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))
