#! /usr/bin/python3

# SPDX-License-Identifier: GPL-2.0-or-later
#
# tasst - Test A Simple Socket Transport
#         library of test helpers for passt & pasta
#
# Copyright Red Hat
# Author: David Gibson <david@gibson.dropbear.id.au>

import avocado


class TasstSubData:
    def __init__(self, **kwargs):
        self.__dict__.update(kwargs)


# Base class for avocado-based passt/pasta tests
class Tasst(avocado.Test):
    # Fairly short default timeout
    timeout = 10.0

    def subsetup(self, class_, subdata):
        assert isinstance(self, class_)
        setattr(self, class_.__name__, subdata)

    def get_subsetup(self, class_):
        assert hasattr(self, class_.__name__), \
            "{}.setUp() needs to call {}.subsetup()".format(self.__class__.__name__,
                                                            class_.__name__)
        return getattr(self, class_.__name__)