# SPDX-License-Identifier: (GPL-2.0 OR Linux-OpenIB)
# Copyright (c) 2019 Mellanox Technologies, Inc. All rights reserved. See COPYING file
"""
Test module for pyverbs' pd module.
"""
import random

from tests.base import PyverbsAPITestCase
from pyverbs.base import PyverbsRDMAErrno
import pyverbs.device as d
from pyverbs.pd import PD


class PDTest(PyverbsAPITestCase):
    """
    Test various functionalities of the PD class.
    """
    def test_alloc_pd(self):
        """
        Test ibv_alloc_pd()
        """
        for ctx, attr, attr_ex in self.devices:
            with PD(ctx):
                pass

    def test_dealloc_pd(self):
        """
        Test ibv_dealloc_pd()
        """
        for ctx, attr, attr_ex in self.devices:
            with PD(ctx) as pd:
                pd.close()

    def test_multiple_pd_creation(self):
        """
        Test multiple creations and destructions of a PD object
        """
        for ctx, attr, attr_ex in self.devices:
            for i in range(random.randint(1, 200)):
                with PD(ctx) as pd:
                    pd.close()

    @staticmethod
    def test_create_pd_none_ctx():
        """
        Verify that PD can't be created with a None context
        """
        try:
            PD(None)
        except TypeError as te:
            assert 'must not be None' in te.args[0]
        else:
            raise PyverbsRDMAErrno('Created a PD with None context')

    def test_destroy_pd_twice(self):
        """
        Test bad flow cases in destruction of a PD object
        """
        for ctx, attr, attr_ex in self.devices:
            with PD(ctx) as pd:
                # Pyverbs supports multiple destruction of objects, we are
                # not expecting an exception here.
                pd.close()
                pd.close()