def test_create_request():
    request = create_request(SAMPLE_REQUEST)

    assert isinstance(request, ViberUnsubscribedRequest)
    assert request.event_type == SAMPLE_REQUEST['event']
    assert request.timestamp == SAMPLE_REQUEST['timestamp']
    assert request.user_id == SAMPLE_REQUEST['user_id']
Exemplo n.º 2
0
 def parse_request(self, request_data):
     self._logger.debug("parsing request")
     request_dict = json.loads(request_data.decode(
     ) if isinstance(request_data, bytes) else request_data)
     request = create_request(request_dict)
     self._logger.debug(u"parsed request={0}".format(request))
     return request
Exemplo n.º 3
0
    async def webhook_handle(self, request) -> web.Response:
        data = await request.json()
        viber_request = create_request(data)  # type: ViberRequest

        t = self.loop.create_task(self._process_request(viber_request))
        # TODO add error processing

        return web.Response()
Exemplo n.º 4
0
def test_create_request_missing_event():
	sample_request = dict(
		timestamp=datetime.now(),
		message_token="912661846655238145",
		sender=dict(
			id="01234567890A=",
			name="viberUser",
			avatar="http://avatar_url"
		),
		message=dict(
			type=MessageType.TEXT,
			text="HI!"
		))

	with pytest.raises(Exception) as exc:
		create_request(sample_request)
		assert exc.value.message.startswith("request is missing field 'event'")
Exemplo n.º 5
0
def test_create_request():
    request = create_request(SAMPLE_REQUEST)

    assert isinstance(request, ViberSeenRequest)
    assert request.event_type == SAMPLE_REQUEST['event']
    assert request.timestamp == SAMPLE_REQUEST['timestamp']
    assert request.meesage_token == SAMPLE_REQUEST['message_token']
    assert request.user_id == SAMPLE_REQUEST['user_id']
Exemplo n.º 6
0
def test_create_request():
    request = create_request(SAMPLE_REQUEST)

    assert isinstance(request, ViberSubscribedRequest)
    assert request.event_type == SAMPLE_REQUEST['event']
    assert request.timestamp == SAMPLE_REQUEST['timestamp']
    assert request.user.id == SAMPLE_REQUEST['user']['id']
    assert request.user.name == SAMPLE_REQUEST['user']['name']
    assert request.user.avatar == SAMPLE_REQUEST['user']['avatar']
    assert request.user.country == SAMPLE_REQUEST['user']['country']
    assert request.user.language == SAMPLE_REQUEST['user']['language']
def test_create_request():
    request = create_request(SAMPLE_REQUEST)

    assert isinstance(request, ViberMessageRequest)
    assert request.event_type == SAMPLE_REQUEST['event']
    assert request.timestamp == SAMPLE_REQUEST['timestamp']
    assert request.message_token == SAMPLE_REQUEST['message_token']
    assert request.silent == SAMPLE_REQUEST['silent']
    assert request.sender.id == SAMPLE_REQUEST['sender']['id']
    assert request.sender.name == SAMPLE_REQUEST['sender']['name']
    assert request.sender.avatar == SAMPLE_REQUEST['sender']['avatar']
    assert isinstance(request.message, TextMessage)
Exemplo n.º 8
0
def test_create_request():
    request = create_request(SAMPLE_REQUEST)

    assert isinstance(request, ViberConversationStartedRequest)
    assert request.event_type == SAMPLE_REQUEST['event']
    assert request.timestamp == SAMPLE_REQUEST['timestamp']
    assert request.message_token == SAMPLE_REQUEST['message_token']
    assert request.type == SAMPLE_REQUEST['type']
    assert request.context == SAMPLE_REQUEST['context']
    assert request.user.id == SAMPLE_REQUEST['user']['id']
    assert request.user.name == SAMPLE_REQUEST['user']['name']
    assert request.user.avatar == SAMPLE_REQUEST['user']['avatar']
    assert request.user.country == SAMPLE_REQUEST['user']['country']
    assert request.user.language == SAMPLE_REQUEST['user']['language']
    assert request.subscribed == SAMPLE_REQUEST['subscribed']
def test_user_has_no_avatar():
    SAMPLE_REQUEST = dict(event=EventType.MESSAGE,
                          timestamp=datetime.now(),
                          message_token="912661846655238145",
                          sender=dict(id="01234567890A=", name="viberUser"),
                          message=dict(type=MessageType.TEXT, text="HI!"),
                          silent=False)
    request = create_request(SAMPLE_REQUEST)

    assert isinstance(request, ViberMessageRequest)
    assert request.event_type == SAMPLE_REQUEST['event']
    assert request.timestamp == SAMPLE_REQUEST['timestamp']
    assert request.message_token == SAMPLE_REQUEST['message_token']
    assert request.sender.id == SAMPLE_REQUEST['sender']['id']
    assert request.sender.name == SAMPLE_REQUEST['sender']['name']
    assert isinstance(request.message, TextMessage)
Exemplo n.º 10
0
    def parse_message(self, request: Request) -> Message:
        # NOTE: There is no way to get the body
        #       after processing the request in DRF.
        # # Verify signature
        # sign = request.META.get('HTTP_X_VIBER_CONTENT_SIGNATURE')
        # if not self.bot.verify_signature(request.body, sign):
        #     raise IMApiException(f'Viber message not verified; '
        #                          f'Data={request.data}; Sign={sign};')

        # Parse message data in to viber types
        vb_request = create_request(request.data)

        try:
            return self._get_message(vb_request)
        except Exception as err:
            # TODO: remove this after development
            log.exception(f'Parse message error; Message={vb_request}; '
                          f'Error={err};')
            return Message(MessageType.UNDEFINED)
def test_request_is_chat_extension():
    SAMPLE_REQUEST = dict(event=EventType.MESSAGE,
                          timestamp=datetime.now(),
                          message_token="912661846655238145",
                          chat_id=5048086924903818307,
                          reply_type='message',
                          sender=dict(id="01234567890A=", name="viberUser"),
                          message=dict(type=MessageType.TEXT, text="HI!"),
                          silent=False)
    request = create_request(SAMPLE_REQUEST)

    assert isinstance(request, ViberMessageRequest)
    assert request.event_type == SAMPLE_REQUEST['event']
    assert request.timestamp == SAMPLE_REQUEST['timestamp']
    assert request.message_token == SAMPLE_REQUEST['message_token']
    assert request.sender.id == SAMPLE_REQUEST['sender']['id']
    assert request.sender.name == SAMPLE_REQUEST['sender']['name']
    assert request.chat_id == SAMPLE_REQUEST['chat_id']
    assert request.reply_type == SAMPLE_REQUEST['reply_type']
    assert isinstance(request.message, TextMessage)
Exemplo n.º 12
0
def test_missing_user_optional_params():
    request_dict = dict(event=EventType.CONVERSATION_STARTED,
                        timestamp=datetime.now(),
                        message_token="912661846655238145",
                        type="open",
                        context="context info",
                        user=dict(id="01234567890A=", name="viberUser"))

    request = create_request(request_dict)  # should not fail

    assert isinstance(request, ViberConversationStartedRequest)
    assert request.event_type == request_dict['event']
    assert request.timestamp == request_dict['timestamp']
    assert request.message_token == request_dict['message_token']
    assert request.type == request_dict['type']
    assert request.context == request_dict['context']
    assert request.user.id == request_dict['user']['id']
    assert request.user.name == request_dict['user']['name']
    assert request.user.avatar is None
    assert request.user.country is None
    assert request.user.language is None
    assert request.subscribed is None
Exemplo n.º 13
0
	def parse_request(self, request_data):
		self._logger.debug("parsing request")
		request_dict = json.loads(request_data)
		request = create_request(request_dict)
		self._logger.debug(u"parsed request={0}".format(request))
		return request