# -*- coding: utf-8 -*-
"""
Tests for the rabbitmq state
"""
# Import python libs
from __future__ import absolute_import, print_function, unicode_literals

# Import Salt Testing libs
from tests.support.case import ModuleCase
from tests.support.helpers import skip_if_not_root
from tests.support.mixins import SaltReturnAssertsMixin


@skip_if_not_root
class RabbitUserTestCase(ModuleCase, SaltReturnAssertsMixin):
    """
    Validate the rabbitmq user states.
    """

    def setUp(self):
        super(RabbitUserTestCase, self).setUp()
        rabbit_installed = self.run_function("cmd.has_exec", ["rabbitmqctl"])

        if not rabbit_installed:
            self.skipTest("rabbitmq-server not installed")

    def test_present(self):
        """
        rabbitmq_user.present null_name
        """
        ret = self.run_state("rabbitmq_user.present", name="null_name", test=True)
        self.assertSaltFalseReturn(ret)
        self.assertInSaltComment("User 'null_name' is set to be created", ret)

    def absent(self):
        """
        rabbitmq_user.absent null_name
        """
        ret = self.run_state("rabbitmq_user.absent", name="null_name", test=True)
        self.assertSaltFalseReturn(ret)

"""
Tests for the rabbitmq state
"""

from tests.support.case import ModuleCase
from tests.support.helpers import skip_if_not_root
from tests.support.mixins import SaltReturnAssertsMixin


@skip_if_not_root
class RabbitUserTestCase(ModuleCase, SaltReturnAssertsMixin):
    """
    Validate the rabbitmq user states.
    """

    def setUp(self):
        super().setUp()
        rabbit_installed = self.run_function("cmd.has_exec", ["rabbitmqctl"])

        if not rabbit_installed:
            self.skipTest("rabbitmq-server not installed")

    def test_present(self):
        """
        rabbitmq_user.present null_name
        """
        ret = self.run_state("rabbitmq_user.present", name="null_name", test=True)
        self.assertSaltFalseReturn(ret)
        self.assertInSaltComment("User 'null_name' is set to be created", ret)

    def absent(self):
        """
        rabbitmq_user.absent null_name
        """
        ret = self.run_state("rabbitmq_user.absent", name="null_name", test=True)
        self.assertSaltFalseReturn(ret)

Drop Py2 and six on tests/integration/states/test_rabbitmq_user.py