Statistics
| Branch: | Tag: | Revision:

root / test / ganeti.bdev_unittest.py @ d4fa5c23

History | View | Annotate | Download (5.4 kB)

1 3840729d Iustin Pop
#!/usr/bin/python
2 3840729d Iustin Pop
#
3 3840729d Iustin Pop
4 3840729d Iustin Pop
# Copyright (C) 2006, 2007 Google Inc.
5 3840729d Iustin Pop
#
6 3840729d Iustin Pop
# This program is free software; you can redistribute it and/or modify
7 3840729d Iustin Pop
# it under the terms of the GNU General Public License as published by
8 3840729d Iustin Pop
# the Free Software Foundation; either version 2 of the License, or
9 3840729d Iustin Pop
# (at your option) any later version.
10 3840729d Iustin Pop
#
11 3840729d Iustin Pop
# This program is distributed in the hope that it will be useful, but
12 3840729d Iustin Pop
# WITHOUT ANY WARRANTY; without even the implied warranty of
13 3840729d Iustin Pop
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
14 3840729d Iustin Pop
# General Public License for more details.
15 3840729d Iustin Pop
#
16 3840729d Iustin Pop
# You should have received a copy of the GNU General Public License
17 3840729d Iustin Pop
# along with this program; if not, write to the Free Software
18 3840729d Iustin Pop
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
19 3840729d Iustin Pop
# 02110-1301, USA.
20 3840729d Iustin Pop
21 3840729d Iustin Pop
22 3840729d Iustin Pop
"""Script for unittesting the bdev module"""
23 3840729d Iustin Pop
24 3840729d Iustin Pop
25 6634816b Iustin Pop
import os
26 3840729d Iustin Pop
import unittest
27 3840729d Iustin Pop
28 3840729d Iustin Pop
from ganeti import bdev
29 6b90c22e Iustin Pop
from ganeti import errors
30 3840729d Iustin Pop
31 3840729d Iustin Pop
32 3840729d Iustin Pop
class TestDRBD8Runner(unittest.TestCase):
33 3840729d Iustin Pop
  """Testing case for DRBD8"""
34 3840729d Iustin Pop
35 3840729d Iustin Pop
  @staticmethod
36 3840729d Iustin Pop
  def _has_disk(data, dname, mname):
37 3840729d Iustin Pop
    """Check local disk corectness"""
38 3840729d Iustin Pop
    retval = (
39 3840729d Iustin Pop
      "local_dev" in data and
40 3840729d Iustin Pop
      data["local_dev"] == dname and
41 3840729d Iustin Pop
      "meta_dev" in data and
42 3840729d Iustin Pop
      data["meta_dev"] == mname and
43 3840729d Iustin Pop
      "meta_index" in data and
44 3840729d Iustin Pop
      data["meta_index"] == 0
45 3840729d Iustin Pop
      )
46 3840729d Iustin Pop
    return retval
47 3840729d Iustin Pop
48 3840729d Iustin Pop
  @staticmethod
49 6634816b Iustin Pop
  def _get_contents(name):
50 6634816b Iustin Pop
    """Returns the contents of a file"""
51 6634816b Iustin Pop
52 6634816b Iustin Pop
    prefix = os.environ.get("srcdir", None)
53 6634816b Iustin Pop
    if prefix:
54 e8230860 Michael Hanselmann
      name = prefix + "/test/" + name
55 6634816b Iustin Pop
    fh = open(name, "r")
56 6634816b Iustin Pop
    try:
57 6634816b Iustin Pop
      data = fh.read()
58 6634816b Iustin Pop
    finally:
59 6634816b Iustin Pop
      fh.close()
60 6634816b Iustin Pop
    return data
61 6634816b Iustin Pop
62 6634816b Iustin Pop
  @staticmethod
63 3840729d Iustin Pop
  def _has_net(data, local, remote):
64 3840729d Iustin Pop
    """Check network connection parameters"""
65 3840729d Iustin Pop
    retval = (
66 3840729d Iustin Pop
      "local_addr" in data and
67 3840729d Iustin Pop
      data["local_addr"] == local and
68 3840729d Iustin Pop
      "remote_addr" in data and
69 3840729d Iustin Pop
      data["remote_addr"] == remote
70 3840729d Iustin Pop
      )
71 3840729d Iustin Pop
    return retval
72 3840729d Iustin Pop
73 3840729d Iustin Pop
  def testParserCreation(self):
74 3840729d Iustin Pop
    """Test drbdsetup show parser creation"""
75 3840729d Iustin Pop
    bdev.DRBD8._GetShowParser()
76 3840729d Iustin Pop
77 3840729d Iustin Pop
  def testParserBoth(self):
78 3840729d Iustin Pop
    """Test drbdsetup show parser for disk and network"""
79 6634816b Iustin Pop
    data = self._get_contents("data/bdev-both.txt")
80 3840729d Iustin Pop
    result = bdev.DRBD8._GetDevInfo(data)
81 3840729d Iustin Pop
    self.failUnless(self._has_disk(result, "/dev/xenvg/test.data",
82 3840729d Iustin Pop
                                   "/dev/xenvg/test.meta"),
83 3840729d Iustin Pop
                    "Wrong local disk info")
84 3840729d Iustin Pop
    self.failUnless(self._has_net(result, ("192.168.1.1", 11000),
85 3840729d Iustin Pop
                                  ("192.168.1.2", 11000)),
86 3840729d Iustin Pop
                    "Wrong network info")
87 3840729d Iustin Pop
88 3840729d Iustin Pop
  def testParserNet(self):
89 3840729d Iustin Pop
    """Test drbdsetup show parser for disk and network"""
90 6634816b Iustin Pop
    data = self._get_contents("data/bdev-net.txt")
91 3840729d Iustin Pop
    result = bdev.DRBD8._GetDevInfo(data)
92 3840729d Iustin Pop
    self.failUnless(("local_dev" not in result and
93 3840729d Iustin Pop
                     "meta_dev" not in result and
94 3840729d Iustin Pop
                     "meta_index" not in result),
95 3840729d Iustin Pop
                    "Should not find local disk info")
96 3840729d Iustin Pop
    self.failUnless(self._has_net(result, ("192.168.1.1", 11002),
97 3840729d Iustin Pop
                                  ("192.168.1.2", 11002)),
98 3840729d Iustin Pop
                    "Wrong network info")
99 3840729d Iustin Pop
100 3840729d Iustin Pop
  def testParserDisk(self):
101 3840729d Iustin Pop
    """Test drbdsetup show parser for disk and network"""
102 6634816b Iustin Pop
    data = self._get_contents("data/bdev-disk.txt")
103 3840729d Iustin Pop
    result = bdev.DRBD8._GetDevInfo(data)
104 3840729d Iustin Pop
    self.failUnless(self._has_disk(result, "/dev/xenvg/test.data",
105 3840729d Iustin Pop
                                   "/dev/xenvg/test.meta"),
106 3840729d Iustin Pop
                    "Wrong local disk info")
107 3840729d Iustin Pop
    self.failUnless(("local_addr" not in result and
108 3840729d Iustin Pop
                     "remote_addr" not in result),
109 3840729d Iustin Pop
                    "Should not find network info")
110 3840729d Iustin Pop
111 6b90c22e Iustin Pop
112 6b90c22e Iustin Pop
class TestDRBD8Status(unittest.TestCase):
113 6b90c22e Iustin Pop
  """Testing case for DRBD8 /proc status"""
114 6b90c22e Iustin Pop
115 6b90c22e Iustin Pop
  def setUp(self):
116 6b90c22e Iustin Pop
    """Read in txt data"""
117 e8230860 Michael Hanselmann
    proc_data = "test/data/proc_drbd8.txt"
118 ae9da390 Iustin Pop
    prefix = os.environ.get("srcdir", None)
119 ae9da390 Iustin Pop
    if prefix:
120 ae9da390 Iustin Pop
      proc_data = prefix + "/" + proc_data
121 ae9da390 Iustin Pop
    self.proc_data = bdev.DRBD8._GetProcData(filename=proc_data)
122 6b90c22e Iustin Pop
    self.mass_data = bdev.DRBD8._MassageProcData(self.proc_data)
123 6b90c22e Iustin Pop
124 6b90c22e Iustin Pop
  def testMinorNotFound(self):
125 6b90c22e Iustin Pop
    """Test not-found-minor in /proc"""
126 6b90c22e Iustin Pop
    self.failUnless(9 not in self.mass_data)
127 6b90c22e Iustin Pop
128 6b90c22e Iustin Pop
  def testLineNotMatch(self):
129 6b90c22e Iustin Pop
    """Test wrong line passed to DRBD8Status"""
130 6b90c22e Iustin Pop
    self.assertRaises(errors.BlockDeviceError, bdev.DRBD8Status, "foo")
131 6b90c22e Iustin Pop
132 6b90c22e Iustin Pop
  def testMinor0(self):
133 6b90c22e Iustin Pop
    """Test connected, primary device"""
134 6b90c22e Iustin Pop
    stats = bdev.DRBD8Status(self.mass_data[0])
135 6b90c22e Iustin Pop
    self.failUnless(stats.is_connected and stats.is_primary and
136 6b90c22e Iustin Pop
                    stats.peer_secondary and stats.is_disk_uptodate)
137 6b90c22e Iustin Pop
138 6b90c22e Iustin Pop
  def testMinor1(self):
139 6b90c22e Iustin Pop
    """Test connected, secondary device"""
140 6b90c22e Iustin Pop
    stats = bdev.DRBD8Status(self.mass_data[1])
141 6b90c22e Iustin Pop
    self.failUnless(stats.is_connected and stats.is_secondary and
142 6b90c22e Iustin Pop
                    stats.peer_primary and stats.is_disk_uptodate)
143 6b90c22e Iustin Pop
144 6b90c22e Iustin Pop
  def testMinor4(self):
145 6b90c22e Iustin Pop
    """Test WFconn device"""
146 6b90c22e Iustin Pop
    stats = bdev.DRBD8Status(self.mass_data[4])
147 6b90c22e Iustin Pop
    self.failUnless(stats.is_wfconn and stats.is_primary and
148 6b90c22e Iustin Pop
                    stats.rrole == 'Unknown' and
149 6b90c22e Iustin Pop
                    stats.is_disk_uptodate)
150 6b90c22e Iustin Pop
151 6b90c22e Iustin Pop
  def testMinor6(self):
152 6b90c22e Iustin Pop
    """Test diskless device"""
153 6b90c22e Iustin Pop
    stats = bdev.DRBD8Status(self.mass_data[6])
154 6b90c22e Iustin Pop
    self.failUnless(stats.is_connected and stats.is_secondary and
155 6b90c22e Iustin Pop
                    stats.peer_primary and stats.is_diskless)
156 6b90c22e Iustin Pop
157 6b90c22e Iustin Pop
  def testMinor8(self):
158 6b90c22e Iustin Pop
    """Test standalone device"""
159 6b90c22e Iustin Pop
    stats = bdev.DRBD8Status(self.mass_data[8])
160 6b90c22e Iustin Pop
    self.failUnless(stats.is_standalone and
161 6b90c22e Iustin Pop
                    stats.rrole == 'Unknown' and
162 6b90c22e Iustin Pop
                    stats.is_disk_uptodate)
163 6b90c22e Iustin Pop
164 3840729d Iustin Pop
if __name__ == '__main__':
165 3840729d Iustin Pop
  unittest.main()