/ - Diff - Synnefo - Greek Research and Technology Network's projects

Revision 0e1f3323

     from signal import signal, SIGINT, SIGTERM
     import setproctitle
     from ganeti import utils
     from ganeti import jqueue
     from ganeti import constants
     from ganeti import serializer
     from ganeti import utils, jqueue, constants, serializer, cli
     from ganeti.ssconf import SimpleConfigReader
-...
     from synnefo.lib.amqp import AMQPClient
     def get_time_from_status(op, job):
         """Generate a unique message identifier for a ganeti job.
-...
         raise InvalidBackendStatus(status, job)
     def get_instance_nics(instance, logger):
         """Query Ganeti to a get the instance's NICs.
         @type instance: string
         @param instance: the name of the instance
         @rtype: List of dicts
         @retrun: Dictionary containing the instance's NICs. Each dictionary
                  contains the following keys: 'network', 'ip', 'mac', 'mode',
                  'link' and 'firewall'
         """
         fields = ["nic.networks", "nic.ips", "nic.macs", "nic.modes", "nic.links",
                   "tags"]
         # Get Ganeti client
         client = cli.GetClient()
         info = client.QueryInstances([instance], fields, use_locking=False)
         networks, ips, macs, modes, links, tags = info[0]
         nic_keys = ["network", "ip", "mac", "mode", "link"]
         nics = zip(networks, ips, macs, modes, links)
         nics = map(lambda x: dict(zip(nic_keys, x)), nics)
         # Get firewall from instance Tags
         # Tags are of the form synnefo:network:N:firewall_mode
         for tag in tags:
             t = tag.split(":")
             if t[0:2] == ["synnefo", "network"]:
                 if len(t) != 4:
                     logger.error("Malformed synefo tag %s", tag)
                     continue
                 try:
                     index = int(t[2])
                     nics[index]['firewall'] = t[3]
                 except ValueError:
                     logger.error("Malformed synnefo tag %s", tag)
                 except IndexError:
                     logger.error("Found tag %s for non-existent NIC %d",
                                  tag, index)
         return nics
     class InvalidBackendStatus(Exception):
         def __init__(self, status, job):
             self.status = status
-...
                             "logmsg": logmsg,
                             "jobId": job_id})
                 if op_id in ["OP_INSTANCE_CREATE", "OP_INSTANCE_SET_PARAMS",
                              "OP_INSTANCE_STARTUP"]:
                     if op.status == "success":
                         nics = get_instance_nics(msg["instance"], self.logger)
                         msg["nics"] = nics
                 msg = json.dumps(msg)
                 self.logger.debug("Delivering msg: %s (key=%s)", msg, routekey)

     #!/usr/bin/env python
+    #
     # -*- coding: utf-8 -*-
+    #
     # Copyright 2011 GRNET S.A. All rights reserved.
     # Copyright 2013 GRNET S.A. All rights reserved.
+    #
     # Redistribution and use in source and binary forms, with or
     # without modification, are permitted provided that the following
-...
     # interpreted as representing official policies, either expressed
     # or implied, of GRNET S.A.
+    #
+    #
     """Unit Tests for the Ganeti-specific interfaces in synnefo.ganeti
     Provides unit tests for the code implementing
     the Ganeti notification daemon and the Ganeti hook in Synnefo.
     """
     # This assumes a test-specific configuration file
     # is in the same directory as the unit test script
     import os
     os.environ["SYNNEFO_CONFIG_DIR"] = os.path.dirname(__file__)
     import sys
     import logging
     import unittest
     from synnefo.ganeti.eventd import get_instance_nics
     from mock import patch
     from synnefo.ganeti.hook import ganeti_net_status
     log = logging.getLogger()
     # Use backported unittest functionality if Python < 2.7
     try:
         import unittest2 as unittest
     except ImportError:
         if sys.version_info < (2, 7):
             raise Exception("The unittest2 package is required for Python < 2.7")
         import unittest
     class GanetiHookTestCase(unittest.TestCase):
         def setUp(self):
             # Example Ganeti environment, based on from
             # http://docs.ganeti.org/ganeti/master/html/hooks.html?highlight=hooks#examples
             self.env = {
                 'GANETI_CLUSTER': 'cluster1.example.com',
                 'GANETI_DATA_DIR': '/var/lib/ganeti',
                 'GANETI_FORCE': 'False',
                 'GANETI_HOOKS_PATH': 'instance-start',
                 'GANETI_HOOKS_PHASE': 'post',
                 'GANETI_HOOKS_VERSION': '2',
                 'GANETI_INSTANCE_DISK0_MODE': 'rw',
                 'GANETI_INSTANCE_DISK0_SIZE': '128',
                 'GANETI_INSTANCE_DISK_COUNT': '1',
                 'GANETI_INSTANCE_DISK_TEMPLATE': 'drbd',
                 'GANETI_INSTANCE_MEMORY': '128',
                 'GANETI_INSTANCE_TAGS': 'tag1 synnefo:network:0:protected tag2',
                 'GANETI_INSTANCE_NAME': 'instance2.example.com',
                 'GANETI_INSTANCE_NIC0_BRIDGE': 'xen-br0',
                 'GANETI_INSTANCE_NIC0_IP': '147.102.3.1',
                 'GANETI_INSTANCE_NIC0_MAC': '00:01:de:ad:be:ef',
                 'GANETI_INSTANCE_NIC1_MAC': '00:01:de:ad:ba:be',
                 'GANETI_INSTANCE_NIC2_MAC': '00:01:02:03:04:05',
                 'GANETI_INSTANCE_NIC2_IP': '147.102.3.98',
                 'GANETI_INSTANCE_NIC_COUNT': '3',
                 'GANETI_INSTANCE_OS_TYPE': 'debootstrap',
                 'GANETI_INSTANCE_PRIMARY': 'node3.example.com',
                 'GANETI_INSTANCE_SECONDARY': 'node5.example.com',
                 'GANETI_INSTANCE_STATUS': 'down',
                 'GANETI_INSTANCE_VCPUS': '1',
                 'GANETI_MASTER': 'node1.example.com',
                 'GANETI_OBJECT_TYPE': 'INSTANCE',
                 'GANETI_OP_CODE': 'OP_INSTANCE_STARTUP',
                 'GANETI_OP_TARGET': 'instance2.example.com'
+            }
     @patch("ganeti.cli.GetClient")
     class GanetiNICTestCase(unittest.TestCase):
         def test_no_nics(self, client):
             ret = [[[], [], [], [], [], []]]
             client.return_value.QueryInstances.return_value = ret
             self.assertEqual(get_instance_nics('test', log), [])
         def test_ganeti_net_status(self):
             e = self.env
             expected = {
                 'type': 'ganeti-net-status',
                 'instance': 'instance2.example.com',
                 'nics': [
+                    {
                         'ip': '147.102.3.1', 'mac': '00:01:de:ad:be:ef',
                         'link': 'xen-br0', 'ipv6': '2001:db8::201:deff:fead:beef',
                         'firewall': 'protected'
                     },
                     { 'mac': '00:01:de:ad:ba:be' },
                     { 'ip': '147.102.3.98', 'mac': '00:01:02:03:04:05' }
+                ]
+            }
         def test_one_nic(self, client):
             ret = [[["network"], ["ip"], ["mac"], ["mode"], ["link"],
                     ["tag1", "tag2"]]]
             client.return_value.QueryInstances.return_value = ret
             nics0 = get_instance_nics("test", log)
             nics1 = [{"network": "network",
                       "ip": "ip",
                       "mac": "mac",
                       "mode": "mode",
                       "link": "link"}]
             self.assertEqual(nics0, nics1)
             self.assertEqual(ganeti_net_status(logging, e), expected)
         def test_two_nics(self, client):
             ret = [[["network1", "network2"], ["ip1", "ip2"], ["mac1", "mac2"],
                     ["mode1", "mode2"], ["link1", "link2"], ["tag1", "tag2"]]]
             client.return_value.QueryInstances.return_value = ret
             nics0 = get_instance_nics("test", log)
             nics1 = [{"network": "network1",
                       "ip": "ip1",
                       "mac": "mac1",
                       "mode": "mode1",
                       "link": "link1"},
                       {"network": "network2",
                        "ip": "ip2",
                        "mac": "mac2",
                        "mode": "mode2",
                        "link": "link2"}]
             self.assertEqual(nics0, nics1)
         def test_firewall(self, client):
             ret = [[["network"], ["ip"], ["mac"], ["mode"], ["link"],
                 ["tag1", "synnefo:network:0:protected"]]]
             client.return_value.QueryInstances.return_value = ret
             nics0 = get_instance_nics("test", log)
             nics1 = [{"network": "network",
                       "ip": "ip",
                       "mac": "mac",
                       "mode": "mode",
                       "link": "link",
                       "firewall": "protected"}]
             self.assertEqual(nics0, nics1)
     if __name__ == '__main__':
         unittest.main()

Also available in: Unified diff