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")