Hide keyboard shortcuts

Hot-keys on this page

r m x p   toggle line displays

j k   next/prev highlighted chunk

0   (zero) top of page

1   (one) first highlighted chunk

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

55

56

57

58

59

60

61

62

63

64

65

66

67

68

69

70

71

72

73

74

75

76

77

78

79

80

81

82

83

84

85

86

87

88

89

90

91

92

93

94

95

96

97

98

99

100

101

102

103

104

105

106

107

108

109

110

111

112

113

""" submit failure or test session information to a pastebin service. """ 

from __future__ import absolute_import 

from __future__ import division 

from __future__ import print_function 

 

import sys 

import tempfile 

 

import six 

 

import pytest 

 

 

def pytest_addoption(parser): 

group = parser.getgroup("terminal reporting") 

group._addoption( 

"--pastebin", 

metavar="mode", 

action="store", 

dest="pastebin", 

default=None, 

choices=["failed", "all"], 

help="send failed|all info to bpaste.net pastebin service.", 

) 

 

 

@pytest.hookimpl(trylast=True) 

def pytest_configure(config): 

if config.option.pastebin == "all": 

tr = config.pluginmanager.getplugin("terminalreporter") 

# if no terminal reporter plugin is present, nothing we can do here; 

# this can happen when this function executes in a slave node 

# when using pytest-xdist, for example 

if tr is not None: 

# pastebin file will be utf-8 encoded binary file 

config._pastebinfile = tempfile.TemporaryFile("w+b") 

oldwrite = tr._tw.write 

 

def tee_write(s, **kwargs): 

oldwrite(s, **kwargs) 

if isinstance(s, six.text_type): 

s = s.encode("utf-8") 

config._pastebinfile.write(s) 

 

tr._tw.write = tee_write 

 

 

def pytest_unconfigure(config): 

if hasattr(config, "_pastebinfile"): 

# get terminal contents and delete file 

config._pastebinfile.seek(0) 

sessionlog = config._pastebinfile.read() 

config._pastebinfile.close() 

del config._pastebinfile 

# undo our patching in the terminal reporter 

tr = config.pluginmanager.getplugin("terminalreporter") 

del tr._tw.__dict__["write"] 

# write summary 

tr.write_sep("=", "Sending information to Paste Service") 

pastebinurl = create_new_paste(sessionlog) 

tr.write_line("pastebin session-log: %s\n" % pastebinurl) 

 

 

def create_new_paste(contents): 

""" 

Creates a new paste using bpaste.net service. 

 

:contents: paste contents as utf-8 encoded bytes 

:returns: url to the pasted contents 

""" 

import re 

 

if sys.version_info < (3, 0): 

from urllib import urlopen, urlencode 

else: 

from urllib.request import urlopen 

from urllib.parse import urlencode 

 

params = { 

"code": contents, 

"lexer": "python3" if sys.version_info[0] == 3 else "python", 

"expiry": "1week", 

} 

url = "https://bpaste.net" 

response = urlopen(url, data=urlencode(params).encode("ascii")).read() 

m = re.search(r'href="/raw/(\w+)"', response.decode("utf-8")) 

if m: 

return "%s/show/%s" % (url, m.group(1)) 

else: 

return "bad response: " + response 

 

 

def pytest_terminal_summary(terminalreporter): 

import _pytest.config 

 

if terminalreporter.config.option.pastebin != "failed": 

return 

tr = terminalreporter 

if "failed" in tr.stats: 

terminalreporter.write_sep("=", "Sending information to Paste Service") 

for rep in terminalreporter.stats.get("failed"): 

try: 

msg = rep.longrepr.reprtraceback.reprentries[-1].reprfileloc 

except AttributeError: 

msg = tr._getfailureheadline(rep) 

tw = _pytest.config.create_terminal_writer( 

terminalreporter.config, stringio=True 

) 

rep.toterminal(tw) 

s = tw.stringio.getvalue() 

assert len(s) 

pastebinurl = create_new_paste(s) 

tr.write_line("%s --> %s" % (msg, pastebinurl))