Synnefo » snf-ganeti

from cStringIO import StringIO

class TestShellWriter(unittest.TestCase):

  def test(self):

    buf = StringIO()

    sw = utils.ShellWriter(buf)

    sw.Write("#!/bin/bash")

    sw.Write("if true; then")

    sw.IncIndent()

    try:

      sw.Write("echo true")

      sw.Write("for i in 1 2 3")

      sw.Write("do")

      sw.IncIndent()

      try:

        self.assertEqual(sw._indent, 2)

        sw.Write("date")

      finally:

        sw.DecIndent()

      sw.Write("done")

    finally:

      sw.DecIndent()

    sw.Write("echo %s", utils.ShellQuote("Hello World"))

    sw.Write("exit 0")

    self.assertEqual(sw._indent, 0)

    output = buf.getvalue()

    self.assert_(output.endswith("\n"))

    lines = output.splitlines()

    self.assertEqual(len(lines), 9)

    self.assertEqual(lines[0], "#!/bin/bash")

    self.assert_(re.match(r"^\s+date$", lines[5]))

    self.assertEqual(lines[7], "echo 'Hello World'")

  def testEmpty(self):

    buf = StringIO()

    sw = utils.ShellWriter(buf)

    sw = None

    self.assertEqual(buf.getvalue(), "")