tests/parser/pre.py

   1 import unittest
   2 from tests.baseclass import *
   3
   4 from pykickstart import constants
   5 from pykickstart.errors import KickstartParseError
   6 from pykickstart import version
   7
   8 class Simple_Terminated_TestCase(ParserTest):
   9     ks = """
  10 %pre
  11 ls /tmp
  12 %end
  13 """
  14
  15     def runTest(self):
  16         self.parser.readKickstartFromString(self.ks)
  17         self.assertEqual(len(self.handler.scripts), 1)
  18
  19         # Verify the script defaults.
  20         script = self.handler.scripts[0]
  21         self.assertEqual(script.interp, "/bin/sh")
  22         self.assertFalse(script.inChroot)
  23         self.assertEqual(script.lineno, 2)
  24         self.assertFalse(script.errorOnFail)
  25         self.assertEqual(script.type, constants.KS_SCRIPT_PRE)
  26
  27         # Also verify the body, which is the most important part.
  28         self.assertEqual(script.script.rstrip(), "ls /tmp")
  29
  30 class Simple_Header_TestCase(ParserTest):
  31     ks = """
  32 %pre --interpreter /usr/bin/python --erroronfail --log=/tmp/blah
  33 ls /tmp
  34 %end
  35 """
  36
  37     def runTest(self):
  38         self.parser.readKickstartFromString(self.ks)
  39         self.assertEqual(len(self.handler.scripts), 1)
  40
  41         # Verify the changes we made in the header.
  42         script = self.handler.scripts[0]
  43         self.assertEqual(script.interp, "/usr/bin/python")
  44         self.assertFalse(script.inChroot)
  45         self.assertTrue(script.errorOnFail)
  46         self.assertEqual(script.lineno, 2)
  47         self.assertEqual(script.type, constants.KS_SCRIPT_PRE)
  48         self.assertEqual(script.logfile, "/tmp/blah")
  49
  50         # Also verify the body, which is the most important part.
  51         self.assertEqual(script.script.rstrip(), "ls /tmp")
  52
  53 class Pre_Given_Nochroot_TestCase(ParserTest):
  54     ks = """
  55 %pre --nochroot
  56 ls /tmp
  57 %end
  58 """
  59
  60     def runTest(self):
  61         self.assertRaises(KickstartParseError, self.parser.readKickstartFromString, self.ks)
  62
  63 class Multiple_Terminated_TestCase(ParserTest):
  64     ks = """
  65 %pre
  66 ls /tmp
  67 %end
  68
  69 %pre
  70 ls /var
  71 %end
  72 """
  73
  74     def runTest(self):
  75         self.parser.readKickstartFromString(self.ks)
  76         self.assertEqual(len(self.handler.scripts), 2)
  77
  78         # Verify the script defaults.
  79         script = self.handler.scripts[0]
  80         self.assertEqual(script.interp, "/bin/sh")
  81         self.assertFalse(script.inChroot)
  82         self.assertFalse(script.errorOnFail)
  83         self.assertEqual(script.lineno, 2)
  84         self.assertEqual(script.type, constants.KS_SCRIPT_PRE)
  85
  86         # Also verify the body, which is the most important part.
  87         self.assertEqual(script.script.rstrip(), "ls /tmp")
  88
  89         script = self.handler.scripts[1]
  90         self.assertEqual(script.script.rstrip(), "ls /var")
  91
  92 class Simple_Unterminated_TestCase(Simple_Terminated_TestCase):
  93     version = version.F7
  94
  95     ks = """
  96 %pre
  97 ls /tmp
  98 """
  99
 100 class Simple_Unterminated_Fails_TestCase(Simple_Unterminated_TestCase):
 101     version = version.F8
 102
 103     def runTest(self):
 104         self.assertRaises(KickstartParseError, self.parser.readKickstartFromString, self.ks)
 105
 106 class Multiple_Unterminated_TestCase(Multiple_Terminated_TestCase):
 107     version = version.F7
 108
 109     ks = """
 110 %pre
 111 ls /tmp
 112
 113 %pre
 114 ls /var
 115 """
 116
 117 class Multiple_Unterminated_Fails_TestCase(Multiple_Unterminated_TestCase):
 118     version = version.F8
 119
 120     def runTest(self):
 121         self.assertRaises(KickstartParseError, self.parser.readKickstartFromString, self.ks)
 122
 123 if __name__ == "__main__":
 124     unittest.main()