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 setUp(self): self.override = {} self.override['STORE'] = '' self.override['DUMP'] = '' self.parser = GruntParser(StringReader(""), self.override)