fmt+refactor

tests/query_follows_test.py

@@ -0,0 +1,78 @@
+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()