def testLongLine(self):
dummydata = (1024 * "Hello World! ")
myfile = self._CreateTempFile()
utils.WriteFile(myfile, data=dummydata)
datastrict = utils.ReadOneLineFile(myfile, strict=True)
datalax = utils.ReadOneLineFile(myfile, strict=False)
self.assertEqual(dummydata, datastrict)
self.assertEqual(dummydata, datalax)
def testNewline(self):
myfile = self._CreateTempFile()
myline = "myline"
for nl in ["", "\n", "\r\n"]:
dummydata = "%s%s" % (myline, nl)
utils.WriteFile(myfile, data=dummydata)
datalax = utils.ReadOneLineFile(myfile, strict=False)
self.assertEqual(myline, datalax)
datastrict = utils.ReadOneLineFile(myfile, strict=True)
self.assertEqual(myline, datastrict)
def testEmptylines(self):
myfile = self._CreateTempFile()
myline = "myline"
for nl in ["\n", "\r\n"]:
for ol in ["", "otherline"]:
dummydata = "%s%s%s%s%s%s" % (nl, nl, myline, nl, ol, nl)
utils.WriteFile(myfile, data=dummydata)
self.assert_(set("\r\n") & set(dummydata))
datalax = utils.ReadOneLineFile(myfile, strict=False)
self.assertEqual(myline, datalax)
if ol:
self.assertRaises(errors.GenericError, utils.ReadOneLineFile,
myfile, strict=True)
else:
datastrict = utils.ReadOneLineFile(myfile, strict=True)
self.assertEqual(myline, datastrict)
def testWhitespaceAndMultipleLines(self):
myfile = self._CreateTempFile()
for nl in ["", "\n", "\r\n"]:
for ws in [" ", "\t", "\t\t \t", "\t "]:
dummydata = (1024 * ("Foo bar baz %s%s" % (ws, nl)))
utils.WriteFile(myfile, data=dummydata)
datalax = utils.ReadOneLineFile(myfile, strict=False)
if nl:
self.assert_(set("\r\n") & set(dummydata))
self.assertRaises(errors.GenericError, utils.ReadOneLineFile,
myfile, strict=True)
explen = len("Foo bar baz ") + len(ws)
self.assertEqual(len(datalax), explen)
self.assertEqual(datalax, dummydata[:explen])
self.assertFalse(set("\r\n") & set(datalax))
else:
datastrict = utils.ReadOneLineFile(myfile, strict=True)
self.assertEqual(dummydata, datastrict)
self.assertEqual(dummydata, datalax)
def testWrongHostname(self):
self._CreateValidConfigDir()
utils.WriteFile(self.config_path,
data=serializer.DumpJson(GetMinimalConfig()))
hostname = netutils.GetHostname().name
assert hostname != utils.ReadOneLineFile(self.ss_master_node_path)
self.assertRaises(Exception, _RunUpgrade, self.tmpdir, False, True,
ignore_hostname=False)
def CheckHostname(path):
"""Ensures hostname matches ssconf value.
@param path: Path to ssconf file
"""
ssconf_master_node = utils.ReadOneLineFile(path)
hostname = netutils.GetHostname().name
if ssconf_master_node == hostname:
return True
logging.warning("Warning: ssconf says master node is '%s', but this"
" machine's name is '%s'; this tool must be run on"
" the master node", ssconf_master_node, hostname)
return False
def testNotStrict(self):
data = utils.ReadOneLineFile(testutils.TestDataFilename("cert1.pem"),
strict=False)
self.assertEqual(len(data), 27)
self.assertEqual(data, "-----BEGIN CERTIFICATE-----")
def testDefault(self):
data = utils.ReadOneLineFile(testutils.TestDataFilename("cert1.pem"))
self.assertEqual(len(data), 27)
self.assertEqual(data, "-----BEGIN CERTIFICATE-----")
def GetClusterDomainSecret():
"""Reads the cluster domain secret.
"""
return utils.ReadOneLineFile(pathutils.CLUSTER_DOMAIN_SECRET_FILE,
strict=True)