#!/usr/bin/python2.5
#
# Copyright 2009 the Melange authors.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#   http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.


__authors__ = [
  '"Sverre Rabbelier" <sverre@rabbelier.nl>',
  ]


import unittest

from google.appengine.api import users
from google.appengine.api import memcache

from soc.cache import sidebar


class SidebarCacheTest(unittest.TestCase):
  """Tests that the sidebar properly uses caching.
  """

  def setUp(self):
    self.called = 0
    self.user = users.get_current_user()
    self.memcache_key = "sidebar_for_'id'"

  def tearDown(self):
    memcache.flush_all()

  def testGetCurrentUser(self):
    """Sanity check to see if get_current_user returns same value.
    """

    self.assertEqual(self.user, users.get_current_user())

  def testKey(self):
    """Test that the key method returns a unique key.
    """

    self.assertEqual("sidebar_for_users.User(email='test@example.com')",
                     sidebar.key(self.user))

  def testGet(self):
    """Test that get without setting something returns None.
    """

    self.assertEqual((None, self.memcache_key), sidebar.get('id'))

  def testGetPut(self):
    """Test that getting after putting gives back what you put in.
    """

    sidebar.put(42, self.memcache_key)
    self.assertEqual((42, self.memcache_key), sidebar.get('id'))

  def testFlush(self):
    """Test that getting after putting and flushing returns None.
    """

    sidebar.put(42, self.memcache_key)
    sidebar.flush('id')
    self.assertEqual((None, self.memcache_key), sidebar.get('id'))

  def testCache(self):
    """Test that the result of a cached sidebar is cached.
    """

    @sidebar.cache
    def getAnswer(x):
      self.called = self.called + 1
      return 42

    self.assertEqual(42, getAnswer('id'))
    self.assertEqual(42, getAnswer('id'))
    self.assertEqual(self.called, 1)