def test_can_create_package(self):
'''
Tests that a proper wheel package is created from the create_package method of
the AgentPackage class.
'''
agent_name = AGENT_TESTCASE1_NAME
package_tmp_dir = os.path.join(self.tmpdir, 'create_package')
expected_package_name = 'listeneragent-0.1-py2-none-any.whl'
returned_package = create_package(
self.get_agent_fixture(agent_name), package_tmp_dir)
self.assertIsNotNone(
returned_package, "Invalid package name {}".format(returned_package))
self.assertTrue(os.path.exists(returned_package))
self.assertEqual(
expected_package_name, os.path.basename(returned_package))
# Wheel is in the correct location.
self.assertEqual(
os.path.join(package_tmp_dir, expected_package_name), returned_package)
self.assertTrue(os.path.exists(returned_package))
try:
wf = WheelFile(returned_package)
# sets up the expected hashes for all of the wheel directory.
self.assertIsNone(wf.verify())
# Reading the files
# if the hash doesn't match it will throw an exception.
for o in wf.zipfile.infolist():
wf.zipfile.open(o).read()
wf.zipfile.close()
finally:
shutil.rmtree(package_tmp_dir)
def test_can_extract_package(self):
wheelhouse = os.path.join(self.tmpdir, 'extract_package')
expected_install_at = os.path.join(wheelhouse, 'listeneragent-0.1')
test_wheel_name = 'listeneragent-0.1-py2-none-any.whl'
wheel_file = os.path.join(self.fixtureDir, test_wheel_name)
installed_at = extract_package(wheel_file, wheelhouse)
try:
self.assertIsNotNone(installed_at)
self.assertTrue(os.path.isdir(installed_at))
self.assertEqual(expected_install_at, installed_at)
# use the wheel file to verify that everything was extracted
# properly.
wf = WheelFile(wheel_file)
self.assertIsNone(wf.verify())
for o in wf.zipfile.infolist():
self.assertTrue(
os.path.exists(
os.path.join(expected_install_at, o.filename)))
wf.zipfile.close()
finally:
shutil.rmtree(installed_at)
shutil.rmtree(wheelhouse)
def test_can_extract_package(self):
wheelhouse = os.path.join(self.tmpdir, 'extract_package')
expected_install_at = os.path.join(wheelhouse, 'listeneragent-0.1')
test_wheel_name = 'listeneragent-0.1-py2-none-any.whl'
wheel_file = os.path.join(self.fixtureDir, test_wheel_name)
installed_at = extract_package(wheel_file, wheelhouse)
try:
self.assertIsNotNone(installed_at)
self.assertTrue(os.path.isdir(installed_at))
self.assertEqual(expected_install_at, installed_at)
# use the wheel file to verify that everything was extracted
# properly.
wf = WheelFile(wheel_file)
self.assertIsNone(wf.verify())
for o in wf.zipfile.infolist():
self.assertTrue(
os.path.exists(os.path.join(expected_install_at, o.filename)))
wf.zipfile.close()
finally:
shutil.rmtree(installed_at)
shutil.rmtree(wheelhouse)
def test_can_create_package(self):
'''
Tests that a proper wheel package is created from the create_package method of
the AgentPackage class.
'''
agent_name = AGENT_TESTCASE1_NAME
package_tmp_dir = os.path.join(self.tmpdir, 'create_package')
expected_package_name = 'listeneragent-0.1-py2-none-any.whl'
returned_package = create_package(
self.get_agent_fixture(agent_name), package_tmp_dir)
self.assertIsNotNone(
returned_package, "Invalid package name {}".format(returned_package))
self.assertTrue(os.path.exists(returned_package))
self.assertEqual(
expected_package_name, os.path.basename(returned_package))
# Wheel is in the correct location.
self.assertEqual(
os.path.join(package_tmp_dir, expected_package_name), returned_package)
self.assertTrue(os.path.exists(returned_package))
try:
wf = WheelFile(returned_package)
# sets up the expected hashes for all of the wheel directory.
self.assertIsNone(wf.verify())
# Reading the files
# if the hash doesn't match it will throw an exception.
for o in wf.zipfile.infolist():
wf.zipfile.open(o).read()
wf.zipfile.close()
finally:
shutil.rmtree(package_tmp_dir)
