from robot.libraries.BuiltIn import BuiltIn


class LogLevels(object):
    ROBOT_LISTENER_API_VERSION = 2

    def __init__(self):
        self.ROBOT_LIBRARY_LISTENER = self
        self.messages = []

    def _log_message(self, msg):
        self.messages.append('%s: %s' % (msg['level'], msg['message']))

    def logged_messages_should_be(self, *expected):
        BuiltIn().should_be_equal('\n'.join(self.messages), '\n'.join(expected))
