Statistics
| Branch: | Tag: | Revision:

root / test / ganeti.hypervisor.hv_chroot_unittest.py @ 55cc0a44

History | View | Annotate | Download (1.6 kB)

1
#!/usr/bin/python
2
#
3

    
4
# Copyright (C) 2011 Google Inc.
5
#
6
# This program is free software; you can redistribute it and/or modify
7
# it under the terms of the GNU General Public License as published by
8
# the Free Software Foundation; either version 2 of the License, or
9
# (at your option) any later version.
10
#
11
# This program is distributed in the hope that it will be useful, but
12
# WITHOUT ANY WARRANTY; without even the implied warranty of
13
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
14
# General Public License for more details.
15
#
16
# You should have received a copy of the GNU General Public License
17
# along with this program; if not, write to the Free Software
18
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
19
# 02110-1301, USA.
20

    
21

    
22
"""Script for testing ganeti.hypervisor.hv_chroot"""
23

    
24
import unittest
25
import tempfile
26
import shutil
27

    
28
from ganeti import constants
29
from ganeti import objects
30
from ganeti import hypervisor
31

    
32
from ganeti.hypervisor import hv_chroot
33

    
34
import testutils
35

    
36

    
37
class TestConsole(unittest.TestCase):
38
  def setUp(self):
39
    self.tmpdir = tempfile.mkdtemp()
40

    
41
  def tearDown(self):
42
    shutil.rmtree(self.tmpdir)
43

    
44
  def test(self):
45
    instance = objects.Instance(name="fake.example.com", primary_node="node837")
46
    cons = hv_chroot.ChrootManager.GetInstanceConsole(instance, {}, {},
47
                                                      root_dir=self.tmpdir)
48
    self.assertTrue(cons.Validate())
49
    self.assertEqual(cons.kind, constants.CONS_SSH)
50
    self.assertEqual(cons.host, instance.primary_node)
51

    
52

    
53
if __name__ == "__main__":
54
  testutils.GanetiTestProgram()