"""Tests for BaseWorkerGroup implementation and usage."""

import pytest

from ray.train._internal.base_worker_group import BaseWorkerGroup
from ray.train._internal.worker_group import WorkerGroup as V1WorkerGroup
from ray.train.v2._internal.execution.worker_group.worker_group import (
    WorkerGroup as V2WorkerGroup,
)


def test_interface_abstract_methods():
    """Test that BaseWorkerGroup enforces its abstract methods."""
    # Should not be able to instantiate interface directly
    with pytest.raises(TypeError):
        BaseWorkerGroup()

    # Should not be able to create incomplete implementation
    class IncompleteWorkerGroup(BaseWorkerGroup):
        def execute(self, func, *args, **kwargs):
            pass

        # Missing other abstract methods

    with pytest.raises(TypeError):
        IncompleteWorkerGroup()


def test_real_implementations_inherit_interface():
    """Smoke test that real WorkerGroup implementations inherit from interface."""
    # Test inheritance
    assert issubclass(V1WorkerGroup, BaseWorkerGroup)
    assert issubclass(V2WorkerGroup, BaseWorkerGroup)

    # Test that all abstract methods are implemented
    # If any abstract methods are missing, __abstractmethods__ will be non-empty
    assert (
        len(V1WorkerGroup.__abstractmethods__) == 0
    ), f"V1 WorkerGroup missing abstract methods: {V1WorkerGroup.__abstractmethods__}"
    assert (
        len(V2WorkerGroup.__abstractmethods__) == 0
    ), f"V2 WorkerGroup missing abstract methods: {V2WorkerGroup.__abstractmethods__}"


if __name__ == "__main__":
    pytest.main([__file__])