"""Test protocol util."""

import pytest

from grass.protocols import rest, utils


def test_validate_invalid_protocol():
    """Test if an invalid protocol is valid."""
    with pytest.raises(AttributeError) as exc:

        class FakeProtocol:
            def delete(self):
                pass

            def get(self):
                pass

        utils.validate_protocol(FakeProtocol)

    assert str(exc.value).find("has no attribute 'head'") > -1


def test_validate_protocol():
    """Test a valid protocol like Rest."""
    utils.validate_protocol(rest)