Statistics
| Branch: | Tag: | Revision:

root / test / import-export_unittest-helper @ 53dbf14c

History | View | Annotate | Download (2.5 kB)

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

    
4
# Copyright (C) 2010 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
"""Helpers for testing import-export daemon"""
23

    
24
import os
25
import sys
26
import errno
27
import time
28

    
29
from ganeti import constants
30
from ganeti import utils
31
from ganeti import objects
32
from ganeti import serializer
33

    
34

    
35
RETRY_INTERVAL = 0.1
36
TIMEOUT = int(os.getenv("TIMEOUT", 10))
37
VALIDITY = int(os.getenv("VALIDITY", 1))
38

    
39

    
40
def Log(msg, *args):
41
  if args:
42
    line = msg % args
43
  else:
44
    line = msg
45

    
46
  sys.stderr.write("%0.6f, pid %s: %s\n" % (time.time(), os.getpid(), line))
47
  sys.stderr.flush()
48

    
49

    
50
def _GetImportExportData(filename):
51
  try:
52
    data = utils.ReadFile(filename)
53
  except EnvironmentError, err:
54
    Log("%s = %s", filename, err)
55
    if err.errno != errno.ENOENT:
56
      raise
57
    raise utils.RetryAgain()
58

    
59
  Log("%s = %s", filename, data.strip())
60

    
61
  return objects.ImportExportStatus.FromDict(serializer.LoadJson(data))
62

    
63

    
64
def _CheckConnected(filename):
65
  if not _GetImportExportData(filename).connected:
66
    Log("Not connected")
67
    raise utils.RetryAgain()
68

    
69
  Log("Connected")
70

    
71

    
72
def _CheckListenPort(filename):
73
  port = _GetImportExportData(filename).listen_port
74
  if not port:
75
    Log("No port")
76
    raise utils.RetryAgain()
77
  Log("Listening on %s", port)
78
  return port
79

    
80

    
81
def WaitForListenPort(filename):
82
  return utils.Retry(_CheckListenPort, RETRY_INTERVAL, TIMEOUT,
83
                     args=(filename, ))
84

    
85

    
86
def WaitForConnected(filename):
87
  utils.Retry(_CheckConnected, RETRY_INTERVAL, TIMEOUT, args=(filename, ))
88

    
89

    
90
def main():
91
  (filename, what) = sys.argv[1:]
92

    
93
  if what == "listen-port":
94
    print WaitForListenPort(filename)
95
  elif what == "connected":
96
    WaitForConnected(filename)
97
  elif what == "gencert":
98
    utils.GenerateSelfSignedSslCert(filename, validity=VALIDITY)
99
  else:
100
    raise Exception("Unknown command '%s'" % what)
101

    
102

    
103
if __name__ == "__main__":
104
  main()