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)