Python Mocket.register方法代码示例

# 需要导入模块: from mocket.mocket import Mocket [as 别名]
# 或者: from mocket.mocket.Mocket import register [as 别名]
    def _test_write(self):
        frame = "%s>%s:%s" % (
            self.random(6),
            ','.join([self.random(6), self.random(6), self.random(6)]),
            ' '.join([
                self.random(), 'test_write', self.random()])
        )
        aprs_frame = aprs.Frame(frame)
        kiss_frame = aprs_frame.encode_kiss()

        ks = kiss.TCPKISS(host=self.random_host, port=self.random_port)
        a = (self.random_host, self.random_port)

        entry = MocketEntry(a, kiss_frame)
        Mocket.register(entry)
        self._logger.debug(a)
        self._logger.debug(entry.get_response())

        ks.interface = create_connection(a)
        ks._write_handler = ks.interface.sendall

        def _pass(): pass

        ks.stop = _pass

        ks.write(kiss_frame)

# 需要导入模块: from mocket.mocket import Mocket [as 别名]
# 或者: from mocket.mocket.Mocket import register [as 别名]
    def test_write_and_read(self):
        """Tests writing-to and reading-from TCP Host."""
        frame = "%s>%s:%s" % (
            self.random(6),
            ','.join([self.random(6), self.random(6), self.random(6)]),
            ' '.join([
                self.random(), 'test_write_and_read', self.random()])
        )
        aprs_frame = aprs.Frame(frame)
        kiss_frame = aprs_frame.encode_kiss()

        ks = kiss.TCPKISS(host=self.random_host, port=self.random_port)
        a = (self.random_host, self.random_port)

        entry = MocketEntry(a, [kiss_frame])
        entry_1 = MocketEntry(('localhost', 80), True)
        Mocket.register(entry)

        ks.interface = create_connection(a)
        ks._write_handler = ks.interface.sendall

        def _pass(): pass
        ks.stop = _pass

        ks.write(kiss_frame)

        _read_data = ks.read(len(kiss_frame), readmode=False)

        self._logger.info(
            '_read_data(%s)="%s"', len(_read_data), _read_data)

# 需要导入模块: from mocket.mocket import Mocket [as 别名]
# 或者: from mocket.mocket.Mocket import register [as 别名]
 def single_register(cls, method, uri, body='', status=200, headers=None, match_querystring=True):
     entry = cls(
         uri, method, Response(
             body=body, status=status, headers=headers
         ), match_querystring=match_querystring
     )
     Mocket.register(entry)
     return entry

# 需要导入模块: from mocket.mocket import Mocket [as 别名]
# 或者: from mocket.mocket.Mocket import register [as 别名]
 def test_register(self):
     entry_1 = MocketEntry(('localhost', 80), True)
     entry_2 = MocketEntry(('localhost', 80), True)
     entry_3 = MocketEntry(('localhost', 8080), True)
     Mocket.register(entry_1, entry_2, entry_3)
     self.assertEqual(Mocket._entries, {
         ('localhost', 80): [entry_1, entry_2],
         ('localhost', 8080): [entry_3],
     })

# 需要导入模块: from mocket.mocket import Mocket [as 别名]
# 或者: from mocket.mocket.Mocket import register [as 别名]
    def test_client_unsubscribes_to_event(self):

        ack = C.ACTIONS_ACK + C.MESSAGE_PART_SEPARATOR + C.ACTIONS_ACK + C.MESSAGE_SEPARATOR
        unsubscribe_ack = C.TOPIC_EVENT + C.MESSAGE_PART_SEPARATOR + C.ACTIONS_ACK + C.MESSAGE_PART_SEPARATOR + C.ACTIONS_UNSUBSCRIBE + C.MESSAGE_PART_SEPARATOR + "test1" + C.MESSAGE_SEPARATOR
        unsubscribe = C.TOPIC_EVENT + C.MESSAGE_PART_SEPARATOR + C.ACTIONS_UNSUBSCRIBE + C.MESSAGE_PART_SEPARATOR + "test1" + C.MESSAGE_SEPARATOR
        Mocket.register(MocketEntry(('localhost', 8989), [str.encode(ack), str.encode(unsubscribe_ack)]))
        ds = DeepStreamClient('localhost', 8989)
        ds.login({"username": "XXX", "password": "YYY"}, None)
        time.sleep(0.1)
        ds.event.unsubscribe("test1", None)
        time.sleep(0.1)
        assert Mocket.last_request() == (str(str.encode(unsubscribe)))

# 需要导入模块: from mocket.mocket import Mocket [as 别名]
# 或者: from mocket.mocket.Mocket import register [as 别名]
 def test_client_receives_login_confirmation(self):
     """
     Scenario: The client receives a login confirmation
         When the server sends the message A|A+
         Then the clients connection state is "OPEN"
             And the last login was successful
     """
     ack = C.ACTIONS_ACK + C.MESSAGE_PART_SEPARATOR + C.ACTIONS_ACK + C.MESSAGE_SEPARATOR
     Mocket.register(MocketEntry(('localhost', 8989), [str.encode(ack)]))
     ds = DeepStreamClient('localhost', 8989)
     ds.login({"username": "XXX", "password": "YYY"}, None)
     time.sleep(0.1)
     assert ds.get_connection_state() == C.CONNECTION_STATE_OPEN

# 需要导入模块: from mocket.mocket import Mocket [as 别名]
# 或者: from mocket.mocket.Mocket import register [as 别名]
 def test_client_made_too_many_unsuccessful_authentication_attempts(self):
     """
     Scenario: The client has made too many unsuccessful authentication attempts
         Given the client is initialised
         When the client logs in with username "XXX" and password "ZZZ"
             But the server sends the message A|E|TOO_MANY_AUTH_ATTEMPTS|Stoo many authentication attempts+
         Then the last login failed with error "TOO_MANY_AUTH_ATTEMPTS" and message "too many authentication attempts"
     """
     too_many_auth_attempts = C.TOPIC_AUTH + C.MESSAGE_PART_SEPARATOR + C.ACTIONS_ERROR + C.MESSAGE_PART_SEPARATOR + "TOO_MANY_AUTH_ATTEMPTS" + C.MESSAGE_SEPARATOR
     Mocket.register(MocketEntry(('localhost', 8989), [str.encode(too_many_auth_attempts)]))
     ds = DeepStreamClient('localhost', 8989)
     ds.login({"username": "XXX", "password": "YYY"}, None)
     time.sleep(0.1)
     assert ds.get_connection_state() == C.CONNECTION_STATE_AUTHENTICATING

# 需要导入模块: from mocket.mocket import Mocket [as 别名]
# 或者: from mocket.mocket.Mocket import register [as 别名]
 def test_client_receives_invalid_authentication_data(self):
     """
     Scenario: The client's authentication data is rejected
         Given the client is initialised
         When the client logs in with username "XXX" and password "ZZZ"
             But the server sends the message A|E|INVALID_AUTH_DATA|Sinvalid authentication data+
         Then the last login failed with error "INVALID_AUTH_DATA" and message "invalid authentication data"
     """
     invalid_auth_data = C.TOPIC_AUTH + C.MESSAGE_PART_SEPARATOR + C.ACTIONS_ERROR + C.MESSAGE_PART_SEPARATOR + "INVALID_AUTH_DATA" + C.MESSAGE_PART_SEPARATOR + "Sinvalid authentication data" + C.MESSAGE_SEPARATOR
     Mocket.register(MocketEntry(('localhost', 8989), [str.encode(invalid_auth_data)]))
     ds = DeepStreamClient('localhost', 8989)
     ds.login({"username": "XXX", "password": "YYY"}, None)
     time.sleep(0.1)
     assert ds.get_connection_state() == C.CONNECTION_STATE_AWAITING_AUTHENTICATION

# 需要导入模块: from mocket.mocket import Mocket [as 别名]
# 或者: from mocket.mocket.Mocket import register [as 别名]
 def test_client_sends_login_credentials(self):
     """
     Scenario: The client sends login credentials
         Given the test server is ready
             And the client is initialised
         When the client logs in with username "XXX" and password "YYY"
         Then the last message the server recieved is A|REQ|{"username":"XXX","password":"YYY"}+
             And the clients connection state is "AUTHENTICATING"
     """
     Mocket.register(MocketEntry(('localhost', 8989), 0))
     ds = DeepStreamClient('localhost', 8989)
     ds.login({"username": "XXX", "password": "YYY"}, None)
     time.sleep(0.1)
     auth_msg = C.TOPIC_AUTH + C.MESSAGE_PART_SEPARATOR + C.ACTIONS_REQUEST + C.MESSAGE_PART_SEPARATOR + "{\"password\": \"YYY\", \"username\": \"XXX\"}" + C.MESSAGE_SEPARATOR
     assert (Mocket.last_request()) == (str(str.encode(auth_msg)))
     assert ds.get_connection_state() == C.CONNECTION_STATE_AUTHENTICATING

# 需要导入模块: from mocket.mocket import Mocket [as 别名]
# 或者: from mocket.mocket.Mocket import register [as 别名]
 def test_client_listens_to_event_prefix(self):
     '''
     Scenario: The client listens to eventPrefix
         When the client listens to events matching "eventPrefix/.*"
         Then the last message the server received is E|L|eventPrefix/.*+
         Then the server sends the message E|A|L|eventPrefix/.*+
     '''
     ack = C.ACTIONS_ACK + C.MESSAGE_PART_SEPARATOR + C.ACTIONS_ACK + C.MESSAGE_SEPARATOR
     listen_ack = C.TOPIC_EVENT + C.MESSAGE_PART_SEPARATOR + C.ACTIONS_ACK + C.MESSAGE_PART_SEPARATOR + C.ACTIONS_LISTEN + C.MESSAGE_PART_SEPARATOR + "regex/\*" + C.MESSAGE_SEPARATOR
     listen = C.TOPIC_EVENT + C.MESSAGE_PART_SEPARATOR + C.ACTIONS_LISTEN + C.MESSAGE_PART_SEPARATOR + "regex/\*" + C.MESSAGE_SEPARATOR
     Mocket.register(MocketEntry(('localhost', 8989), [str.encode(ack), str.encode(listen_ack)]))
     ds = DeepStreamClient('localhost', 8989)
     ds.login({"username": "XXX", "password": "YYY"}, None)
     time.sleep(0.1)
     ds.event.listen("regex/\*", None)
     time.sleep(0.1)
     assert Mocket.last_request() == (str(str.encode(listen)))

# 需要导入模块: from mocket.mocket import Mocket [as 别名]
# 或者: from mocket.mocket.Mocket import register [as 别名]
 def test_client_subscribes_to_event(self):
     """
     Scenario: The client subscribes to an event
         Given the client subscribes to an event named "test1"
         Then the last message the server received is E|S|test1+
     	Then the server sends the message E|A|S|test1+
     """
     ack = C.ACTIONS_ACK + C.MESSAGE_PART_SEPARATOR + C.ACTIONS_ACK + C.MESSAGE_SEPARATOR
     subscribe_ack = C.TOPIC_EVENT + C.MESSAGE_PART_SEPARATOR + C.ACTIONS_ACK + C.MESSAGE_PART_SEPARATOR + C.ACTIONS_SUBSCRIBE + C.MESSAGE_PART_SEPARATOR + "test1" + C.MESSAGE_SEPARATOR
     subscribe = C.TOPIC_EVENT + C.MESSAGE_PART_SEPARATOR + C.ACTIONS_SUBSCRIBE + C.MESSAGE_PART_SEPARATOR + "test1" + C.MESSAGE_SEPARATOR
     Mocket.register(MocketEntry(('localhost', 8989), [str.encode(ack), str.encode(subscribe_ack)]))
     ds = DeepStreamClient('localhost', 8989)
     ds.login({"username": "XXX", "password": "YYY"}, None)
     time.sleep(0.1)
     ds.event.subscribe("test1", None)
     time.sleep(0.1)
     assert Mocket.last_request() == (str(str.encode(subscribe)))

# 需要导入模块: from mocket.mocket import Mocket [as 别名]
# 或者: from mocket.mocket.Mocket import register [as 别名]
 def test_client_cant_connect_after_TOO_MANY_AUTH_ATTEMPTS(self):
     """
     Scenario: The client can't made further authentication attempts after it received TOO_MANY_AUTH_ATTEMPTS
         Given the server resets its message count
         When the client logs in with username "XXX" and password "ZZZ"
         Then the server has received 0 messages
             And the client throws a "IS_CLOSED" error with message "this client's connection was closed"
     """
     too_many_auth_attempts = C.TOPIC_AUTH + C.MESSAGE_PART_SEPARATOR + C.ACTIONS_ERROR + C.MESSAGE_PART_SEPARATOR + "TOO_MANY_AUTH_ATTEMPTS" + C.MESSAGE_SEPARATOR
     Mocket.register(MocketEntry(('localhost', 8989), [str.encode(too_many_auth_attempts)]))
     ds = DeepStreamClient('localhost', 8989)
     credentials = {"username": "XXX", "password": "YYY"}
     ds.login(credentials, None)
     time.sleep(0.1)
     assert ds.get_connection_state() == C.CONNECTION_STATE_AUTHENTICATING
     time.sleep(0.1)
     with pytest.raises(Exception) as excinfo:
         ds.login(credentials, None)
     assert 'client\'s connection was closed' in str(excinfo.value)

# 需要导入模块: from mocket.mocket import Mocket [as 别名]
# 或者: from mocket.mocket.Mocket import register [as 别名]
    def test_client_throws_error_when_connection_lost(self):
        """
        Scenario: The client loses its connection to the server
            When the connection to the server is lost
            Given two seconds later
            Then the client throws a "connectionError" error with message "Can't connect! Deepstream server unreachable"
                And the clients connection state is "RECONNECTING"
        """
        ack = C.ACTIONS_ACK + C.MESSAGE_PART_SEPARATOR + C.ACTIONS_ACK + C.MESSAGE_SEPARATOR
        Mocket.register(MocketEntry(('localhost', 8989), [str.encode(ack)]))
        ds = DeepStreamClient('localhost', 8989)
        ds.login({"username": "XXX", "password": "YYY"}, None)
        time.sleep(1)
        Mocket.reset()
        time.sleep(1)

        with self.assertRaises(Exception) as c:
            ds.event.subscribe('event1', None)
            time.sleep(2)
        self.assertTrue("Can't connect! Deepstream server unreachable" in c.exception.args[0])

# 需要导入模块: from mocket.mocket import Mocket [as 别名]
# 或者: from mocket.mocket.Mocket import register [as 别名]
 def test_getentry(self):
     entry = MocketEntry(('localhost', 80), True)
     Mocket.register(entry)
     self.assertEqual(Mocket.get_entry('localhost', 80, True), entry)

# 需要导入模块: from mocket.mocket import Mocket [as 别名]
# 或者: from mocket.mocket.Mocket import register [as 别名]
 def test_ko(self):
     Mocket.register(MocketEntry(('localhost', 8080), ['Blue Pill.\r\n']))
     neo = Neo()
     self.assertFalse(neo.iknow())
     self.assertEqual(len(Mocket._requests), 1)