#!/usr/bin/env python

#

# A library that provides a Python interface to the Telegram Bot API

# Copyright (C) 2015-2022

# Leandro Toledo de Souza <devs@python-telegram-bot.org>

#

# This program is free software: you can redistribute it and/or modify

# it under the terms of the GNU Lesser Public License as published by

# the Free Software Foundation, either version 3 of the License, or

# (at your option) any later version.

#

# This program is distributed in the hope that it will be useful,

# but WITHOUT ANY WARRANTY; without even the implied warranty of

# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the

# GNU Lesser Public License for more details.

#

# You should have received a copy of the GNU Lesser Public License

# along with this program. If not, see [http://www.gnu.org/licenses/].

import asyncio

import datetime

import pytest

import pytz

from telegram import (

Bot,

CallbackQuery,

Chat,

ChatInviteLink,

ChatJoinRequest,

ChosenInlineResult,

Message,

PreCheckoutQuery,

ShippingQuery,

Update,

User,

)

from telegram.ext import CallbackContext, ChatJoinRequestHandler, JobQueue

message = Message(1, None, Chat(1, ""), from_user=User(1, "", False), text="Text")

params = [

{"message": message},

{"edited_message": message},

{"callback_query": CallbackQuery(1, User(1, "", False), "chat", message=message)},

{"channel_post": message},

{"edited_channel_post": message},

{"chosen_inline_result": ChosenInlineResult("id", User(1, "", False), "")},

{"shipping_query": ShippingQuery("id", User(1, "", False), "", None)},

{"pre_checkout_query": PreCheckoutQuery("id", User(1, "", False), "", 0, "")},

{"callback_query": CallbackQuery(1, User(1, "", False), "chat")},

]

ids = (

"message",

"edited_message",

"callback_query",

"channel_post",

"edited_channel_post",

"chosen_inline_result",

"shipping_query",

"pre_checkout_query",

"callback_query_without_message",

)

@pytest.fixture(scope="class", params=params, ids=ids)

def false_update(request):

return Update(update_id=2, **request.param)

@pytest.fixture(scope="class")

def time():

return datetime.datetime.now(tz=pytz.utc)

@pytest.fixture(scope="class")

def chat_join_request(time, bot):

return ChatJoinRequest(

chat=Chat(1, Chat.SUPERGROUP),

from_user=User(2, "first_name", False),

date=time,

bio="bio",

invite_link=ChatInviteLink(

"https://invite.link",

User(42, "creator", False),

creates_join_request=False,

name="InviteLink",

is_revoked=False,

is_primary=False,

),

bot=bot,

)

@pytest.fixture(scope="function")

def chat_join_request_update(bot, chat_join_request):

return Update(0, chat_join_request=chat_join_request)

class TestChatJoinRequestHandler:

test_flag = False

def test_slot_behaviour(self, mro_slots):

action = ChatJoinRequestHandler(self.callback)

for attr in action.__slots__:

assert getattr(action, attr, "err") != "err", f"got extra slot '{attr}'"

assert len(mro_slots(action)) == len(set(mro_slots(action))), "duplicate slot"

@pytest.fixture(autouse=True)

def reset(self):

self.test_flag = False

async def callback(self, update, context):

self.test_flag = (

isinstance(context, CallbackContext)

and isinstance(context.bot, Bot)

and isinstance(update, Update)

and isinstance(context.update_queue, asyncio.Queue)

and isinstance(context.job_queue, JobQueue)

and isinstance(context.user_data, dict)

and isinstance(context.chat_data, dict)

and isinstance(context.bot_data, dict)

and isinstance(

update.chat_join_request,

ChatJoinRequest,

)

)

def test_other_update_types(self, false_update):

handler = ChatJoinRequestHandler(self.callback)

assert not handler.check_update(false_update)

assert not handler.check_update(True)

async def test_context(self, app, chat_join_request_update):

handler = ChatJoinRequestHandler(callback=self.callback)

app.add_handler(handler)

async with app:

await app.process_update(chat_join_request_update)

assert self.test_flag