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_session(self): collection = mongomock.MongoClient().db.collection user = MockUser() mock_user = user.mock_user() session_obj = Session(mock_user) session_obj.collection = collection return session_obj
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)