Example #1
0
async def test_add_torrent_url(core):
    request = await create_mock_request(
        core=core,
        url='http://localhost/test.torrent')
    data = Path(torrent_dir, 'random_one_file.torrent').read_bytes()

    class AsyncContextManager:
        async def __aenter__(self):
            return self

        async def __aexit__(self, exc_type, exc, tb):
            return

        async def read(self):
            return data

    class ClientSession:
        async def __aenter__(self):
            return self

        async def __aexit__(self, exc_type, exc_val, exc_tb):
            return

        def get(self, url):
            return AsyncContextManager()

    with patch('aiohttp.ClientSession', new=ClientSession):
        body, response = await json_response(torrent.post_torrent(request))
        assert response.status == 201
Example #2
0
async def test_add_torrent_url():
    request = create_mock_request(url='http://localhost/test.torrent')
    data = open(
        os.path.join(torrent_dir, 'random_one_file.torrent'), 'rb').read()

    class AsyncContextManager:
        async def __aenter__(self):
            return self

        async def __aexit__(self, exc_type, exc, tb):
            return

        async def read(self):
            print('read')
            return data

    class ClientSession:
        def __enter__(self):
            return self

        def __exit__(self, exc_type, exc_val, exc_tb):
            return

        def get(self, url):
            return AsyncContextManager()

    with patch('aiohttp.ClientSession', new=ClientSession):
        body, response = await json_response(torrent.post_torrent(request))
        assert response.status == 201
Example #3
0
async def test_add_torrent_bad_number_args():
    request = create_mock_request(
        url='http://testing/test.torrent',
        info_hash='a0'*20)

    with assert_raises(aiohttp.errors.HttpProcessingError) as e:
        await json_response(torrent.post_torrent(request))
    assert e.exception.code == 400
Example #4
0
async def test_post_torrent_info_hash():
    request = create_mock_request(
        info_hash='44a040be6d74d8d290cd20128788864cbf770719')

    body, response = await json_response(torrent.post_torrent(request))
    assert 'info_hash' in body
    info_hash = body['info_hash']
    assert info_hash == '44a040be6d74d8d290cd20128788864cbf770719'
Example #5
0
async def test_add_torrent_bad_number_args(core):
    request = await create_mock_request(
        core=core,
        url='http://testing/test.torrent',
        info_hash='a0'*20)

    with pytest.raises(aiohttp.web.HTTPBadRequest):
        _, response = await json_response(torrent.post_torrent(request))
        assert response.status == 400
Example #6
0
async def test_add_torrent_lt_runtime_error():
    request = create_mock_request(filename='random_one_file.torrent')

    add_torrent = MagicMock()
    add_torrent.side_effect = RuntimeError()

    with patch('spritzle.core.core.session.add_torrent', add_torrent):
        with assert_raises(aiohttp.errors.HttpProcessingError) as e:
            await json_response(torrent.post_torrent(request))
        assert e.exception.code == 500
Example #7
0
async def test_add_torrent_bad_args(core):
    request = await create_mock_request(
        core=core,
        filename='random_one_file.torrent',
        args={'bad_key': True},
    )

    with pytest.raises(aiohttp.web.HTTPBadRequest):
        _, response = await json_response(torrent.post_torrent(request))
        assert response.status == 400
Example #8
0
async def test_add_torrent_lt_runtime_error(core):
    request = await create_mock_request(core=core,
                                        filename='random_one_file.torrent')

    add_torrent = MagicMock()
    add_torrent.side_effect = RuntimeError()
    request.app['spritzle.core'].session.add_torrent = add_torrent
    with pytest.raises(aiohttp.web.HTTPInternalServerError):
        _, response = await json_response(torrent.post_torrent(request))
        assert response.status == 500
Example #9
0
async def test_post_torrent_bad_body():
    request = create_mock_request(filename='random_one_file.torrent')
    async def json():
        return b'\xc3\x28'.decode("utf8")

    request.json = json
    body, response = await json_response(torrent.post_torrent(request))
    assert 'info_hash' in body

    info_hash = body['info_hash']

    assert info_hash == '44a040be6d74d8d290cd20128788864cbf770719'
Example #10
0
async def test_post_torrent():
    request = create_mock_request(filename='random_one_file.torrent')

    body, response = await json_response(torrent.post_torrent(request))
    assert 'info_hash' in body
    info_hash = body['info_hash']

    assert response.headers['LOCATION'] == \
        'http://localhost:8080/torrent/{}'.format(info_hash)
    assert response.status == 201

    assert info_hash == '44a040be6d74d8d290cd20128788864cbf770719'

    request = MagicMock()
    request.match_info = {}
    tlist, response = await json_response(torrent.get_torrent(request))
    assert tlist == ['44a040be6d74d8d290cd20128788864cbf770719']
Example #11
0
async def test_post_torrent(core):
    request = await create_mock_request(core=core,
                                        filename='random_one_file.torrent',
                                        tags=['foo'])

    body, response = await json_response(torrent.post_torrent(request))
    assert 'info_hash' in body
    info_hash = body['info_hash']

    assert response.headers['LOCATION'] == \
        f'http://localhost:8080/torrent/{info_hash}'
    assert response.status == 201

    assert info_hash == '44a040be6d74d8d290cd20128788864cbf770719'

    request.match_info = {}
    tlist, response = await json_response(torrent.get_torrent(request))
    assert tlist == ['44a040be6d74d8d290cd20128788864cbf770719']
    return request
Example #12
0
async def test_add_torrent_bad_file():
    request = create_mock_request('empty.torrent')

    with assert_raises(aiohttp.errors.HttpProcessingError) as e:
        await json_response(torrent.post_torrent(request))
    assert e.exception.code == 400
Example #13
0
async def test_add_torrent_bad_file(core):
    request = await create_mock_request(core=core, filename='empty.torrent')

    with pytest.raises(aiohttp.web.HTTPBadRequest):
        _, response = await json_response(torrent.post_torrent(request))
        assert response.status == 400