Python badge Examples

Example #1

File: badge_test.py Project: 13625025773/h

        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)

Example #2

File: badge_test.py Project: chinmaygghag/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()

Example #3

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}

Example #4

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}

Example #5

File: badge_test.py Project: 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}

Example #6

File: badge_test.py Project: 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}

Example #7

File: badge_test.py Project: 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()

Example #8

File: badge_test.py Project: chinmaygghag/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}

Example #9

File: badge_test.py Project: chinmaygghag/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}

Example #10

File: badge_test.py Project: 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}

Example #11

File: badge_test.py Project: 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}

Example #12

File: badge_test.py Project: 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}

Example #13

File: badge_test.py Project: fuelpress/i.fuel.press

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}

Example #14

File: badge_test.py Project: hypothesis/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}

Example #15

File: badge_test.py Project: jbc22/h

def test_badge_raises_if_no_uri():
    with pytest.raises(httpexceptions.HTTPBadRequest):
        badge(mock.Mock(params={}))

Example #16

File: badge_test.py Project: hypothesis/h

def test_badge_raises_if_no_uri():
    with pytest.raises(httpexceptions.HTTPBadRequest):
        badge(mock.Mock(params={}))