+
−#!/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)
+
−
+
−