示例#1
0
        def caller(uri, annotated=True, blocked=False):
            if annotated:
                factories.DocumentURI(uri=uri)
                pyramid_request.db.flush()

            Blocklist.is_blocked.return_value = blocked

            pyramid_request.params["uri"] = uri
            return badge(pyramid_request)
示例#2
0
def test_badge_does_not_search_if_uri_never_annotated(models, pyramid_request, search_run):
    pyramid_request.params['uri'] = 'http://example.com'
    models.Blocklist.is_blocked.return_value = False

    result = badge(pyramid_request)

    assert result == {'total': 0}
    models.Blocklist.is_blocked.assert_not_called()
    search_run.assert_not_called()
示例#3
0
def test_badge_returns_0_if_blocked(models, search_run):
    request = mock.Mock(params={'uri': 'test_uri'})
    models.Blocklist.is_blocked.return_value = True
    search_run.return_value = {'total': 29}

    result = badge(request)

    assert not search_run.called
    assert result == {'total': 0}
示例#4
0
def test_badge_returns_number_from_search(models, search_run):
    request = mock.Mock(params={'uri': 'test_uri'})
    models.Blocklist.is_blocked.return_value = False
    search_run.return_value = mock.Mock(total=29)

    result = badge(request)

    search_run.assert_called_once_with({'uri': 'test_uri', 'limit': 0})
    assert result == {'total': 29}
示例#5
0
文件: badge_test.py 项目: gnott/h
def test_badge_returns_0_if_blocked(models, search_run):
    request = mock.Mock(params={'uri': 'test_uri'})
    models.Blocklist.is_blocked.return_value = True
    search_run.return_value = {'total': 29}

    result = badge(request)

    assert not search_run.called
    assert result == {'total': 0}
示例#6
0
文件: badge_test.py 项目: gnott/h
def test_badge_returns_number_from_search(models, search_run):
    request = mock.Mock(params={'uri': 'test_uri'})
    models.Blocklist.is_blocked.return_value = False
    search_run.return_value = mock.Mock(total=29)

    result = badge(request)

    search_run.assert_called_once_with({'uri': 'test_uri', 'limit': 0})
    assert result == {'total': 29}
示例#7
0
文件: badge_test.py 项目: jbc22/h
def test_badge_does_not_search_if_uri_never_annotated(models, pyramid_request,
                                                      search_run):
    pyramid_request.params["uri"] = "http://example.com"
    models.Blocklist.is_blocked.return_value = False

    result = badge(pyramid_request)

    assert result == {"total": 0}
    models.Blocklist.is_blocked.assert_not_called()
    search_run.assert_not_called()
示例#8
0
def test_badge_returns_number_from_search(models, pyramid_request, search_run, mark_uri_as_annotated):
    mark_uri_as_annotated('http://example.com')

    pyramid_request.params['uri'] = 'http://example.com'
    models.Blocklist.is_blocked.return_value = False
    search_run.return_value = mock.Mock(total=29)

    result = badge(pyramid_request)

    search_run.assert_called_once_with({'uri': 'http://example.com', 'limit': 0})
    assert result == {'total': 29}
示例#9
0
def test_badge_returns_0_if_blocked(models, pyramid_request, search_run, mark_uri_as_annotated):
    mark_uri_as_annotated('http://blocked-domain.com')

    pyramid_request.params['uri'] = 'http://blocked-domain.com'
    models.Blocklist.is_blocked.return_value = True
    search_run.return_value = {'total': 29}

    result = badge(pyramid_request)

    models.Blocklist.is_blocked.assert_called_with(mock.ANY, 'http://blocked-domain.com')
    assert not search_run.called
    assert result == {'total': 0}
示例#10
0
文件: badge_test.py 项目: jbc22/h
def test_badge_returns_0_if_blocked(models, pyramid_request, search_run,
                                    mark_uri_as_annotated):
    mark_uri_as_annotated("http://blocked-domain.com")

    pyramid_request.params["uri"] = "http://blocked-domain.com"
    models.Blocklist.is_blocked.return_value = True
    search_run.return_value = {"total": 29}

    result = badge(pyramid_request)

    models.Blocklist.is_blocked.assert_called_with(
        mock.ANY, "http://blocked-domain.com")
    assert not search_run.called
    assert result == {"total": 0}
示例#11
0
文件: badge_test.py 项目: yumatch/h
def test_badge_returns_0_if_blocked(models, pyramid_request, search_run,
                                    mark_uri_as_annotated):
    mark_uri_as_annotated('http://blocked-domain.com')

    pyramid_request.params['uri'] = 'http://blocked-domain.com'
    models.Blocklist.is_blocked.return_value = True
    search_run.return_value = {'total': 29}

    result = badge(pyramid_request)

    models.Blocklist.is_blocked.assert_called_with(
        mock.ANY, 'http://blocked-domain.com')
    assert not search_run.called
    assert result == {'total': 0}
示例#12
0
文件: badge_test.py 项目: yumatch/h
def test_badge_returns_number_from_search(models, pyramid_request, search_run,
                                          mark_uri_as_annotated):
    mark_uri_as_annotated('http://example.com')

    pyramid_request.params['uri'] = 'http://example.com'
    models.Blocklist.is_blocked.return_value = False
    search_run.return_value = mock.Mock(total=29)

    result = badge(pyramid_request)

    search_run.assert_called_once_with({
        'uri': 'http://example.com',
        'limit': 0
    })
    assert result == {'total': 29}
示例#13
0
def test_badge_returns_number_from_search(
    models, pyramid_request, search_run, mark_uri_as_annotated
):
    mark_uri_as_annotated("http://example.com")

    pyramid_request.params["uri"] = "http://example.com"
    models.Blocklist.is_blocked.return_value = False
    search_run.return_value = mock.Mock(total=29)

    result = badge(pyramid_request)

    search_run.assert_called_once_with(
        MultiDict({"uri": "http://example.com", "limit": 0})
    )
    assert result == {"total": 29}
示例#14
0
def test_badge_returns_0_if_blocked(
    models, pyramid_request, search_run, mark_uri_as_annotated
):
    mark_uri_as_annotated("http://blocked-domain.com")

    pyramid_request.params["uri"] = "http://blocked-domain.com"
    models.Blocklist.is_blocked.return_value = True
    search_run.return_value = {"total": 29}

    result = badge(pyramid_request)

    models.Blocklist.is_blocked.assert_called_with(
        mock.ANY, "http://blocked-domain.com"
    )
    assert not search_run.called
    assert result == {"total": 0}
示例#15
0
文件: badge_test.py 项目: jbc22/h
def test_badge_raises_if_no_uri():
    with pytest.raises(httpexceptions.HTTPBadRequest):
        badge(mock.Mock(params={}))
示例#16
0
def test_badge_raises_if_no_uri():
    with pytest.raises(httpexceptions.HTTPBadRequest):
        badge(mock.Mock(params={}))