def setup(self): self.fh = WhateverIO() self.env = {} self.t = MultiTool(env=self.env, fh=self.fh) self.t.cluster_from_argv = Mock(name='cluster_from_argv') self.t._cluster_from_argv = Mock(name='cluster_from_argv') self.t.Cluster = Mock(name='Cluster') self.t.carp = Mock(name='.carp') self.t.usage = Mock(name='.usage') self.t.splash = Mock(name='.splash') self.t.say = Mock(name='.say') self.t.ok = Mock(name='.ok') self.cluster = self.t.Cluster.return_value def _cluster_from_argv(argv): p = self.t.OptionParser(argv) p.parse() return p, self.cluster self.t.cluster_from_argv.return_value = self.cluster self.t._cluster_from_argv.side_effect = _cluster_from_argv
def setup(self): self.out = WhateverIO() self.err = WhateverIO() self.cmd = Command(self.app, stdout=self.out, stderr=self.err)
def test_run(self): out = WhateverIO() r = report(app=self.app, stdout=out) self.assertEqual(r.run(), EX_OK) self.assertTrue(out.getvalue())
def test_list_bindings_no_support(self): l = list_(app=self.app, stderr=WhateverIO()) management = Mock() management.get_bindings.side_effect = NotImplementedError() with self.assertRaises(Error): l.list_bindings(management)
def test_to_dot(self): s = WhateverIO() self.graph1().to_dot(s) assert s.getvalue()
def setup(self): self.fh = WhateverIO() self.adm = self.create_adm() self.shell = AMQShell(connect=self.adm.connect, out=self.fh)
def setup(self): self.fh = WhateverIO() self.env = {} self.t = MultiTool(env=self.env, fh=self.fh)
def setup(self): self.out = WhateverIO() self.dumper = Dumper(out=self.out)
def test_run(self): out = WhateverIO() r = report(app=self.app, stdout=out) assert r.run() == EX_OK assert out.getvalue()
def _mocked(url, *args, **kwargs): response_data, headers = response_method(url) return addinfourl(WhateverIO(response_data), headers, url)
def setup(self): self.fh = WhateverIO() self.env = {} with patch('celery.apps.multi.os.mkdir'): self.t = MultiTool(env=self.env, fh=self.fh)