def test_package_pip(self, mock_labbook): lb = mock_labbook[2] package_manager_dir = os.path.join(lb.root_dir, '.gigantum', 'env', 'package_manager') with open(os.path.join(package_manager_dir, 'pip3_docker.yaml'), 'w') as apt_dep: content = os.linesep.join([ 'manager: pip3', 'package: docker', 'version: "2.0.1"', 'from_base: false' ]) apt_dep.write(content) ib = ImageBuilder(lb) pkg_lines = [l for l in ib._load_packages() if 'RUN' in l] assert 'RUN pip install docker==2.0.1' in pkg_lines with open(os.path.join(package_manager_dir, 'pip3_docker.yaml'), 'w') as apt_dep: content = os.linesep.join([ 'manager: pip', 'package: docker', 'version: "2.0.1"', 'from_base: true' ]) apt_dep.write(content) ib = ImageBuilder(lb) pkg_lines = [l for l in ib._load_packages() if 'RUN' in l] assert 'RUN pip install docker==2.0.1' not in pkg_lines
def test_package_apt(self, mock_labbook): lb = mock_labbook[2] package_manager_dir = os.path.join(lb.root_dir, '.gigantum', 'env', 'package_manager') with open(os.path.join(package_manager_dir, 'apt_docker.yaml'), 'w') as apt_dep: content = os.linesep.join([ 'manager: apt', 'package: docker', 'version: 1.2.3', 'from_base: false' ]) apt_dep.write(content) ib = ImageBuilder(lb) pkg_lines = [l for l in ib._load_packages() if 'RUN' in l] assert 'RUN apt-get -y --no-install-recommends install docker' in pkg_lines