# Copyright 2007 Google Inc.
#
# This program is free software; you can redistribute it and/or
# modify it under the terms of the GNU General Public License
# as published by the Free Software Foundation; either version 2
# of the License, or (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software Foundation,
# Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.

"""Unit tests for caches/caches.py."""

__author__ = 'jaq@google.com (Jamie Wilkinson)'

import os
import stat
import tempfile
import unittest

import mox

from nss_cache import config
from nss_cache.caches import caches


class FakeCacheCls(caches.Cache):

  CACHE_FILENAME = 'shadow'
  def __init__(self, config, map_name):
    super(FakeCacheCls, self).__init__(config, map_name)

  def Write(self, map_data):
    return 0

  def GetCacheFilename(self):
    return os.path.join(self.output_dir,
                        self.CACHE_FILENAME + '.test')


class TestCls(mox.MoxTestBase):

  def setUp(self):
    self.workdir = tempfile.mkdtemp()
    self.config = {'dir': self.workdir}

  def tearDown(self):
    os.rmdir(self.workdir)

  def testCopyOwnerMissing(self):
    expected = os.stat(os.path.join('/etc', config.MAP_SHADOW))
    expected = stat.S_IMODE(expected.st_mode)
    cache = FakeCacheCls(config=self.config, map_name=config.MAP_SHADOW)
    cache._Begin()
    cache._Commit()
    data = os.stat(os.path.join(self.workdir, cache.GetCacheFilename()))
    self.assertEqual(expected, stat.S_IMODE(data.st_mode))
    os.unlink(cache.GetCacheFilename())

  def testCopyOwnerPresent(self):
    expected = os.stat(os.path.join('/etc/', config.MAP_SHADOW))
    expected = stat.S_IMODE(expected.st_mode)
    cache = FakeCacheCls(config=self.config, map_name=config.MAP_SHADOW)
    cache._Begin()
    cache._Commit()
    data = os.stat(os.path.join(self.workdir, cache.GetCacheFilename()))
    self.assertEqual(expected, stat.S_IMODE(data.st_mode))
    os.unlink(cache.GetCacheFilename())


class TestCache(mox.MoxTestBase):
  def testWriteMap(self):
    cache_map = caches.Cache({}, config.MAP_PASSWORD, None)
    self.mox.StubOutWithMock(cache_map, '_Commit')
    self.mox.StubOutWithMock(cache_map, 'Write')
    self.mox.StubOutWithMock(cache_map, 'Verify')
    cache_map._Commit()
    cache_map.Write('writable_map').AndReturn('entries_written')
    cache_map.Verify('entries_written').AndReturn(True)
    self.mox.ReplayAll()

    self.assertEqual(0, cache_map.WriteMap('writable_map'))


if __name__ == '__main__':
  unittest.main()
