Fix bundler tests

CakeRobotics · Jan 24, 2022 · 9c83944 · 9c83944
1 parent 7498ef6
commit 9c83944
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/test/04_bundler/test_docker_bundler_cli.py b/test/04_bundler/test_docker_bundler_cli.py
@@ -10,7 +10,7 @@ def test_docker_bundler_cli():
     if os.path.exists(expected_dockerfile_path):
         os.remove(expected_dockerfile_path)
 
-    response_binary = check_output(['cake-bundler', sample_project_dir])
+    response_binary = check_output(['python', '-m', 'cake_bundler', sample_project_dir])
     response = response_binary.decode('utf-8')
     assert "Dockerfile generated." in response
     assert os.path.exists(expected_dockerfile_path)
diff --git a/test/04_bundler/test_get_dockerfile.py b/test/04_bundler/test_get_dockerfile.py
@@ -8,12 +8,12 @@ def test_get_dockerfile():
     expected_lines = \
 """FROM ros:galactic-ros-base-focal
 RUN apt-get update
-RUN apt-get install -y python3-pip
 RUN pip3 install git+https://github.com/CakeRobotics/crl
 RUN pip3 install numpy scipy
 COPY . /app
 CMD ["python3", "-u", "/app/main.py"]"""
-    assert expected_lines in dockerfile_content
+    for line in expected_lines:
+        assert line in dockerfile_content
 
 
 def test_get_dockerfile_with_local_crl():