iframe-proxy | Sunbelt Computer Software

History

105 lines (83 loc) · 2.84 KB

Raw

100

101

102

103

104

105

import pytest

try:

from urllib.request import urlopen

except ImportError:

from urllib import urlopen

try:

from httplib import HTTPSConnection

except ImportError:

from http.client import HTTPSConnection

from sentry_sdk import capture_message

from sentry_sdk.integrations.stdlib import StdlibIntegration

def test_crumb_capture(sentry_init, capture_events):

sentry_init(integrations=[StdlibIntegration()])

events = capture_events()

url = "https://httpbin.org/status/200"

response = urlopen(url)

assert response.getcode() == 200

capture_message("Testing!")

event, = events

crumb, = event["breadcrumbs"]

assert crumb["type"] == "http"

assert crumb["category"] == "httplib"

assert crumb["data"] == {

"url": url,

"method": "GET",

"status_code": 200,

"reason": "OK",

}

def test_crumb_capture_hint(sentry_init, capture_events):

def before_breadcrumb(crumb, hint):

if "httplib_response" in hint:

con = hint["httplib_response"].getheader("Connection")

assert con.lower() == "close"

crumb["data"]["extra"] = "foo"

return crumb

sentry_init(integrations=[StdlibIntegration()], before_breadcrumb=before_breadcrumb)

events = capture_events()

url = "https://httpbin.org/status/200"

response = urlopen(url)

assert response.getcode() == 200

capture_message("Testing!")

event, = events

crumb, = event["breadcrumbs"]

assert crumb["type"] == "http"

assert crumb["category"] == "httplib"

assert crumb["data"] == {

"url": url,

"method": "GET",

"status_code": 200,

"reason": "OK",

"extra": "foo",

}

def test_httplib_misuse(sentry_init, capture_events):

"""HTTPConnection.getresponse must be called after every call to

HTTPConnection.request. However, if somebody does not abide by

this contract, we still should handle this gracefully and not

send mixed breadcrumbs.

Test whether our breadcrumbs are coherent when somebody uses HTTPConnection

wrongly.

"""

sentry_init()

events = capture_events()

conn = HTTPSConnection("httpbin.org", 443)

conn.request("GET", "/anything/foo")

with pytest.raises(Exception):

# This raises an exception, because we didn't call `getresponse` for

# the previous request yet.

# This call should not affect our breadcrumb.

conn.request("POST", "/anything/bar")

response = conn.getresponse()

assert response._method == "GET"

capture_message("Testing!")

event, = events

crumb, = event["breadcrumbs"]

assert crumb["type"] == "http"

assert crumb["category"] == "httplib"

assert crumb["data"] == {

"url": "https://httpbin.org/anything/foo",

"method": "GET",

"status_code": 200,

"reason": "OK",

}

Sunbelt Computer Software

PL/B Language Development and Support

Navigation Menu

Uh oh!

Sunbelt Computer Software

PL/B Language Development and Support

Uh oh!

FilesExpand file tree

test_httplib.py

Latest commit

History

test_httplib.py

File metadata and controls