def testBasicWorker2_16(self):
self.setupStep(
transfer.StringDownload("Hello World", "hello.txt"),
worker_version={'*': '2.16'})
self.step.worker = Mock()
self.step.remote = Mock()
# A place to store what gets read
read = []
self.expectCommands(
Expect('downloadFile', dict(
slavedest="hello.txt", workdir='wkdir',
blocksize=16384, maxsize=None, mode=None,
reader=ExpectRemoteRef(remotetransfer.StringFileReader)))
+ Expect.behavior(downloadString(read.append))
+ 0)
self.expectOutcome(
result=SUCCESS, state_string="downloading to hello.txt")
yield self.runStep()
self.assertEqual(b''.join(read), b"Hello World")
def testBasic(self):
self.setupStep(transfer.StringDownload("Hello World", "hello.txt"))
self.step.worker = Mock()
self.step.remote = Mock()
# A place to store what gets read
read = []
self.expectCommands(
Expect('downloadFile', dict(
slavedest="hello.txt", workdir='wkdir',
blocksize=16384, maxsize=None, mode=None,
reader=ExpectRemoteRef(remotetransfer.StringFileReader)))
+ Expect.behavior(downloadString(read.append))
+ 0)
self.expectOutcome(result=SUCCESS, state_string="downloading to hello.txt")
d = self.runStep()
@d.addCallback
def checkCalls(res):
self.assertEquals(''.join(read), "Hello World")
return d
def testBasic(self):
s = transfer.StringDownload("Hello World", "hello.txt")
s.build = Mock()
s.build.getProperties.return_value = Properties()
s.build.getSlaveCommandVersion.return_value = 1
s.step_status = Mock()
s.buildslave = Mock()
s.remote = Mock()
s.start()
for c in s.remote.method_calls:
name, command, args = c
commandName = command[3]
kwargs = command[-1]
if commandName == 'downloadFile':
self.assertEquals(kwargs['slavedest'], 'hello.txt')
reader = kwargs['reader']
data = reader.remote_read(100)
self.assertEquals(data, "Hello World")
break
else:
self.assert_(False, "No downloadFile command found")
def test_init_positional_args(self):
self.assertRaises(TypeError, lambda: transfer.StringDownload())
self.assertRaises(TypeError,
lambda: transfer.StringDownload('srcfile'))
def test_init_workerdest_positional(self):
with assertNotProducesWarnings(DeprecatedWorkerAPIWarning):
step = transfer.StringDownload('srcfile', 'dstfile')
self.assertEqual(step.workerdest, 'dstfile')
def test_init_positional_args(self):
with self.assertRaises(TypeError):
transfer.StringDownload()
with self.assertRaises(TypeError):
transfer.StringDownload('srcfile')
def test_init_workerdest_positional(self):
step = transfer.StringDownload('srcfile', 'dstfile')
self.assertEqual(step.workerdest, 'dstfile')
def test_init_workerdest_keyword(self):
step = transfer.StringDownload('srcfile', workerdest='dstfile')
self.assertEqual(step.workerdest, 'dstfile')
def testModeConfError(self):
with self.assertRaisesRegex(config.ConfigErrors,
"StringDownload step's mode must be an integer or None,"
" got 'not-a-number'"):
transfer.StringDownload("string", "file", mode="not-a-number")
