X-Git-Url: https://code.grnet.gr/git/ganeti-local/blobdiff_plain/64dd91631666ca2e641dd061d172ced48e1b82e8..9bbb3c37aad87c5835094c89c18fbb94cc081656:/test/ganeti.rapi.resources_unittest.py

diff --git a/test/ganeti.rapi.resources_unittest.py b/test/ganeti.rapi.resources_unittest.py
index d0a85a0..8fc40ee 100755
--- a/test/ganeti.rapi.resources_unittest.py
+++ b/test/ganeti.rapi.resources_unittest.py
@@ -24,55 +24,121 @@
 
 import os
 import unittest
+import tempfile
+import time
 
+from ganeti import errors
+from ganeti.rapi import httperror
 from ganeti.rapi import resources
+from ganeti.rapi import RESTHTTPServer
 
 
 class MapperTests(unittest.TestCase):
+  """Tests for remote API URI mapper."""
 
   def setUp(self):
-    self.con = resources.CONNECTOR
-    self.map = resources.Mapper(self.con)
-    
+    self.map = resources.Mapper()
+
+  def _TestUri(self, uri, result):
+    self.assertEquals(self.map.getController(uri), result)
+
+  def _TestFailingUri(self, uri):
+    self.failUnlessRaises(httperror.HTTPNotFound, self.map.getController, uri)
+
   def testMapper(self):
-    """Testing for Mapper."""
-    
-    self.failUnless(self.map.getController('/tags') == 
-                    ('R_tags',
-                     ['/tags'],
-                     {}))
-
-    self.failUnless(self.map.getController('/tag') == None)
-    
-    self.failUnless(self.map.getController('/instances/www.test.com') == 
-                    ('R_instances_name',
-                     ['www.test.com'],
-                     {}))
-    
-    self.failUnless(self.map.getController(
-        '/instances/www.test.com/tags?f=5&f=6&alt=html') == 
-                    ('R_instances_name_tags',
-                     ['www.test.com'],
-                     {'alt':['html'], 'f':['5', '6']}))
+    """Testing resources.Mapper"""
+
+    self._TestUri("/tags", (resources.R_tags, [], {}))
+
+    self._TestUri('/instances/www.test.com',
+                  (resources.R_instances_name,
+                   ['www.test.com'],
+                   {}))
+
+    self._TestUri('/instances/www.test.com/tags?f=5&f=6&alt=html',
+                  (resources.R_instances_name_tags,
+                   ['www.test.com'],
+                   {'alt': ['html'],
+                    'f': ['5', '6'],
+                   }))
+
+    self._TestFailingUri("/tag")
+    self._TestFailingUri("/instances/does/not/exist")
 
 
 class R_RootTests(unittest.TestCase):
   """Testing for R_root class."""
-  
+
   def setUp(self):
     self.root = resources.R_root(None, None, None)
-    self.root.result = []
-  
+
   def testGet(self):
-    self.root._get()
-    self.failUnless(self.root.result == 
-                    [{'name': 'instances', 'uri': '/instances'},
-                     {'name': 'info', 'uri': '/info'},
-                     {'name': 'os', 'uri': '/os'},
-                     {'name': 'status', 'uri': '/status'},
-                     {'name': 'tags', 'uri': '/tags'},
-                     {'name': 'nodes', 'uri': '/nodes'}])
-    
+    expected = [
+      {'name': 'info', 'uri': '/info'},
+      {'name': 'instances', 'uri': '/instances'},
+      {'name': 'nodes', 'uri': '/nodes'},
+      {'name': 'os', 'uri': '/os'},
+      {'name': 'tags', 'uri': '/tags'},
+      {'name': 'version', 'uri': '/version'},
+      ]
+    self.assertEquals(self.root.GET(), expected)
+
+
+class HttpLogfileTests(unittest.TestCase):
+  """Rests for HttpLogfile class."""
+
+  class FakeRequest:
+    FAKE_ADDRESS = "1.2.3.4"
+
+    def address_string(self):
+      return self.FAKE_ADDRESS
+
+  def setUp(self):
+    self.tmpfile = tempfile.NamedTemporaryFile()
+    self.logfile = RESTHTTPServer.HttpLogfile(self.tmpfile.name)
+
+  def testFormatLogTime(self):
+    self._TestInTimezone(1208646123.0, "Europe/London",
+                         "19/Apr/2008:23:02:03 +0000")
+    self._TestInTimezone(1208646123, "Europe/Zurich",
+                         "19/Apr/2008:23:02:03 +0000")
+    self._TestInTimezone(1208646123, "Australia/Sydney",
+                         "19/Apr/2008:23:02:03 +0000")
+
+  def _TestInTimezone(self, seconds, timezone, expected):
+    """Tests HttpLogfile._FormatLogTime with a specific timezone
+
+    """
+    # Preserve environment
+    old_TZ = os.environ.get("TZ", None)
+    try:
+      os.environ["TZ"] = timezone
+      time.tzset()
+      result = self.logfile._FormatLogTime(seconds)
+    finally:
+      # Restore environment
+      if old_TZ is not None:
+        os.environ["TZ"] = old_TZ
+      elif "TZ" in os.environ:
+        del os.environ["TZ"]
+      time.tzset()
+
+    self.assertEqual(result, expected)
+
+  def testClose(self):
+    self.logfile.Close()
+
+  def testCloseAndWrite(self):
+    request = self.FakeRequest()
+    self.logfile.Close()
+    self.assertRaises(errors.ProgrammerError, self.logfile.LogRequest,
+                      request, "Message")
+
+  def testLogRequest(self):
+    request = self.FakeRequest()
+    self.logfile.LogRequest(request, "This is only a %s", "test")
+    self.logfile.Close()
+
 
 if __name__ == '__main__':
   unittest.main()