"""Common secure document test constructs."""

import pytest

from payee.connectors.secure_data.document import SecureDocument
from payee.connectors.secure_data.fields.text_field import TextField


@pytest.fixture
def test_document_class():
    """Test document class."""

    class TestDocument(SecureDocument):
        """Test document class."""

        _document_type = 'test'
        _fields = {
            'id': TextField('id'),
            'name': TextField('name', obscure_in_response=True),
        }

    return TestDocument