From dfac48ab45e768e50ba50e2647e3e8c9f94e4137 Mon Sep 17 00:00:00 2001 From: Egor Date: Mon, 29 Jun 2020 00:15:14 +0500 Subject: [PATCH] feat: add test for deleted file --- aiogram/contrib/fsm_storage/mongo.py | 2 +- tests/contrib/fsm_storage/test_aiomongo.py | 15 +++++++++++++++ 2 files changed, 16 insertions(+), 1 deletion(-) create mode 100644 tests/contrib/fsm_storage/test_aiomongo.py diff --git a/aiogram/contrib/fsm_storage/mongo.py b/aiogram/contrib/fsm_storage/mongo.py index d7501d68..8d2a8cea 100644 --- a/aiogram/contrib/fsm_storage/mongo.py +++ b/aiogram/contrib/fsm_storage/mongo.py @@ -1 +1 @@ -from .mongo_aiomongo import MongoStorage \ No newline at end of file +from .mongo_aiomongo import MongoStorage diff --git a/tests/contrib/fsm_storage/test_aiomongo.py b/tests/contrib/fsm_storage/test_aiomongo.py new file mode 100644 index 00000000..4940711d --- /dev/null +++ b/tests/contrib/fsm_storage/test_aiomongo.py @@ -0,0 +1,15 @@ +import importlib + +import aiogram + + +def test_file_deleted(): + try: + major, minor, _ = aiogram.__version__.split(".") + except ValueError: # raised if version is major.minor + major, minor = aiogram.__version__.split(".") + if major == "2" and int(minor) >= 11: + mongo_aiomongo = importlib.util.find_spec("aiogram.contrib.fsm_storage.mongo_aiomongo") + assert mongo_aiomongo is False, "Remove aiogram.contrib.fsm_storage.mongo_aiomongo file, and replace storage " \ + "in aiogram.contrib.fsm_storage.mongo with storage " \ + "from aiogram.contrib.fsm_storage.mongo_motor"