def test_restore(self):
"""Restore a project."""
factory = stubs.Factory()
fp = filepath.FilePath(self.mktemp())
fp.setContent(CONFIG1)
config = configfile.ConfigFile()
config.restore(factory, fp)
self.assertIsNotNone(factory.get_brick_by_name("sender"))
def test_save(self):
"""Save a project."""
factory = stubs.Factory()
factory.new_brick("switch", "sw")
fp = filepath.FilePath(self.mktemp())
config = configfile.ConfigFile()
config.save(factory, fp)
self.assertEqual(fp.getContent(), "[Switch:sw]\n\n")
def test_restore(self):
"""Restore a project."""
#NB -> Twisted FilePath returns files opened in binary mode
factory = stubs.Factory()
fp = filepath.FilePath(self.mktemp())
fp.setContent(file_bytes_from_text(CONFIG1))
config = configfile.ConfigFile()
config.restore(factory, fp)
self.assertIsNotNone(factory.get_brick_by_name("sender"))
def test_config_resume(self):
self.factory.del_brick(self.vm)
self.assertEqual(len(self.factory.bricks), 0)
sio = StringIO.StringIO(HOSTONLY_CONFIG)
configfile.ConfigFile().restore_from(self.factory, sio)
self.assertEqual(len(self.factory.bricks), 1)
vm1 = self.factory.get_brick_by_name("vm")
self.assertEqual(len(vm1.plugs), 1)
plug = vm1.plugs[0]
self.assertEqual(plug.mac, "00:11:22:33:44:55")
self.assertIs(plug.sock, vm.hostonly_sock)
def test_config_save(self):
sio = StringIO.StringIO()
configfile.ConfigFile().save_to(self.factory, sio)
self.assertEqual(sio.getvalue(), HOSTONLY_CONFIG)