5 releases
0.2.0 | Sep 20, 2024 |
---|---|
0.1.3 | Aug 1, 2024 |
0.1.2 | Jul 30, 2024 |
0.1.1 | Jul 29, 2024 |
0.1.0 | Jul 17, 2024 |
#211 in Testing
150 downloads per month
265KB
5.5K
SLoC
teloxide_tests
A crate that allows you to unit test your teloxide bots easily! No internet, accounts or anything required!
What this crate has
- Easy testing of handlers with access to raw bot requests (see hello_world_bot)
- Support of dependencies, changes of
me
and multiple updates (see album_bot) - Syntactic sugar and native support for storage, dialogue and states (see calculator_bot)
- Fake file getting and downloading (see file_download_bot)
- Ability to be used with databases (see phrase_bot)
Examples
Simplified [hello_world_bot
](https://github.com/LasterAlex/teloxide_tests/blob/master/examples/hello_world_bot/src/main.rs)
#[tokio::test]
async fn test_hello_world() {
let message = MockMessageText::new().text("Hi!");
let bot = MockBot::new(message, handler_tree());
// Sends the message as if it was from a user
bot.dispatch().await;
let responses = bot.get_responses();
let message = responses
.sent_messages
.last()
.expect("No sent messages were detected!");
assert_eq!(message.text(), Some("Hello World!"));
}
[file_download_bot
](https://github.com/LasterAlex/teloxide_tests/blob/master/examples/file_download_bot/src/main.rs)
#[tokio::test]
async fn test_not_a_document() {
let bot = MockBot::new(MockMessageText::new().text("Hi!"), handler_tree());
// Syntactic sugar
bot.dispatch_and_check_last_text("Not a document").await;
}
#[tokio::test]
async fn test_download_document_and_check() {
let bot = MockBot::new(MockMessageDocument::new(), handler_tree());
bot.dispatch_and_check_last_text("Downloaded!").await;
}
[calculator_bot
](https://github.com/LasterAlex/teloxide_tests/blob/master/examples/calculator_bot/src/tests.rs)
#[tokio::test]
async fn test_what_is_the_first_number() {
let bot = MockBot::new(MockCallbackQuery::new().data("add"), handler_tree());
bot.dependencies(deps![get_bot_storage().await]);
bot.set_state(State::WhatDoYouWant).await;
bot.dispatch_and_check_last_text_and_state(
text::ENTER_THE_FIRST_NUMBER,
State::GetFirstNumber {
operation: "add".to_owned(),
},
)
.await;
}
You can see more useful examples at examples/ and the docs at docs.rs
It is highly reccomended you read at least hello_world_bot
(there is a lot of comments that explain how to use this crate which i removed in the README) and calculator_bot
(it teaches about the syntactic sugar and working with dialogue)
How to implement it?
Hopefully it is as easy as doing what happens in ./examples
- Import the
teloxide_tests
- Make your handler tree into a separate function (we are going to test it, after all)
- Create a mocked bot with something that can be turned into an update, like MockMessageText or MockMessagePhoto
- Add dependencies and/or a different bot using .dependencies(deps![]) and .me(MockedMe::new().build())
- Dispatch it with .dispatch().await
- Get the responces with .get_responces()
- Do the testing with the gotten responces
- If you want to re-use the current bot and state with a new update, just call .update(MockMessageText::new()) and follow from the 5th step!
Do NOT use raw MockBot fields like bot.updates or bot.me to mutate the bot, unless you know what you are doing. Use given abstractions, and if some feature is missing, you can mention it in the github repo (or write it in the telegram group @teloxide_tests)
Pitfalls
-
Race conditions. They are, to my knoledge, the most difficult.
-
And also when you use a method that is still not supported by this crate. Please reffer to the docs to see, what endpoints are implemented in the latest release (or look at server/routes files to look at the current endpoints)
-
Maybe also the fact that the fake server actually checks the messages and files that are present, and it starts with a clean state. You can't just send a file by file_id or forward a message by an arbitrary message_id that was sent long ago, the bot wouldn't know what to do with it, so you need to separately add it by dispatching the bot with that update, so that it gets added as the user message to memory (you can change file_id and message_id in the mocked structs to anything you need).
Some errors associated with these race conditions:
-
trait
Send
is not implemented forstd::sync::MutexGuard<'static, ()>
This means you can't share the bot between any threads, as you should not in any circumstance.
-
PoisonError(...)
You shouldn't see it, i tried to mitigate it, but if you do, it's not the problem, it just means that something else panicked and now the bot doesn't know, what to do. Just fix whatever was causing the problem, and poison errors should be gone.
-
Stupid bugs that change every time you run a test
You can use the crate serial_test, or try to add
drop(bot);
at the end of every test, and do everything AFTER callingMockBot::new()
, as the bot creation makes a safe lock that prevent any race conditions.
Contributing
Please see CONTRIBUTING.md
Todo
- Add dataset
- Add dataset of chats
- Add dataset of common messages
- Add dataset of queries (low priority)
- Add dataset of messages (low priority)
- Add structs without a category (low priority)
- Add fake server
- Add most common endpoints
- Add all common messages
- Add inline queries (low priority)
- Add all queries (low priority)
- Add all messages (super low priority)
- Add everything else (may never be done)
- Make mocked bot that sends requests to fake server
- Add tests to that bot
- Make it into a library
- Publish it when it is ready
Special thanks to
The teloxide team! They made an absolutely incredible library with amazing internal documentation, which helped me a lot during development! It is an amazing project, and i'm happy i'm able to add to it something useful!
Dependencies
~28–44MB
~807K SLoC