#! /usr/bin/env python3

# SPDX-License-Identifier: GPL-2.0-or-later
#
# Copyright Red Hat
# Author: David Gibson <david@gibson.dropbear.id.au>

"""
Test writer facing interface to avocodo-classless
"""

from collections import Counter

from .manifest import manifest_add


def classless_test(**kwargs):
    """Mark a function as a classless test with options"""
    def dec(func):
        manifest_add(func.__module__, func.__name__, func, **kwargs)
        return func
    return dec


# Simple version of classless_test with default options
test = classless_test()


def test_output(*tests):
    """Decorate a function to test its output with a list of other functions"""
    def dec(func):
        for testfn in tests:
            manifest_add(func.__module__,
                         f"{testfn.__name__}∘{func.__name__}",
                         lambda testfn=testfn: testfn(func()))

        return func
    return dec


#
# Assertion helpers without unnecessary OOP nonsense
#


def assert_eq(left, right):
    """assert_eq(left, right)

    If left != right, fail with message showing both values
    """
    assert (left == right), f'{left} != {right}'


def assert_in(member, group):
    """assert_in(member, group)

    If member not in group, fail with a message showing both values
    """
    assert member in group, f'{member} not in {group}'


def assert_eq_unordered(left, right):
    """assert_eq_unordered(left, right)

    Fails if sequences left and right are different, ignoring order.
    """
    assert Counter(left) == Counter(right), \
        f'{left} != {right} (ignoring order)'


def assert_raises(exc, func, *args, **kwargs):
    """assert_raises(exc, func, *args, **kwargs)

    If func(*args, **kwargs) does not raise exc, fail
    """
    try:
        func(*args, **kwargs)
        raise AssertionError(f'Expected {exc.__name__} exception')
    except exc:
        pass