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

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

import pytest 

import sys 

import tempfile 

 

 

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): 

    import py 

18    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 py.builtin._istext(s): 

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

                config._pastebinfile.write(s) 

            tr._tw.write = tee_write 

 

def pytest_unconfigure(config): 

36    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 

79    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))