xn.py.test

#!/usr/bin/env python3
# coding: utf-8

# Copyright (c) 2018 Trevor Taylor
# 
# Permission to use, copy, modify, and/or distribute this software for
# any purpose with or without fee is hereby granted, provided that all
# copyright notices and this permission notice appear in all copies.
# 
# THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
# WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
# MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
# ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
# WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
# ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
# OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
#
from sys import path
from os.path import dirname
if path[0]==dirname(__file__): path.pop(0)
from xju.xn import in_context,readable_repr,AllFailed,in_function_context,Xn,capitalise,first_para_of, first_line_of
from xju.assert_ import Assert
from typing import cast

import io
import sys

from xju.assert_ import Assert

def test1(prog=sys.argv[0]):
    def a():
        raise Exception('fred')
    def b():
        a()
        pass
    def c():
        b()
        pass
    try:
        c()
    except:
        e=in_context('jock')
        Assert(readable_repr(e))=='''\
Failed to jock because
fred.\
'''
        Assert(str(e)).matches('''\
failed to jock because
[^:]*:40: failed to c[(][)] because
[^:]*:37: failed to b[(][)] because
[^:]*:34: failed to a[(][)] because
[^:]*:32: fred'''.format(**vars()))
        pass
    pass

class E(Exception):
    def __str__(self):
        return 'deadline reached'
    pass

def test2():
    try:
        try:
            raise E()
        except:
            raise in_context('await message') from None
        pass
    except E as e:
        Assert(readable_repr(e))=='''\
Failed to await message because
deadline reached.'''
        pass
    pass

def test3():
    try:
        try:
            try:
                raise E()
            except:
                raise in_context('await message') from None
            pass
        except E as e:
            try:
                raise Exception('open failed')
            except:
                raise AllFailed([e,in_context('read default file')]) from None
            pass
        pass
    except:
        Assert(readable_repr(in_context('get content')))=='''\
Failed to get content because
- failed to await message because
  deadline reached; and
- failed to read default file because
  open failed.'''
        pass
    pass

from inspect import currentframe, getframeinfo

class _Scope():
    def __init__(self,s):
        self.s=s
        cf=currentframe()
        assert cf is not None
        fb=cf.f_back
        assert fb is not None
        frameinfo = getframeinfo(fb)
        self.fl=(frameinfo.filename, frameinfo.lineno)
        pass
    def __enter__(self):
        pass
    def __exit__(self,t,e,b):
        if e:
            raise in_context(self.s,(t,e,b),self.fl) from None
        pass

def test4(prog=sys.argv[0]):
    def a():
        raise Exception('fred')
    def b():
        with _Scope('b'):
            True
            a()
            pass
        pass
    def c():
        b()
        pass
    try:
        c()
    except:
        e=in_context('jock')
    else:
        assert False, 'c should have raised?'
        pass
    Assert(str(e)).matches('''\
failed to jock because
[^:]*:132: failed to c[(][)] because
[^:]*:129: failed to b[(][)] because
[^:]*:123: failed to with _Scope[(]'b'[)]: because
[^:]*:116: failed to raise in_context[(]self.s,[(]t,e,b[)],self.fl[)] from None because
[^:]*:123: failed to b because
[^:]*:125: failed to a[(][)] because
[^:]*:121: fred''')
    Assert(readable_repr(e))=='''\
Failed to jock because
failed to b because
fred.\
'''
    pass

def test6():
    class X:
        def jump(self, height:int):
            '''
            jump {height}m in the air

            - from a standing start
            '''
            try:
                raise Exception('I have lead feet')
            except Exception as e:
                raise in_function_context(X.jump,vars()) from None
            pass
        pass
    try:
        X().jump(10)
    except Exception as e:
        Assert(readable_repr(e))=='Failed to jump 10m in the air because\nI have lead feet.'
    else:
        assert False, 'should not be here'
        pass
    pass

def test8():
    Assert(capitalise('Fred'))=='Fred'
    pass

def test9():
    Assert(str(AllFailed([Exception('fred'),Exception('jock')])))=="fred, and\njock"
    pass

def test10():
    # sometimes exception attributes cannot be copied due to AttributeError
    class EA(Exception):
        def __dir__(self):
            return ['x']+list(Exception.__dir__(self))
        pass
    try:
        try:
            raise EA()
        except EA:
            raise in_context('fred')
            pass
        pass
    except EA as y:
        pass
    pass

def test11():
    Assert(first_para_of("""
  the blind
  mice
  ran 
"""))=="the blind mice ran"

    Assert(first_para_of("""
  the blind

  mice
  ran 
"""))=="the blind"

    Assert(first_para_of(""""""))==""
    Assert(first_para_of(""" fred """))=="fred"

def test12():
    try:
        try:
            raise TimeoutError
        except Exception as e:
            raise in_context('fred') from None
        pass
    except Exception as e:
        Assert(readable_repr(e))=="Failed to fred because\nTimeoutError."
        Assert(str(e)).matches('''\
failed to fred because
[^:]*:222: TimeoutError''')
        pass
    pass

def test13():
    Assert(first_line_of('fred\njock'))=='fred'
    Assert(first_line_of(''))==''
    Assert(first_line_of('fred'))=='fred'

        
tests=[var for name,var in sorted(vars().items())
       if name.startswith('test') and callable(var)]
for t in tests: t()
n=len(tests)
print('PASS - {n} steps'.format(**vars()))
pass