59 lines
1.8 KiB
Python
59 lines
1.8 KiB
Python
|
"""
|
||
|
|
Tests Home Assistant package util methods.
|
||
|
|
"""
|
||
|
|
import unittest
|
||
|
|
import sys
|
||
|
|
import tempfile
|
||
|
|
import homeassistant.util.package as package
|
||
|
|
|
||
|
|
TEST_EXIST_REQ = "pip>=7.0.0"
|
||
|
|
TEST_NEW_REQ = "pyhelloworld3==1.0.0"
|
||
|
|
TEST_ZIP_REQ = \
|
||
|
|
"https://github.com/rmkraus/pyhelloworld3/archive/" \
|
||
|
|
"5ba878316d68ea164e2cf5bd085d0cf1fd76bd15.zip#pyhelloworld3==1.0.0"
|
||
|
|
|
||
|
|
|
||
|
|
class TestPackageUtil(unittest.TestCase):
|
||
|
|
""" Tests for homeassistant.util.package module """
|
||
|
|
|
||
|
|
def setUp(self):
|
||
|
|
""" Create local library for testing """
|
||
|
|
self.lib_dir = tempfile.TemporaryDirectory()
|
||
|
|
|
||
|
|
def tearDown(self):
|
||
|
|
""" Remove local library """
|
||
|
|
del self.lib_dir
|
||
|
|
|
||
|
|
def test_install_existing_package(self):
|
||
|
|
""" Test an install attempt on an existing package """
|
||
|
|
self.assertTrue(package.check_package_exists(
|
||
|
|
TEST_EXIST_REQ, self.lib_dir.name))
|
||
|
|
|
||
|
|
self.assertTrue(package.install_package(TEST_EXIST_REQ))
|
||
|
|
|
||
|
|
def test_install_package_locally(self):
|
||
|
|
""" Test an install attempt to the local library """
|
||
|
|
self.assertFalse(package.check_package_exists(
|
||
|
|
TEST_NEW_REQ, self.lib_dir.name))
|
||
|
|
|
||
|
|
self.assertTrue(package.install_package(
|
||
|
|
TEST_NEW_REQ, True, self.lib_dir.name))
|
||
|
|
|
||
|
|
sys.path.insert(0, self.lib_dir.name)
|
||
|
|
import pyhelloworld3
|
||
|
|
|
||
|
|
self.assertEqual(pyhelloworld3.__version__, '1.0.0')
|
||
|
|
|
||
|
|
def test_install_package_zip(self):
|
||
|
|
""" Test an install attempt from a zip path """
|
||
|
|
self.assertFalse(package.check_package_exists(
|
||
|
|
TEST_ZIP_REQ, self.lib_dir.name))
|
||
|
|
|
||
|
|
self.assertTrue(package.install_package(
|
||
|
|
TEST_ZIP_REQ, True, self.lib_dir.name))
|
||
|
|
|
||
|
|
sys.path.insert(0, self.lib_dir.name)
|
||
|
|
import pyhelloworld3
|
||
|
|
|
||
|
|
self.assertEqual(pyhelloworld3.__version__, '1.0.0')
|