# Copyright 2023 The Fuchsia Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

from __future__ import annotations

import os
import shutil
import tempfile
import unittest

from fuchsia_controller_py import IsolateDir


class IsolateDirectory(unittest.TestCase):
    def test_isolate_dir_creation_empty(self) -> None:
        isolate_dir: IsolateDir | None = IsolateDir()
        assert isolate_dir is not None
        temp_dir = isolate_dir.directory()

        self.assertTrue(os.path.exists(temp_dir))

        isolate_dir = None
        self.assertFalse(os.path.exists(temp_dir))

    def test_isolate_dir_creation_new_dir(self) -> None:
        temp_dir = tempfile.mkdtemp()  # Get a random directory path
        shutil.rmtree(temp_dir)  # Guarantee directory doesn't exist yet
        self.assertFalse(os.path.exists(temp_dir))

        isolate_dir: IsolateDir | None = IsolateDir(dir=temp_dir)
        assert isolate_dir is not None
        self.assertEqual(isolate_dir.directory(), temp_dir)
        self.assertTrue(os.path.exists(temp_dir))

        isolate_dir = None
        self.assertFalse(os.path.exists(temp_dir))

    def test_isolate_dir_creation_existing_dir(self) -> None:
        temp_dir = tempfile.mkdtemp()
        self.assertTrue(os.path.exists(temp_dir))

        isolate_dir: IsolateDir | None = IsolateDir(dir=temp_dir)
        assert isolate_dir is not None
        self.assertEqual(isolate_dir.directory(), temp_dir)
        self.assertTrue(os.path.exists(temp_dir))

        isolate_dir = None
        self.assertFalse(os.path.exists(temp_dir))