"""Shared functions for tests."""

import random
import string


class IntegrationTestHelper:
    """Class for shared functions."""

    @staticmethod
    def random_string(suffix=''):
        """Generate random string with suffix."""
        return ''.join(random
                       .choice(string.ascii_lowercase) for _ in range(6)) \
               + suffix

    @staticmethod
    def get_last_partition_position(consumer):
        """Return partition and latest partition position."""
        consumer.poll(500)
        consumer.seek_to_end()
        topic_partition = consumer.assignment()
        partition = list(topic_partition)[0]
        return partition, consumer.position(partition)