samples/testsuite-support/tests/test_mocked_time.py

import dateutil
import pytest
import pytest_userver.client
 
import testsuite.plugins.mocked_time
import testsuite.utils.http
 
 
# /// [mocked_time]
@pytest.mark.now('2019-12-31T11:22:33Z')
async def test_now(service_client: pytest_userver.client.Client, mocked_time: testsuite.plugins.mocked_time.MockedTime):
    response: testsuite.utils.http.ClientResponse = await service_client.get('/now')
    assert response.status == 200
    assert 'application/json' in response.headers['Content-Type']
    assert response.json() == {'now': '2019-12-31T11:22:33+00:00'}
 
    # Change mocked time and sync state
    mocked_time.sleep(671)
    await service_client.update_server_state()
 
    response = await service_client.get('/now')
    assert response.status == 200
    assert 'application/json' in response.headers['Content-Type']
    assert response.json() == {'now': '2019-12-31T11:33:44+00:00'}
    # /// [mocked_time]
 
 
@pytest.mark.now('2019-12-31T10:00:00+0545')
async def test_now_timezone_custom(service_client, mocked_time):
    response = await service_client.get('/now')
    assert response.status == 200
    assert 'application/json' in response.headers['Content-Type']
    assert response.json() == {'now': '2019-12-31T04:15:00+00:00'}
 
    mocked_time.set(dateutil.parser.parse('2019-12-31T10:00:00+0400'))
 
    response = await service_client.get('/now')
    assert response.status == 200
    assert 'application/json' in response.headers['Content-Type']
    assert response.json() == {'now': '2019-12-31T06:00:00+00:00'}