mock_api.lua
No OneTemporary
Actions

Authored By

Unknown

Size

7 KB

Referenced Files

None

Subscribers

None

mock_api.lua
View Options

	--[[
	mattata v2.0 - Mock Telegram Bot API
	Records all calls and returns configurable responses for testing.
	]]

	local mock_api = {}

	function mock_api.new()
	local api = {
	info = { id = 123456789, username = 'testbot', first_name = 'Test Bot' },
	calls = {},
	}

	local custom_handlers = {}

	local function record(method, ...)
	table.insert(api.calls, { method = method, args = {...} })
	end

	function api.send_message(chat_id, text, parse_mode, ...)
	record('send_message', chat_id, text, parse_mode, ...)
	return { ok = true, result = { message_id = #api.calls, chat = { id = chat_id } } }
	end

	function api.get_chat_member(chat_id, user_id)
	record('get_chat_member', chat_id, user_id)
	if custom_handlers.get_chat_member then
	return custom_handlers.get_chat_member(chat_id, user_id)
	end
	-- Default: regular member
	return { ok = true, result = { status = 'member', user = { id = user_id } } }
	end

	function api.ban_chat_member(chat_id, user_id, until_date)
	record('ban_chat_member', chat_id, user_id, until_date)
	return { ok = true, result = true }
	end

	function api.unban_chat_member(chat_id, user_id)
	record('unban_chat_member', chat_id, user_id)
	return { ok = true, result = true }
	end

	function api.restrict_chat_member(chat_id, user_id, perms_or_until, maybe_perms)
	record('restrict_chat_member', chat_id, user_id, perms_or_until, maybe_perms)
	return { ok = true, result = true }
	end

	function api.delete_message(chat_id, message_id)
	record('delete_message', chat_id, message_id)
	return { ok = true, result = true }
	end

	function api.pin_chat_message(chat_id, message_id, disable_notification)
	record('pin_chat_message', chat_id, message_id, disable_notification)
	return { ok = true, result = true }
	end

	function api.unpin_chat_message(chat_id, message_id)
	record('unpin_chat_message', chat_id, message_id)
	return { ok = true, result = true }
	end

	function api.get_chat(chat_id)
	record('get_chat', chat_id)
	return { ok = true, result = { id = chat_id, first_name = 'Test User' } }
	end

	function api.edit_message_text(chat_id, message_id, text, parse_mode, ...)
	record('edit_message_text', chat_id, message_id, text, parse_mode, ...)
	return { ok = true, result = { message_id = message_id } }
	end

	function api.edit_message_reply_markup(chat_id, message_id, inline_message_id, keyboard)
	record('edit_message_reply_markup', chat_id, message_id, inline_message_id, keyboard)
	return { ok = true, result = { message_id = message_id } }
	end

	function api.answer_callback_query(callback_id, text)
	record('answer_callback_query', callback_id, text)
	return { ok = true }
	end

	function api.get_updates(timeout, offset, limit, allowed)
	record('get_updates', timeout, offset, limit, allowed)
	return { ok = true, result = {} }
	end

	function api.leave_chat(chat_id)
	record('leave_chat', chat_id)
	return { ok = true, result = true }
	end

	function api.inline_keyboard()
	local kb = {}
	function kb:row(...)
	return self
	end
	return kb
	end

	function api.row()
	local r = {}
	function r:callback_data_button(text, data)
	return self
	end
	function r:url_button(text, url)
	return self
	end
	return r
	end

	-- Helper to set custom get_chat_member behavior
	function api.set_admin(chat_id, user_id)
	local original_handler = custom_handlers.get_chat_member
	custom_handlers.get_chat_member = function(cid, uid)
	if cid == chat_id and uid == user_id then
	return {
	ok = true,
	result = {
	status = 'administrator',
	user = { id = uid },
	can_restrict_members = true,
	can_delete_messages = true,
	can_pin_messages = true,
	can_promote_members = true,
	can_invite_users = true,
	}
	}
	end
	if original_handler then
	return original_handler(cid, uid)
	end
	return { ok = true, result = { status = 'member', user = { id = uid } } }
	end
	end

	-- Helper to set the bot as an admin with specified permissions
	function api.set_bot_admin(chat_id, perms)
	perms = perms or {}
	local original_handler = custom_handlers.get_chat_member
	custom_handlers.get_chat_member = function(cid, uid)
	if cid == chat_id and uid == api.info.id then
	return {
	ok = true,
	result = {
	status = 'administrator',
	user = { id = uid },
	can_restrict_members = perms.can_restrict_members or false,
	can_delete_messages = perms.can_delete_messages or false,
	can_pin_messages = perms.can_pin_messages or false,
	can_promote_members = perms.can_promote_members or false,
	can_invite_users = perms.can_invite_users or false,
	}
	}
	end
	if original_handler then
	return original_handler(cid, uid)
	end
	return { ok = true, result = { status = 'member', user = { id = uid } } }
	end
	end

	function api.set_creator(chat_id, user_id)
	local original_handler = custom_handlers.get_chat_member
	custom_handlers.get_chat_member = function(cid, uid)
	if cid == chat_id and uid == user_id then
	return {
	ok = true,
	result = {
	status = 'creator',
	user = { id = uid },
	}
	}
	end
	if original_handler then
	return original_handler(cid, uid)
	end
	return { ok = true, result = { status = 'member', user = { id = uid } } }
	end
	end

	function api.reset()
	api.calls = {}
	custom_handlers = {}
	end

	function api.get_call(method)
	for _, call in ipairs(api.calls) do
	if call.method == method then return call end
	end
	return nil
	end

	function api.get_calls(method)
	local results = {}
	for _, call in ipairs(api.calls) do
	if call.method == method then
	table.insert(results, call)
	end
	end
	return results
	end

	function api.count_calls(method)
	local count = 0
	for _, call in ipairs(api.calls) do
	if call.method == method then count = count + 1 end
	end
	return count
	end

	return api
	end

	return mock_api

File Metadata

Mime Type: text/plain
Expires: Sat, Nov 14, 7:05 PM (19 h, 19 m)
Storage Engine: blob
Storage Format: Raw Data
Storage Handle: 133946
Default Alt Text: mock_api.lua (7 KB)

mock_api.luaNo OneTemporaryActions

mock_api.luaView Options

File Metadata

Event Timeline

mock_api.lua
No OneTemporary
Actions

mock_api.lua
View Options