def test_real(self):
if "TEST_BUILDBOTNET_USAGEDATA" not in os.environ:
raise SkipTest(
"_sendBuildbotNetUsageData real test only run when environment variable"
" TEST_BUILDBOTNET_USAGEDATA is set")
_sendBuildbotNetUsageData({'foo': 'bar'})
def test_real(self):
if "TEST_BUILDBOTNET_USAGEDATA" not in os.environ:
raise SkipTest(
"_sendBuildbotNetUsageData real test only run when environment variable"
" TEST_BUILDBOTNET_USAGEDATA is set")
_sendBuildbotNetUsageData({'foo': 'bar'})
def test_urllib(self):
self.patch(buildbot.buildbot_net_usage_data, '_sendWithRequests',
lambda _, __: None)
class FakeRequest:
def __init__(self, *args, **kwargs):
self.args = args
self.kwargs = kwargs
open_url = []
class urlopen:
def __init__(self, r):
self.request = r
open_url.append(self)
def read(self):
return "ok"
def close(self):
pass
self.patch(urllib_request, "Request", FakeRequest)
self.patch(urllib_request, "urlopen", urlopen)
_sendBuildbotNetUsageData({'foo': 'bar'})
self.assertEqual(len(open_url), 1)
self.assertEqual(open_url[0].request.args,
('https://events.buildbot.net/events/phone_home',
b'{"foo": "bar"}', {
'Content-Length': 14,
'Content-Type': 'application/json'
}))
def test_urllib(self):
self.patch(buildbot.buildbot_net_usage_data, '_sendWithRequests', lambda _, __: None)
class FakeRequest(object):
def __init__(self, *args, **kwargs):
self.args = args
self.kwargs = kwargs
open_url = []
class urlopen(object):
def __init__(self, r):
self.request = r
open_url.append(self)
def read(self):
return "ok"
def close(self):
pass
self.patch(urllib_request, "Request", FakeRequest)
self.patch(urllib_request, "urlopen", urlopen)
_sendBuildbotNetUsageData({'foo': 'bar'})
self.assertEqual(len(open_url), 1)
self.assertEqual(open_url[0].request.args,
('https://events.buildbot.net/events/phone_home',
b'{"foo": "bar"}',
{'Content-Length': 14, 'Content-Type': 'application/json'}))