Revision 6634816b test/ganeti.bdev_unittest.py

b/test/ganeti.bdev_unittest.py
22 22
"""Script for unittesting the bdev module"""
23 23

  
24 24

  
25
import os
25 26
import unittest
26 27

  
27 28
from ganeti import bdev
......
44 45
    return retval
45 46

  
46 47
  @staticmethod
48
  def _get_contents(name):
49
    """Returns the contents of a file"""
50

  
51
    prefix = os.environ.get("srcdir", None)
52
    if prefix:
53
      name = prefix + "/" + name
54
    fh = open(name, "r")
55
    try:
56
      data = fh.read()
57
    finally:
58
      fh.close()
59
    return data
60

  
61

  
62
  @staticmethod
47 63
  def _has_net(data, local, remote):
48 64
    """Check network connection parameters"""
49 65
    retval = (
......
60 76

  
61 77
  def testParserBoth(self):
62 78
    """Test drbdsetup show parser for disk and network"""
63
    data = open("data/bdev-both.txt").read()
79
    data = self._get_contents("data/bdev-both.txt")
64 80
    result = bdev.DRBD8._GetDevInfo(data)
65 81
    self.failUnless(self._has_disk(result, "/dev/xenvg/test.data",
66 82
                                   "/dev/xenvg/test.meta"),
......
71 87

  
72 88
  def testParserNet(self):
73 89
    """Test drbdsetup show parser for disk and network"""
74
    data = open("data/bdev-net.txt").read()
90
    data = self._get_contents("data/bdev-net.txt")
75 91
    result = bdev.DRBD8._GetDevInfo(data)
76 92
    self.failUnless(("local_dev" not in result and
77 93
                     "meta_dev" not in result and
......
83 99

  
84 100
  def testParserDisk(self):
85 101
    """Test drbdsetup show parser for disk and network"""
86
    data = open("data/bdev-disk.txt").read()
102
    data = self._get_contents("data/bdev-disk.txt")
87 103
    result = bdev.DRBD8._GetDevInfo(data)
88 104
    self.failUnless(self._has_disk(result, "/dev/xenvg/test.data",
89 105
                                   "/dev/xenvg/test.meta"),

Also available in: Unified diff