import os
import sys
import stat
import copy
import lxml.etree
from mock import Mock, MagicMock, patch
from Bcfg2.Client.Tools.POSIX.Directory import *

# add all parent testsuite directories to sys.path to allow (most)
# relative imports in python 2.4
path = os.path.dirname(__file__)
while path != "/":
    if os.path.basename(path).lower().startswith("test"):
        sys.path.append(path)
    if os.path.basename(path) == "testsuite":
        break
    path = os.path.dirname(path)
from Test__init import get_posix_object
from Testbase import TestPOSIXTool
from common import *

class TestPOSIXDirectory(TestPOSIXTool):
    test_obj = POSIXDirectory

    @patch("os.listdir")
    @patch("Bcfg2.Client.Tools.POSIX.base.POSIXTool.verify")
    @patch("Bcfg2.Client.Tools.POSIX.Directory.%s._exists" % test_obj.__name__)
    def test_verify(self, mock_exists, mock_verify, mock_listdir):
        entry = lxml.etree.Element("Path", name="/test", type="directory",
                                   mode='0644', owner='root', group='root')

        mock_exists.return_value = False
        self.assertFalse(self.ptool.verify(entry, []))
        mock_exists.assert_called_with(entry)

        mock_exists.reset_mock()
        exists_rv = MagicMock()
        exists_rv.__getitem__.return_value = stat.S_IFREG | 420 # 0o644
        mock_exists.return_value = exists_rv
        self.assertFalse(self.ptool.verify(entry, []))
        mock_exists.assert_called_with(entry)

        mock_exists.reset_mock()
        mock_verify.return_value = False
        exists_rv.__getitem__.return_value = stat.S_IFDIR | 420 # 0o644
        self.assertFalse(self.ptool.verify(entry, []))
        mock_exists.assert_called_with(entry)
        mock_verify.assert_called_with(self.ptool, entry, [])

        mock_exists.reset_mock()
        mock_verify.reset_mock()
        mock_verify.return_value = True
        self.assertTrue(self.ptool.verify(entry, []))
        mock_exists.assert_called_with(entry)
        mock_verify.assert_called_with(self.ptool, entry, [])

        mock_exists.reset_mock()
        mock_verify.reset_mock()
        entry.set("prune", "true")
        orig_entry = copy.deepcopy(entry)

        entries = ["foo", "bar", "bar/baz"]
        mock_listdir.return_value = entries
        modlist = [os.path.join(entry.get("name"), entries[0])]
        self.assertFalse(self.ptool.verify(entry, modlist))
        mock_exists.assert_called_with(entry)
        mock_verify.assert_called_with(self.ptool, entry, modlist)
        mock_listdir.assert_called_with(entry.get("name"))
        expected = [os.path.join(entry.get("name"), e)
                    for e in entries
                    if os.path.join(entry.get("name"), e) not in modlist]
        actual = [e.get("path") for e in entry.findall("Prune")]
        self.assertItemsEqual(expected, actual)

        mock_verify.reset_mock()
        mock_exists.reset_mock()
        mock_listdir.reset_mock()
        entry = copy.deepcopy(orig_entry)
        modlist = [os.path.join(entry.get("name"), e)
                   for e in entries]
        self.assertTrue(self.ptool.verify(entry, modlist))
        mock_exists.assert_called_with(entry)
        mock_verify.assert_called_with(self.ptool, entry, modlist)
        mock_listdir.assert_called_with(entry.get("name"))
        self.assertEqual(len(entry.findall("Prune")), 0)

    @patch("os.unlink")
    @patch("Bcfg2.Client.Tools.POSIX.base.POSIXTool.install")
    @patch("Bcfg2.Client.Tools.POSIX.Directory.%s._exists" % test_obj.__name__)
    @patch("Bcfg2.Client.Tools.POSIX.Directory.%s._makedirs" %
           test_obj.__name__)
    def test_install(self, mock_makedirs, mock_exists, mock_install,
                     mock_unlink):
        entry = lxml.etree.Element("Path", name="/test/foo/bar",
                                   type="directory", mode='0644',
                                   owner='root', group='root')

        self.ptool._makedirs = Mock()
        self.ptool._remove = Mock()

        def reset():
            mock_exists.reset_mock()
            mock_install.reset_mock()
            mock_unlink.reset_mock()
            self.ptool._makedirs.reset_mock()
            self.ptool._remove.reset_mock()

        self.ptool._makedirs.return_value = True
        mock_exists.return_value = False
        mock_install.return_value = True
        self.assertTrue(self.ptool.install(entry))
        mock_exists.assert_called_with(entry)
        mock_install.assert_called_with(self.ptool, entry)
        self.ptool._makedirs.assert_called_with(entry)

        reset()
        exists_rv = MagicMock()
        exists_rv.__getitem__.return_value = stat.S_IFREG | 420 # 0o644
        mock_exists.return_value = exists_rv
        self.assertTrue(self.ptool.install(entry))
        mock_unlink.assert_called_with(entry.get("name"))
        mock_exists.assert_called_with(entry)
        self.ptool._makedirs.assert_called_with(entry)
        mock_install.assert_called_with(self.ptool, entry)

        reset()
        exists_rv.__getitem__.return_value = stat.S_IFDIR | 420 # 0o644
        mock_install.return_value = True
        self.assertTrue(self.ptool.install(entry))
        mock_exists.assert_called_with(entry)
        mock_install.assert_called_with(self.ptool, entry)

        reset()
        mock_install.return_value = False
        self.assertFalse(self.ptool.install(entry))
        mock_install.assert_called_with(self.ptool, entry)

        entry.set("prune", "true")
        prune = ["/test/foo/bar/prune1", "/test/foo/bar/prune2"]
        for path in prune:
            lxml.etree.SubElement(entry, "Prune", path=path)

        reset()
        mock_install.return_value = True

        self.assertTrue(self.ptool.install(entry))
        mock_exists.assert_called_with(entry)
        mock_install.assert_called_with(self.ptool, entry)
        self.assertItemsEqual([c[0][0].get("path")
                               for c in self.ptool._remove.call_args_list],
                              prune)