import os
from mocker import MockerTestCase
from cloudinit import util
from cloudinit.sources import DataSourceOpenNebula as ds

TEST_VARS = {
    'var1': 'single',
    'var2': 'double word',
    'var3': 'multi\nline\n',
    'var4': "'single'",
    'var5': "'double word'",
    'var6': "'multi\nline\n'",
    'var7': 'single\\t',
    'var8': 'double\\tword',
    'var9': 'multi\\t\nline\n'}

USER_DATA = '#cloud-config\napt_upgrade: true'
SSH_KEY = 'ssh-rsa AAAAB3NzaC1....sIkJhq8wdX+4I3A4cYbYP ubuntu@server-460-%i'
HOSTNAME = 'foo.example.com'
PUBLIC_IP = '10.0.0.3'

CMD_IP_OUT = '''\
1: lo: <LOOPBACK,UP,LOWER_UP> mtu 16436 qdisc noqueue state UNKNOWN \    link/loopback 00:00:00:00:00:00 brd 00:00:00:00:00:00
2: eth0: <BROADCAST,MULTICAST,UP,LOWER_UP> mtu 1500 qdisc mq state UP qlen 1000\    link/ether 02:00:0a:12:01:01 brd ff:ff:ff:ff:ff:ff
'''


class TestOpenNebulaDataSource(MockerTestCase):

    def setUp(self):
        super(TestOpenNebulaDataSource, self).setUp()
        self.tmp = self.makeDir()

    def test_seed_dir_non_contextdisk(self):
        my_d = os.path.join(self.tmp, 'non-contextdisk')
        self.assertRaises(ds.NonContextDiskDir, ds.read_context_disk_dir, my_d)

    def test_seed_dir_bad_context_sh(self):
        my_d = os.path.join(self.tmp, 'bad-context-sh')
        os.mkdir(my_d)
        with open(os.path.join(my_d, "context.sh"), "w") as fp:
            fp.write('/bin/false\n')
            fp.close()
        self.assertRaises(ds.NonContextDiskDir, ds.read_context_disk_dir, my_d)

    def test_context_sh_parser(self):
        my_d = os.path.join(self.tmp, 'context-sh-parser')
        populate_dir(my_d, TEST_VARS)
        results = ds.read_context_disk_dir(my_d)

        self.assertTrue('metadata' in results)
        self.assertEqual(TEST_VARS, results['metadata'])

    def test_ssh_key(self):
        public_keys = []
        for c in range(4):
            for k in ('SSH_KEY', 'SSH_PUBLIC_KEY'):
                my_d = os.path.join(self.tmp, "%s-%i" % (k, c))
                populate_dir(my_d, {k: '\n'.join(public_keys)})
                results = ds.read_context_disk_dir(my_d)

                self.assertTrue('metadata' in results)
                self.assertTrue('public-keys' in results['metadata'])
                self.assertEqual(public_keys, results['metadata']['public-keys'])

            public_keys.append(SSH_KEY % (c + 1,))

    def test_user_data(self):
        for k in ('USER_DATA', 'USERDATA'):
            my_d = os.path.join(self.tmp, k)
            populate_dir(my_d, {k: USER_DATA})
            results = ds.read_context_disk_dir(my_d)

            self.assertTrue('userdata' in results)
            self.assertEqual(USER_DATA, results['userdata'])

    def test_hostname(self):
        for k in ('HOSTNAME', 'PUBLIC_IP', 'IP_PUBLIC', 'ETH0_IP'):
            my_d = os.path.join(self.tmp, k)
            populate_dir(my_d, {k: PUBLIC_IP})
            results = ds.read_context_disk_dir(my_d)

            self.assertTrue('metadata' in results)
            self.assertTrue('local-hostname' in results['metadata'])
            self.assertEqual(PUBLIC_IP, results['metadata']['local-hostname'])

    def test_find_candidates(self):
        devs_with_answers = {
            "TYPE=iso9660": ["/dev/vdb"],
            "LABEL=CDROM": ["/dev/sr0"],
        }

        def my_devs_with(criteria):
            return devs_with_answers[criteria]

        try:
            orig_find_devs_with = util.find_devs_with
            util.find_devs_with = my_devs_with
            self.assertEqual(["/dev/sr0", "/dev/vdb"], ds.find_candidate_devs())
        finally:
            util.find_devs_with = orig_find_devs_with


class TestOpenNebulaNetwork(MockerTestCase):

    def setUp(self):
        super(TestOpenNebulaNetwork, self).setUp()

    def test_lo(self):
        net = ds.OpenNebulaNetwork('', {})
        self.assertEqual(net.gen_conf(), u'''\
auto lo
iface lo inet loopback
''')

    def test_eth0(self):
        net = ds.OpenNebulaNetwork(CMD_IP_OUT, {})
        self.assertEqual(net.gen_conf(), u'''\
auto lo
iface lo inet loopback

auto eth0
iface eth0 inet static
  address 10.18.1.1
  network 10.18.1.0
  netmask 255.255.255.0
''')

    def test_eth0_override(self):
        context_sh = {
            'dns': '1.2.3.8',
            'eth0_ip': '1.2.3.4',
            'eth0_network': '1.2.3.0',
            'eth0_mask': '255.255.0.0',
            'eth0_gateway': '1.2.3.5',
            'eth0_domain': 'example.com',
            'eth0_dns': '1.2.3.6 1.2.3.7'}

        net = ds.OpenNebulaNetwork(CMD_IP_OUT, context_sh)
        self.assertEqual(net.gen_conf(), u'''\
auto lo
iface lo inet loopback

auto eth0
iface eth0 inet static
  address 1.2.3.4
  network 1.2.3.0
  netmask 255.255.0.0
  gateway 1.2.3.5
  dns-search example.com
  dns-nameservers 1.2.3.8 1.2.3.6 1.2.3.7
''')


def populate_dir(seed_dir, files):
    os.mkdir(seed_dir)
    with open(os.path.join(seed_dir, "context.sh"), "w") as fp:
        fp.write("# Context variables generated by OpenNebula\n")
        for (name, content) in files.iteritems():
            fp.write('%s="%s"\n' % (name.upper(), content))
        fp.close()

# vi: ts=4 expandtab