def testBasic(self):
self.setupStep(
DownloadSecretsToWorker([
(os.path.join(self.temp_path, "pathA"), "something"),
(os.path.join(self.temp_path, "pathB"), "something more")
]))
args1 = {
'maxsize': None,
'mode': None,
'reader': ExpectRemoteRef(remotetransfer.StringFileReader),
'blocksize': 32 * 1024,
'workerdest': os.path.join(self.temp_path, "pathA"),
'workdir': "wkdir"
}
args2 = {
'maxsize': None,
'mode': None,
'reader': ExpectRemoteRef(remotetransfer.StringFileReader),
'blocksize': 32 * 1024,
'workerdest': os.path.join(self.temp_path, "pathB"),
'workdir': "wkdir"
}
self.expectCommands(
Expect('downloadFile', args1) + 0,
Expect('downloadFile', args2) + 0,
)
self.expectOutcome(result=SUCCESS, state_string="finished")
d = self.runStep()
return d
def testBasic(self):
self.setup_step(
DownloadSecretsToWorker([
(os.path.join(self.temp_path, "pathA"), "something"),
(os.path.join(self.temp_path, "pathB"), "something more")
]))
self.expect_commands(
ExpectDownloadFile(
maxsize=None,
mode=stat.S_IRUSR | stat.S_IWUSR,
reader=ExpectRemoteRef(remotetransfer.StringFileReader),
blocksize=32 * 1024,
workerdest=os.path.join(self.temp_path, "pathA"),
workdir="wkdir").exit(0),
ExpectDownloadFile(
maxsize=None,
mode=stat.S_IRUSR | stat.S_IWUSR,
reader=ExpectRemoteRef(remotetransfer.StringFileReader),
blocksize=32 * 1024,
workerdest=os.path.join(self.temp_path, "pathB"),
workdir="wkdir").exit(0),
)
self.expect_outcome(result=SUCCESS, state_string="finished")
d = self.run_step()
return d
def addSteps(self, steps, withSecrets=[]):
if withSecrets:
self.addStep(DownloadSecretsToWorker(withSecrets))
for s in steps:
self.addStep(s)
if withSecrets:
self.addStep(RemoveWorkerFileSecret(withSecrets))
def withSecrets(self, secrets):
self.addStep(DownloadSecretsToWorker(secrets))
yield self
self.addStep(RemoveWorkerFileSecret(secrets))
