def test_connect(self):
terminal = self.terminal
protocol = DeviceProtocol(self.terminal, self.configuration)
protocol.connect()
assert_equals(terminal.output[-2], DeviceProtocol._default_login_banner)
assert_equals(terminal.output[-1], DeviceProtocol._default_prompt)
assert_equals(len(terminal.output), 2)
def test_logout(self):
terminal = self.terminal
protocol = DeviceProtocol(self.terminal, self.configuration)
protocol._logout()
assert_equals(terminal.output[-1], DeviceProtocol._default_logout_banner)
assert_false(terminal.connected)
def test_logout_on_empty_context(self):
protocol = DeviceProtocol(self.terminal, self.configuration)
protocol.command('exit')
assert_false(self.terminal.connected)
