def registerScript(self, file_name, alias_overrides):
"""
Parses and registers the pig script.
file_name: The Pig script file.
alias_overrides: The list of aliases to override in the Pig script.
"""
grunt = GruntParser(FileReader(File(file_name)), alias_overrides)
grunt.setInteractive(False)
grunt.setParams(self)
grunt.parseStopOnError(True)
class TestPigServer(unittest.TestCase):
parser = None
override = None
def setUp(self):
self.override = {}
self.override['STORE'] = ''
self.override['DUMP'] = ''
self.parser = GruntParser(StringReader(""), self.override)
def testRemoveStores(self):
self.assertEquals("", self.parser.override("STORE output INTO '/path';"))
del self.override["STORE"]
self.assertEquals("STORE output INTO '/path';", self.parser.override("STORE output INTO '/path';"))
def testRemoveDumps(self):
self.assertEquals("", self.parser.override("DUMP output;"))
del self.override["DUMP"]
self.assertEquals("DUMP output;", self.parser.override("DUMP output;"))
def testReplaceLoad(self):
self.override["A"] = "A = LOAD 'file';"
self.assertEquals("A = LOAD 'file';",
self.parser.override("A = LOAD 'input.txt' AS (query:CHARARRAY);"))
def testGetStoreAlias(self):
del self.override["STORE"]
self.parser.override("STORE output INTO '/path'")
self.assertEquals("output", self.override.get("LAST_STORE_ALIAS"))
class TestPigServer(unittest.TestCase):
parser = None
override = None
def setUp(self):
self.override = {}
self.override['STORE'] = ''
self.override['DUMP'] = ''
self.parser = GruntParser(StringReader(""), self.override)
def testRemoveStores(self):
self.assertEquals("",
self.parser.override("STORE output INTO '/path';"))
del self.override["STORE"]
self.assertEquals("STORE output INTO '/path';",
self.parser.override("STORE output INTO '/path';"))
def testRemoveDumps(self):
self.assertEquals("", self.parser.override("DUMP output;"))
del self.override["DUMP"]
self.assertEquals("DUMP output;", self.parser.override("DUMP output;"))
def testReplaceLoad(self):
self.override["A"] = "A = LOAD 'file';"
self.assertEquals(
"A = LOAD 'file';",
self.parser.override("A = LOAD 'input.txt' AS (query:CHARARRAY);"))
def testGetStoreAlias(self):
del self.override["STORE"]
self.parser.override("STORE output INTO '/path'")
self.assertEquals("output", self.override.get("LAST_STORE_ALIAS"))
def registerScript(self, file_name, alias_overrides):
"""
Parses and registers the pig script.
file_name: The Pig script file.
alias_overrides: The list of aliases to override in the Pig script.
"""
grunt = GruntParser(FileReader(File(file_name)), alias_overrides)
grunt.setInteractive(False)
grunt.setParams(self)
grunt.parseStopOnError(True)
def setUp(self):
self.override = {}
self.override['STORE'] = ''
self.override['DUMP'] = ''
self.parser = GruntParser(StringReader(""), self.override)
def setUp(self):
self.override = {}
self.override['STORE'] = ''
self.override['DUMP'] = ''
self.parser = GruntParser(StringReader(""), self.override)
