# -*- coding: utf-8 -*-
"""
    :codeauthor: Ethan Devenport <ethand@stackpointcloud.com>
"""

# Import Python Libs
from __future__ import absolute_import, print_function, unicode_literals

# Import Third-Party Libs
from tests.integration.cloud.helpers.cloud_test_base import TIMEOUT, CloudTest

# Import Salt Testing Libs
from tests.support.unit import skipIf

try:
    # pylint: disable=unused-import
    from profitbricks.client import ProfitBricksService

    HAS_PROFITBRICKS = True
except ImportError:
    HAS_PROFITBRICKS = False


@skipIf(HAS_PROFITBRICKS is False, "salt-cloud requires >= profitbricks 4.1.0")
class ProfitBricksTest(CloudTest):
    """
    Integration tests for the ProfitBricks cloud provider
    """

    PROVIDER = "profitbricks"
    REQUIRED_PROVIDER_CONFIG_ITEMS = ("username", "password", "datacenter_id")

    def setUp(self):
        super(ProfitBricksTest, self).setUp()
        username = self.provider_config.get("username")
        password = self.provider_config.get("password")

        # A default username and password must be hard-coded as defaults as per issue #46265
        # If they are 'foo' and 'bar' it is the same as not being set

        self.skipTest(
            "Conf items are missing that must be provided to run these tests:  username, password"
            "\nCheck tests/integration/files/conf/cloud.providers.d/{0}.conf".format(
                self.PROVIDER
            )
        )

    def test_list_images(self):
        """
        Tests the return of running the --list-images command for ProfitBricks
        """
        list_images = self.run_cloud("--list-images {0}".format(self.PROVIDER))
        self.assertIn(
            "Ubuntu-16.04-LTS-server-2017-10-01", [i.strip() for i in list_images]
        )

    def test_list_image_alias(self):
        """
        Tests the return of running the -f list_images
        command for ProfitBricks
        """
        cmd = "-f list_images {0}".format(self.PROVIDER)
        list_images = self.run_cloud(cmd)
        self.assertIn("- ubuntu:latest", [i.strip() for i in list_images])

    def test_list_sizes(self):
        """
        Tests the return of running the --list_sizes command for ProfitBricks
        """
        list_sizes = self.run_cloud("--list-sizes {0}".format(self.PROVIDER))
        self.assertIn("Micro Instance:", [i.strip() for i in list_sizes])

    def test_list_datacenters(self):
        """
        Tests the return of running the -f list_datacenters
        command for ProfitBricks
        """
        cmd = "-f list_datacenters {0}".format(self.PROVIDER)
        list_datacenters = self.run_cloud(cmd)
        self.assertIn(
            self.provider_config["datacenter_id"], [i.strip() for i in list_datacenters]
        )

    def test_list_nodes(self):
        """
        Tests the return of running the -f list_nodes command for ProfitBricks
        """
        list_nodes = self.run_cloud("-f list_nodes {0}".format(self.PROVIDER))
        self.assertIn("state:", [i.strip() for i in list_nodes])

        self.assertIn("name:", [i.strip() for i in list_nodes])

    def test_list_nodes_full(self):
        """
        Tests the return of running the -f list_nodes_full
        command for ProfitBricks
        """
        cmd = "-f list_nodes_full {0}".format(self.PROVIDER)
        list_nodes = self.run_cloud(cmd)
        self.assertIn("state:", [i.strip() for i in list_nodes])

        self.assertIn("name:", [i.strip() for i in list_nodes])

    def test_list_location(self):
        """
        Tests the return of running the --list-locations
        command for ProfitBricks
        """
        cmd = "--list-locations {0}".format(self.PROVIDER)
        list_locations = self.run_cloud(cmd)

        self.assertIn("de/fkb", [i.strip() for i in list_locations])

        self.assertIn("de/fra", [i.strip() for i in list_locations])

        self.assertIn("us/las", [i.strip() for i in list_locations])

        self.assertIn("us/ewr", [i.strip() for i in list_locations])

    def test_instance(self):
        """
        Test creating an instance on ProfitBricks
        """
        # check if instance with salt installed returned
        ret_str = self.run_cloud(
            "-p profitbricks-test {0}".format(self.instance_name), timeout=TIMEOUT
        )
        self.assertInstanceExists(ret_str)

        self.assertDestroyInstance()