import pytest
from src.libs.common import to_base64
from src.steps.store import StepsStore


@pytest.mark.usefixtures("node_setup")
class TestSorting(StepsStore):
    @pytest.mark.parametrize("ascending", ["true", "false"])
    def test_store_sort_ascending(self, ascending):
        expected_message_hash_list = []
        for i in range(10):
            message = self.create_message(payload=to_base64(f"Message_{i}"))
            self.publish_message(message=message)
            expected_message_hash_list.append(self.compute_message_hash(self.test_pubsub_topic, message))
        for node in self.store_nodes:
            store_response = self.get_messages_from_store(node, page_size=5, ascending=ascending)
            response_message_hash_list = []
            for index in range(len(store_response.messages)):
                response_message_hash_list.append(store_response.message_hash(index))
            if ascending == "true":
                assert response_message_hash_list == expected_message_hash_list[:5], "Message hash mismatch for acending order"
            else:
                assert response_message_hash_list == expected_message_hash_list[5:], "Message hash mismatch for descending order"