# Copyright (c) Microsoft Corporation. All rights reserved.

# Licensed under the MIT License.

from collections import namedtuple

class TestPath(namedtuple('TestPath', 'root relfile func sub')):

"""Where to find a single test."""

def __new__(cls, root, relfile, func, sub=None):

self = super(TestPath, cls).__new__(

cls,

str(root) if root else None,

str(relfile) if relfile else None,

str(func) if func else None,

[str(s) for s in sub] if sub else None,

)

return self

def __init__(self, *args, **kwargs):

if self.root is None:

raise TypeError('missing id')

if self.relfile is None:

raise TypeError('missing kind')

# self.func may be None (e.g. for doctests).

# self.sub may be None.

class ParentInfo(namedtuple('ParentInfo', 'id kind name root parentid')):

KINDS = ('folder', 'file', 'suite', 'function', 'subtest')

def __new__(cls, id, kind, name, root=None, parentid=None):

self = super(ParentInfo, cls).__new__(

cls,

str(id) if id else None,

str(kind) if kind else None,

str(name) if name else None,

str(root) if root else None,

str(parentid) if parentid else None,

)

return self

def __init__(self, *args, **kwargs):

if self.id is None:

raise TypeError('missing id')

if self.kind is None:

raise TypeError('missing kind')

if self.kind not in self.KINDS:

raise ValueError('unsupported kind {!r}'.format(self.kind))

if self.name is None:

raise TypeError('missing name')

if self.root is None:

if self.parentid is not None or self.kind != 'folder':

raise TypeError('missing root')

elif self.parentid is None:

raise TypeError('missing parentid')

class TestInfo(namedtuple('TestInfo', 'id name path source markers parentid kind')):

"""Info for a single test."""

MARKERS = ('skip', 'skip-if', 'expected-failure')

KINDS = ('function', 'doctest')

def __new__(cls, id, name, path, source, markers, parentid, kind='function'):

self = super(TestInfo, cls).__new__(

cls,

str(id) if id else None,

str(name) if name else None,

path or None,

str(source) if source else None,

[str(marker) for marker in markers or ()],

str(parentid) if parentid else None,

str(kind) if kind else None,

)

return self

def __init__(self, *args, **kwargs):

if self.id is None:

raise TypeError('missing id')

if self.name is None:

raise TypeError('missing name')

if self.path is None:

raise TypeError('missing path')

if self.source is None:

raise TypeError('missing source')

else:

srcfile, _, lineno = self.source.rpartition(':')

if not srcfile or not lineno or int(lineno) < 0:

raise ValueError('bad source {!r}'.format(self.source))

if self.markers:

badmarkers = [m for m in self.markers if m not in self.MARKERS]

if badmarkers:

raise ValueError('unsupported markers {!r}'.format(badmarkers))

if self.parentid is None:

raise TypeError('missing parentid')

if self.kind is None:

raise TypeError('missing kind')

elif self.kind not in self.KINDS:

raise ValueError('unsupported kind {!r}'.format(self.kind))

@property

def root(self):

return self.path.root

@property

def srcfile(self):

return self.source.rpartition(':')[0]

@property

def lineno(self):

return int(self.source.rpartition(':')[-1])