import os
import unittest

from ansible_module.generic import Types, joindict, systemdbool


class TestTypes(unittest.TestCase):
    """tests the Types class"""

    def testsimpletype(self):
        """this tests if an simple type is correctly build"""
        output = Types.str(required=True, help="test", choices=("a", "1"), default="1")
        self.assertIn("type", output)
        self.assertIn("required", output)
        self.assertIn("default", output)
        self.assertIn("choices", output)
        self.assertEqual(output["type"], "str")
        self.assertEqual(Types.str.__name__, "str")
        self.assertEqual(output["required"], True)
        self.assertEqual(output["default"], "1")
        self.assertTupleEqual(output["choices"], ("a", "1"))
        Types.str()

    def testlisttype(self):
        """this tests if the special type list is correctly build"""
        output = Types.list(str)
        Types.list("str")
        self.assertIn("type", output)
        self.assertIn("elements", output)
        self.assertIn("required", output)
        self.assertEqual(output["type"], "list")
        self.assertEqual(output["required"], False)
        self.assertEqual(output["elements"], "str")
        Types.list(Types.dict(a=Types.str(help="")), help="")
        Types.list("str")

    def testdicttype(self):
        output = Types.dict(help="HILFE")
        self.assertIn("type", output)
        self.assertIn("required", output)
        self.assertIn("description", output)
        self.assertIn("option", output)


class TestFuncs(unittest.TestCase):
    def testsystemdbool(self):
        self.assertEqual("Text", systemdbool("Text"))
        self.assertEqual("no", systemdbool(False))
        self.assertEqual("yes", systemdbool(True))

    def testjoindict(self):
        dicts = (
            dict(a=1, b=2),
            dict(b=3, c=4),
        )
        output = dict(a=1, b=3, c=4)
        self.assertDictEqual(output, joindict(*dicts))