class BluecoatTest(unittest.TestCase):
def setUp(self):
bridge_file = bridge.__file__
config_path = os.path.join(os.path.dirname(os.path.abspath(bridge_file)), "testing.conf")
self.test_file = os.path.join(os.path.dirname(os.path.abspath(__file__)), "randdata")
self.config = SafeConfigParser()
self.config.read(config_path)
self.bluecoat_provider = BluecoatProvider('bluecoat-test',
self.config.get("bridge", "bluecoat_url"),
self.config.get("bridge", "bluecoat_api_key"),
self.config.get("bridge", "bluecoat_owner"))
def test_submit_md5sum(self):
#
# NOTE: use a hash of something we have already submitted
# Depending on the BlueCoat MAA you are using you might have to change this hash
#
print self.bluecoat_provider.check_result_for('2c00c4d5a3aa8f6e7bb4cf0b658ee2e0')
def test_submit_binary(self):
with open(self.test_file, 'wb+') as hfile:
hfile.write(os.urandom(1024))
hfile.seek(0)
self.md5sum_test_file = md5(hfile.read()).hexdigest()
hfile.seek(0)
print self.bluecoat_provider.analyze_binary(self.md5sum_test_file, hfile)
#
# Delete the test file
#
os.remove(self.test_file)
def test_submitted_binary(self):
with open(self.test_file, 'wb+') as hfile:
hfile.write(os.urandom(1024))
hfile.seek(0)
self.md5sum_test_file = md5(hfile.read()).hexdigest()
hfile.seek(0)
#
# Unit tests should not be dependent on each other
#
self.bluecoat_provider.analyze_binary(self.md5sum_test_file, hfile)
#
# Now do a seperate check_result_for to make sure that is tested
#
print self.bluecoat_provider.check_result_for(self.md5sum_test_file)
#
# Delete the test file
#
os.remove(self.test_file)
def setUp(self):
bridge_file = bridge.__file__
config_path = os.path.join(os.path.dirname(os.path.abspath(bridge_file)), "testing.conf")
self.test_file = os.path.join(os.path.dirname(os.path.abspath(__file__)), "randdata")
self.config = SafeConfigParser()
self.config.read(config_path)
self.bluecoat_provider = BluecoatProvider('bluecoat-test',
self.config.get("bridge", "bluecoat_url"),
self.config.get("bridge", "bluecoat_api_key"),
self.config.get("bridge", "bluecoat_owner"))
