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)