from unittest.mock import Mock
from resolvers.stat import query_follows


def test_query_follows():
    user_id = 'user123'

    # Mocking database session and ORM models
    mock_session = Mock()
    mock_Author = Mock()
    mock_ShoutAuthor = Mock()
    mock_AuthorFollower = Mock()
    mock_Topic = Mock()
    mock_ShoutTopic = Mock()
    mock_TopicFollower = Mock()

    # Mocking expected query results
    expected_result = {
        'topics': [(1, 5, 10, 15), (2, 8, 12, 20)],  # Example topics query result
        'authors': [(101, 3, 6, 9), (102, 4, 7, 11)],  # Example authors query result
        'communities': [{'id': 1, 'name': 'Дискурс', 'slug': 'discours'}],
    }

    # Set up mocks to return expected results when queried
    mock_session.query().select_from().outerjoin().all.side_effect = [
        expected_result['authors'],  # Authors query result
        expected_result['topics'],  # Topics query result
    ]

    # Call the function to test
    result = query_follows(
        user_id,
        session=mock_session,
        Author=mock_Author,
        ShoutAuthor=mock_ShoutAuthor,
        AuthorFollower=mock_AuthorFollower,
        Topic=mock_Topic,
        ShoutTopic=mock_ShoutTopic,
        TopicFollower=mock_TopicFollower,
    )

    # Assertions
    assert result['topics'] == expected_result['topics']
    assert result['authors'] == expected_result['authors']
    assert result['communities'] == expected_result['communities']

    # Assert that mock session was called with expected queries
    expected_queries = [
        mock_session.query(
            mock_Author.id,
            mock_ShoutAuthor.author_id,
            mock_AuthorFollower.author_id,
            mock_AuthorFollower.follower_id,
        )
        .select_from(mock_Author)
        .outerjoin(mock_ShoutAuthor, mock_Author.id == mock_ShoutAuthor.author_id)
        .outerjoin(mock_AuthorFollower, mock_Author.id == mock_AuthorFollower.author_id)
        .outerjoin(
            mock_AuthorFollower, mock_Author.id == mock_AuthorFollower.follower_id
        )
        .all,
        mock_session.query(
            mock_Topic.id,
            mock_ShoutTopic.topic_id,
            mock_ShoutTopic.topic_id,
            mock_TopicFollower.topic_id,
        )
        .select_from(mock_Topic)
        .outerjoin(mock_ShoutTopic, mock_Topic.id == mock_ShoutTopic.topic_id)
        .outerjoin(mock_ShoutTopic, mock_Topic.id == mock_ShoutTopic.topic_id)
        .outerjoin(mock_TopicFollower, mock_Topic.id == mock_TopicFollower.topic_id)
        .all,
    ]
    mock_session.query.assert_has_calls(expected_queries)


# Run the test
test_query_follows()