Synnefo

#

# Unit Tests for the Ganeti-specific interfaces

#

# Provides unit tests for the code implementing

# the Ganeti notification daemon and the Ganeti hook in Synnefo.

#

# Copyright 2011 Greek Research and Technology Network

#

import logging

from django.test import TestCase

from django.conf import settings

from ganeti.hooks import ganeti_net_status

class GanetiHookTestCase(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'

        }

    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' }

            ]

        }

        self.assertEqual(ganeti_net_status(logging, e), expected)