def asdf_test_censorship_provider_creation(self): import os import tempfile from evilgenius.resources import CensorshipProvider from evilgenius.vagrant import VagrantController, VagrantFile path = os.path.abspath( os.path.join(os.path.dirname(__file__), '..', 'resources', 'censorship-providers', 'dns-censorship', 'dns-censorship.yml')) vagrant_dir = tempfile.mkdtemp() cont = VagrantController(root=vagrant_dir) cens = CensorshipProvider(descriptor_path=path, controller=cont) vfile = VagrantFile(boxes=[cens.box], network=None) with open(os.path.join(vagrant_dir, 'Vagrantfile'), 'wb') as f: f.write(vfile.content) cont.create_box(box=cens.box) self.assertTrue(cens.box is not None) self.assertTrue(len(cont.status().keys()) > 0)
def setUp(self): self.tempdir = tempfile.mkdtemp() self.cont = VagrantController(root=self.tempdir)
class TestVagrantInterface(TestCase): def setUp(self): self.tempdir = tempfile.mkdtemp() self.cont = VagrantController(root=self.tempdir) def test_vagrant_init(self): self.cont.init() with open(os.path.join(self.tempdir, 'Vagrantfile')) as f: self.assertGreater(len(f.read()), 0) def test_vagrant_status(self): def mock_status_output(command): return (0, [ 'Current machine states:\n', '\n', 'default not created (virtualbox)\n', '\n', 'The environment has not yet been created. Run `vagrant up` to\n', 'create the environment. If a machine is not created, only the\n', 'default provider will be shown. So if a provider is not listed,\n', 'then the machine is not created for that environment.\n' ]) self.cont._vagrant = mock_status_output self.assertEqual(self.cont.status(), {'default': 'not created'}) self.assertEqual(self.cont.status(vm='default'), 'not created') def test_vagrant_ssh(self): self.cont.init() self.cont.up() output = self.cont.run_command("echo 'spamham'") self.assertEqual(''.join(output), "spamham\n") self.cont.destroy()
class TestVagrantInterface(TestCase): def setUp(self): self.tempdir = tempfile.mkdtemp() self.cont = VagrantController(root=self.tempdir) def test_vagrant_init(self): self.cont.init() with open(os.path.join(self.tempdir, "Vagrantfile")) as f: self.assertGreater(len(f.read()), 0) def test_vagrant_status(self): def mock_status_output(command): return ( 0, [ "Current machine states:\n", "\n", "default not created (virtualbox)\n", "\n", "The environment has not yet been created. Run `vagrant up` to\n", "create the environment. If a machine is not created, only the\n", "default provider will be shown. So if a provider is not listed,\n", "then the machine is not created for that environment.\n", ], ) self.cont._vagrant = mock_status_output self.assertEqual(self.cont.status(), {"default": "not created"}) self.assertEqual(self.cont.status(vm="default"), "not created") def test_vagrant_ssh(self): self.cont.init() self.cont.up() output = self.cont.run_command("echo 'spamham'") self.assertEqual("".join(output), "spamham\n") self.cont.destroy()