Exemplo n.º 1
0
Arquivo: test.py Projeto: b4lmung/edb
 def setUp(self):
     self.client = Client()
     self.ip1_ptxt = b'127.0.0.1'
     self.ip2_ptxt = b'128.66.0.0'
     self.ip3_ptxt = b'192.0.2.0'
     self.ip1 = self.client.encrypt(self.ip1_ptxt)
     self.ip2 = self.client.encrypt(self.ip2_ptxt)
     self.ip3 = self.client.encrypt(self.ip3_ptxt)
     _Ping.objects.create(source=self.ip1, destination=self.ip2)
     _Ping.objects.create(source=self.ip1, destination=self.ip3)
     _Ping.objects.create(source=self.ip2, destination=self.ip3)
Exemplo n.º 2
0
 def test_keyfile(self):
     keyinfo = crypto.generate_keyinfo(Client.KEY_SCHEMA)
     tmpdir = tempfile.mkdtemp()
     try:
         filename = os.path.join(tmpdir, '.keyinfo')
         crypto.write_keyinfo(keyinfo, filename)
         client = Client(filename)
         ptxt = b"test"
         ctxt = client.encrypt(ptxt)
         self.assertEqual(ptxt, client.decrypt(ctxt))
     finally:
         shutil.rmtree(tmpdir)
Exemplo n.º 3
0
 def setUp(self):
     self.client = Client()