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