예제 #1
0
파일: test.py 프로젝트: 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)
예제 #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)
예제 #3
0
파일: test.py 프로젝트: Pringley/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)
예제 #4
0
파일: test.py 프로젝트: b4lmung/edb
class EncryptedModelTestCase(TestCase):
    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)

    def test_encrypted_filter(self):
        query = self.client.query(self.ip1_ptxt)
        results = _Ping.objects.encrypted_filter(source=query)

        self.assertEqual(2, len(results))
        dests = [result.destination for result in results]
        self.assertIn(self.ip2, dests)
        self.assertIn(self.ip3, dests)
예제 #5
0
파일: test.py 프로젝트: Pringley/edb
class EncryptedModelTestCase(TestCase):

    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)

    def test_encrypted_filter(self):
        query = self.client.query(self.ip1_ptxt)
        results = _Ping.objects.encrypted_filter(source=query)

        self.assertEqual(2, len(results))
        dests = [result.destination for result in results]
        self.assertIn(self.ip2, dests)
        self.assertIn(self.ip3, dests)
예제 #6
0
 def setUp(self):
     self.client = Client()