def create_otp(self): collection = mongomock.MongoClient().db.collection user = MockUser() mock_user = user.mock_user() otp_obj = OTP(mock_user) otp_obj.collection = collection return otp_obj
def create_none_session(self): collection = mongomock.MongoClient().db.collection user = MockUser() mock_none_user = user.mock_none_user() session_obj_none = Session(mock_none_user) session_obj_none.collection = collection return session_obj_none
def _create_replied_message_mock(text): user_denisse = MockUser(id=0, first_name="Denisse") user_ken = MockUser(id=1, first_name="Ken") message = MockMessage(text=text, from_user=user_denisse) replied_message = MockMessage(reply_to_message=message, text="-save", from_user=user_ken) return replied_message
def test_it_sends_help(self): expected_chat_id = 123 expected_message_regex = ( r"Hola, soy Giru.\s+" r"\*Comandos:\*\s+" r"([/\w-]+: [\w\s\?]+?\n" r"\s+- [\w/\?]+\: [\w/\?-]+?\n)+" ) self.bot.send_message = MagicMock() user = MockUser(id=expected_chat_id) update = self.create_mock_update_with_chat_id_message_and_user( chat_id=expected_chat_id, from_user=user ) from giru.core.commands import create_ayuda_cb ayuda_cb = create_ayuda_cb(mock_commands, ayuda_text) ayuda_cb(self.bot, update) kwargs = self.bot.send_message.call_args[1] self.assertEqual(expected_chat_id, kwargs["chat_id"]) self.assertEqual("Markdown", kwargs["parse_mode"]) self.assertRegex(kwargs["text"], expected_message_regex)
def test_exists(): user = MockUser().mock_user() store = Store(user) store.rootpath = os.path.expanduser(tempdir + "/.autodl/") store.path = posixpath.join(store.rootpath, store.user.get("username")) project_path = store.create(project="test_project_dir") project_path_exists = store.exist(project="test_project_dir") assert project_path_exists == True project_path_none = store.exist() assert project_path_none == True
def test_create(): user = MockUser().mock_user() store = Store(user) store.rootpath = os.path.expanduser(tempdir + "/.autodl/") store.path = posixpath.join(store.rootpath, store.user.get("username")) test_path = store.path + "/test_project_dir" project_path = store.create(project="test_project_dir") assert test_path == project_path path_none = store.create() assert store.path == path_none
def test_delete(): user = MockUser().mock_user() store = Store(user) store.rootpath = os.path.expanduser(tempdir + "/.autodl/") store.path = posixpath.join(store.rootpath, store.user.get("username")) project_path = store.create(project="test_project_dir") state, result = store.delete(project="test_project_dir") assert os.path.exists(project_path) == False shared_with_user = MockUser().mock_user() shared_with = shared_with_user.get("username") test_path = store.rootpath + shared_with + "/shared_test_project.json" file = "shared_test_project.json" with open(posixpath.join(store.path, file), "w") as fp: pass state, result = store.delete(project="shared_test_project.json") assert os.path.exists(test_path) == False project_path_none = store.create(project=None) state, result = store.delete(project=None) assert os.path.exists(project_path_none) == False
def test_drunk_reply_is_sent_if_filter_is_matched(self): matcher = AlcoholRelatedFilter() user = MockUser(first_name='Giru', is_bot=True) message_1 = MockMessage(text='denisse quiere romo', from_user=user) self.assertTrue(matcher.filter(message_1)) message_2 = MockMessage(text='pero no hay cuarto', from_user=user) self.assertFalse(matcher.filter(message_2)) message_3 = MockMessage(text='TRAIGANLE ROMO A DENISSE', from_user=user) self.assertTrue(matcher.filter(message_3))
def test_drunk_reply_is_sent_if_filter_is_matched(self): matcher = alcohol_replier user = MockUser(first_name="Giru", is_bot=True) message_1 = MockMessage(text="denisse quiere romo", from_user=user) self.assertTrue(matcher.filter(message_1)) message_2 = MockMessage(text="pero no hay cuarto", from_user=user) self.assertFalse(matcher.filter(message_2)) message_3 = MockMessage(text="TRAIGANLE ROMO A DENISSE", from_user=user) self.assertTrue(matcher.filter(message_3))
def test_enlist(): user = MockUser().mock_user() store = Store(user) store.rootpath = os.path.expanduser(tempdir + "/.autodl/") store.path = posixpath.join(store.rootpath, store.user.get("username")) project_path_1 = store.create(project="test_project_dir_0") project_path_2 = store.create(project="test_project_dir_1") project_path_3 = store.create(project="test_project_dir_2") test_projects = store.enlist() if ( "test_project_dir_0" in test_projects and "test_project_dir_1" in test_projects and "test_project_dir_2" in test_projects ): assert True
import os import sys import pytest from datetime import datetime sys.path.append("..") from authv1.auth import Token, DATE_FORMAT from tests.mocks import MockOS, MockUser mock_os_obj = MockOS() user = MockUser() setup = user.mock_user() def test_create(mocker): mock_os_obj.mock_jwt_secret(mocker) setup_wrong = user.mock_none_user() token = Token(setup).create() assert token is not None token = Token(setup_wrong).create() assert token is None def test_delete(mocker): mock_os_obj.mock_jwt_secret(mocker) token = Token(setup)