forked from ungleich-public/cdist
		
	return output of command execution as string instead of bytestring
Signed-off-by: Steven Armstrong <steven@icarus.ethz.ch>
This commit is contained in:
		
					parent
					
						
							
								78fd611bb0
							
						
					
				
			
			
				commit
				
					
						a254e1f31e
					
				
			
		
					 4 changed files with 6 additions and 6 deletions
				
			
		| 
						 | 
					@ -93,7 +93,7 @@ class Local(object):
 | 
				
			||||||
        assert isinstance(command, (list, tuple)), "list or tuple argument expected, got: %s" % command
 | 
					        assert isinstance(command, (list, tuple)), "list or tuple argument expected, got: %s" % command
 | 
				
			||||||
        self.log.debug("Local run: %s", command)
 | 
					        self.log.debug("Local run: %s", command)
 | 
				
			||||||
        try:
 | 
					        try:
 | 
				
			||||||
            return subprocess.check_output(command, env=env)
 | 
					            return subprocess.check_output(command, env=env).decode()
 | 
				
			||||||
        except subprocess.CalledProcessError:
 | 
					        except subprocess.CalledProcessError:
 | 
				
			||||||
            raise cdist.Error("Command failed: " + " ".join(command))
 | 
					            raise cdist.Error("Command failed: " + " ".join(command))
 | 
				
			||||||
        except OSError as error:
 | 
					        except OSError as error:
 | 
				
			||||||
| 
						 | 
					@ -112,7 +112,7 @@ class Local(object):
 | 
				
			||||||
            self.log.debug("Local run script env: %s", env)
 | 
					            self.log.debug("Local run script env: %s", env)
 | 
				
			||||||
        
 | 
					        
 | 
				
			||||||
        try:
 | 
					        try:
 | 
				
			||||||
            return subprocess.check_output(command, env=env)
 | 
					            return subprocess.check_output(command, env=env).decode()
 | 
				
			||||||
        except subprocess.CalledProcessError as error:
 | 
					        except subprocess.CalledProcessError as error:
 | 
				
			||||||
            script_content = self.run(["cat", script])
 | 
					            script_content = self.run(["cat", script])
 | 
				
			||||||
            self.log.error("Code that raised the error:\n%s", script_content)
 | 
					            self.log.error("Code that raised the error:\n%s", script_content)
 | 
				
			||||||
| 
						 | 
					
 | 
				
			||||||
| 
						 | 
					@ -104,7 +104,7 @@ class Remote(object):
 | 
				
			||||||
        assert isinstance(command, (list, tuple)), "list or tuple argument expected, got: %s" % command
 | 
					        assert isinstance(command, (list, tuple)), "list or tuple argument expected, got: %s" % command
 | 
				
			||||||
        self.log.debug("Remote run: %s", command)
 | 
					        self.log.debug("Remote run: %s", command)
 | 
				
			||||||
        try:
 | 
					        try:
 | 
				
			||||||
            return subprocess.check_output(command, env=env)
 | 
					            return subprocess.check_output(command, env=env).decode()
 | 
				
			||||||
        except subprocess.CalledProcessError:
 | 
					        except subprocess.CalledProcessError:
 | 
				
			||||||
            raise cdist.Error("Command failed: " + " ".join(command))
 | 
					            raise cdist.Error("Command failed: " + " ".join(command))
 | 
				
			||||||
        except OSError as error:
 | 
					        except OSError as error:
 | 
				
			||||||
| 
						 | 
					@ -125,7 +125,7 @@ class Remote(object):
 | 
				
			||||||
            self.log.debug("Remote run script env: %s", env)
 | 
					            self.log.debug("Remote run script env: %s", env)
 | 
				
			||||||
        
 | 
					        
 | 
				
			||||||
        try:
 | 
					        try:
 | 
				
			||||||
            return subprocess.check_output(command, env=env)
 | 
					            return subprocess.check_output(command, env=env).decode()
 | 
				
			||||||
        except subprocess.CalledProcessError as error:
 | 
					        except subprocess.CalledProcessError as error:
 | 
				
			||||||
            script_content = self.run(["cat", script])
 | 
					            script_content = self.run(["cat", script])
 | 
				
			||||||
            self.log.error("Code that raised the error:\n%s", script_content)
 | 
					            self.log.error("Code that raised the error:\n%s", script_content)
 | 
				
			||||||
| 
						 | 
					
 | 
				
			||||||
| 
						 | 
					@ -81,7 +81,7 @@ class LocalTestCase(unittest.TestCase):
 | 
				
			||||||
        fd = open(script, "w")
 | 
					        fd = open(script, "w")
 | 
				
			||||||
        fd.writelines(["#!/bin/sh\n", "echo foobar"])
 | 
					        fd.writelines(["#!/bin/sh\n", "echo foobar"])
 | 
				
			||||||
        fd.close()
 | 
					        fd.close()
 | 
				
			||||||
        self.assertEqual(self.local.run_script(script), b"foobar\n")
 | 
					        self.assertEqual(self.local.run_script(script), "foobar\n")
 | 
				
			||||||
 | 
					
 | 
				
			||||||
    def test_mkdir(self):
 | 
					    def test_mkdir(self):
 | 
				
			||||||
        temp_dir = self.mkdtemp(dir=self.temp_dir)
 | 
					        temp_dir = self.mkdtemp(dir=self.temp_dir)
 | 
				
			||||||
| 
						 | 
					
 | 
				
			||||||
| 
						 | 
					@ -76,7 +76,7 @@ class RemoteTestCase(unittest.TestCase):
 | 
				
			||||||
        fd = open(script, "w")
 | 
					        fd = open(script, "w")
 | 
				
			||||||
        fd.writelines(["#!/bin/sh\n", "echo foobar"])
 | 
					        fd.writelines(["#!/bin/sh\n", "echo foobar"])
 | 
				
			||||||
        fd.close()
 | 
					        fd.close()
 | 
				
			||||||
        self.assertEqual(self.remote.run_script(script), b"foobar\n")
 | 
					        self.assertEqual(self.remote.run_script(script), "foobar\n")
 | 
				
			||||||
 | 
					
 | 
				
			||||||
    def test_mkdir(self):
 | 
					    def test_mkdir(self):
 | 
				
			||||||
        temp_dir = self.mkdtemp(dir=self.temp_dir)
 | 
					        temp_dir = self.mkdtemp(dir=self.temp_dir)
 | 
				
			||||||
| 
						 | 
					
 | 
				
			||||||
		Loading…
	
	Add table
		Add a link
		
	
		Reference in a new issue