import importlib
import os
import shutil

from django.test import TestCase

from django_secrets.startup import check, create_secrets_package


class SecretTest(TestCase):

    def test_adding_a_secret(self):
        self.assertEqual(True, True)

    def test_creating_secrets_folder(self):

        self.assertIn('my_secrets', os.listdir('.'))
        shutil.rmtree("my_secrets")
        self.assertNotIn('my_secrets', os.listdir('.'))
        create_secrets_package(testing=True)
        self.assertIn('my_secrets', os.listdir('.'))

        # test adding back the secret
        os.environ["SECOND_SECRET"] = "blub"
        check()
        from my_secrets import secrets
        if not hasattr(secrets, 'SECOND_SECRET'):  # pragma: no cover / travis import problem fix
            spec = importlib.util.spec_from_file_location('secrets', 'my_secrets/secrets.py')
            secrets = importlib.util.module_from_spec(spec)
            spec.loader.exec_module(secrets)
        self.assertEqual(secrets.SECOND_SECRET, 'blub')

    def test_export(self):
        from django_secrets.management.commands.export_secrets import Command

        command = Command()
        command.handle()

import importlib
import os
import shutil

from django.test import TestCase

from django_secrets.startup import check, create_secrets_package


class SecretTest(TestCase):

    def test_adding_a_secret(self):
        self.assertEqual(True, True)

    def test_creating_secrets_folder(self):

        self.assertIn('my_secrets', os.listdir('.'))
        shutil.rmtree("my_secrets")
        self.assertNotIn('my_secrets', os.listdir('.'))
        create_secrets_package(testing=True)
        self.assertIn('my_secrets', os.listdir('.'))

        # test adding back the secret
        os.environ["SECOND_SECRET"] = "blub"
        check()
        from my_secrets import secrets
        if not hasattr(secrets, 'SECOND_SECRET'):  # pragma: no cover / travis import problem fix
            spec = importlib.util.spec_from_file_location('secrets', 'my_secrets/secrets.py')
            secrets = importlib.util.module_from_spec(spec)
            spec.loader.exec_module(secrets)
        self.assertEqual(secrets.SECOND_SECRET, 'blub')

    def test_export(self):
        from django_secrets.management.commands.export_secrets import Command

        command = Command()
        command.handle()

# test adding back the secret # pragma: no cover / travis import problem fix