class LaunchpadTestCase(unit.TestCase):
def setUp(self):
super().setUp()
self._project = self._make_snapcraft_project()
self.lpc = LaunchpadClient(
project=self._project, build_id="id", user="******", architectures=[]
)
@mock.patch("launchpadlib.launchpad.Launchpad.login_with")
def test_login(self, mock_login):
self.lpc.login()
self.assertThat(self.lpc.user, Equals("user"))
mock_login.assert_called_with(
"snapcraft remote-build {}".format(snapcraft.__version__),
"production",
mock.ANY,
credentials_file=mock.ANY,
version="devel",
)
@mock.patch("launchpadlib.launchpad.Launchpad.login_with")
def test_login_connection_issues(self, mock_login):
mock_login.side_effect = ConnectionRefusedError()
self.assertRaises(errors.LaunchpadHttpsError, self.lpc.login)
mock_login.side_effect = TimeoutError()
self.assertRaises(errors.LaunchpadHttpsError, self.lpc.login)
@mock.patch("snapcraft.internal.remote_build.LaunchpadClient.login")
def test_load_connection_issues(self, mock_login):
self.lpc._lp = LaunchpadImpl()
# ConnectionRefusedError should surface.
self.lpc._lp._load_mock.side_effect = ConnectionRefusedError
self.assertRaises(ConnectionRefusedError, self.lpc._lp_load_url, "foo")
# Load URL should work OK after single connection reset.
self.lpc._lp._load_mock.side_effect = [ConnectionResetError, None]
self.lpc._lp_load_url(url="foo")
mock_login.assert_called()
# Load URL should fail with two connection resets.
self.lpc._lp._load_mock.side_effect = [
ConnectionResetError,
ConnectionResetError,
]
self.assertRaises(ConnectionResetError, self.lpc._lp_load_url, "foo")
mock_login.assert_called()
def test_create_snap(self):
self.lpc._lp = LaunchpadImpl()
self.lpc._create_snap()
self.lpc._lp.snaps.new_mock.assert_called_with(
auto_build=False,
auto_build_archive="/ubuntu/+archive/primary",
auto_build_pocket="Updates",
git_path="master",
git_repository_url="https://[email protected]/~user/+git/id/",
name="id",
owner="/~user",
)
def test_create_snap_with_archs(self):
self.lpc._lp = LaunchpadImpl()
self.lpc.architectures = ["arch1", "arch2"]
self.lpc._create_snap()
self.lpc._lp.snaps.new_mock.assert_called_with(
auto_build=False,
auto_build_archive="/ubuntu/+archive/primary",
auto_build_pocket="Updates",
git_path="master",
git_repository_url="https://[email protected]/~user/+git/id/",
name="id",
owner="/~user",
processors=["/+processors/arch1", "/+processors/arch2"],
)
def test_delete_snap(self):
self.lpc._lp = LaunchpadImpl()
self.lpc._delete_snap()
self.lpc._lp.snaps.getByName_mock.assert_called_with(name="id", owner="/~user")
def test_start_build(self):
self.lpc._lp = LaunchpadImpl()
self.lpc.start_build()
@mock.patch(
"tests.unit.remote_build.test_launchpad.SnapImpl.requestBuilds",
return_value=SnapBuildReqImpl(
status="Failed", error_message="snapcraft.yaml not found..."
),
)
def test_start_build_error(self, mock_rb):
self.lpc._lp = LaunchpadImpl()
raised = self.assertRaises(errors.RemoteBuilderError, self.lpc.start_build)
self.assertThat(str(raised), Contains("snapcraft.yaml not found..."))
@mock.patch(
"tests.unit.remote_build.test_launchpad.SnapImpl.requestBuilds",
return_value=SnapBuildReqImpl(status="Pending", error_message=""),
)
def test_start_build_error_timeout(self, mock_rb):
self.lpc._lp = LaunchpadImpl()
raised = self.assertRaises(
errors.RemoteBuilderNotReadyError, self.lpc.start_build, timeout=0
)
self.assertThat(str(raised), Contains("is not ready"))
@mock.patch("snapcraft.internal.remote_build.LaunchpadClient._download_file")
def test_monitor_build(self, mock_download_file):
open("test_i386.txt", "w").close()
open("test_i386.1.txt", "w").close()
self.lpc._lp = LaunchpadImpl()
self.lpc.start_build()
self.lpc.monitor_build(interval=0)
mock_download_file.assert_has_calls(
[
mock.call(url="url_for/snap_file_i386.snap", dst="snap_file_i386.snap"),
mock.call(
url="url_for/build_log_file_1", gunzip=True, dst="test_i386.2.txt"
),
mock.call(
url="url_for/build_log_file_2", gunzip=True, dst="test_amd64.txt"
),
]
)
@mock.patch("snapcraft.internal.remote_build.LaunchpadClient._download_file")
@mock.patch(
"tests.unit.remote_build.test_launchpad.BuildImpl.getFileUrls", return_value=[]
)
@mock.patch("logging.Logger.error")
def test_monitor_build_error(self, mock_log, mock_urls, mock_download_file):
self.lpc._lp = LaunchpadImpl()
self.lpc.start_build()
self.lpc.monitor_build(interval=0)
mock_download_file.assert_has_calls(
[
mock.call(
url="url_for/build_log_file_2", gunzip=True, dst="test_amd64.txt"
)
]
)
mock_log.assert_called_with("Build failed for arch 'amd64'.")
def test_get_build_status(self):
self.lpc._lp = LaunchpadImpl()
self.lpc.start_build()
build_status = self.lpc.get_build_status()
self.assertThat(
build_status,
Equals({"i386": "Successfully built", "amd64": "Failed to build"}),
)
def _make_snapcraft_project(self):
yaml = textwrap.dedent(
"""\
name: test
base: core18
version: "1.0"
summary: test
description: test
confinement: strict
grade: stable
"""
)
snapcraft_yaml_file_path = self.make_snapcraft_yaml(yaml)
project = snapcraft.project.Project(
snapcraft_yaml_file_path=snapcraft_yaml_file_path
)
return project
def test_git_repository_creation(self):
source_testdir = self.useFixture(TestDir())
source_testdir.create_file("foo")
repo_dir = source_testdir.path
self.assertFalse(os.path.exists(os.path.join(repo_dir, ".git")))
self.lpc._gitify_repository(repo_dir)
self.assertTrue(os.path.exists(os.path.join(repo_dir, ".git")))
@mock.patch("snapcraft.internal.sources.Git.push", return_value=None)
def test_push_source_tree(self, mock_push):
self.lpc._lp = LaunchpadImpl()
source_testdir = self.useFixture(TestDir())
source_testdir.create_file("foo")
repo_dir = source_testdir.path
self.lpc.push_source_tree(repo_dir)
mock_push.assert_called_with(
"https://*****:*****@git.launchpad.net/~user/+git/id/",
"HEAD:master",
force=True,
)
@mock.patch(
"snapcraft.internal.sources.Git.push",
side_effect=SnapcraftPullError(
command="git push HEAD:master https://user:access-token@url", exit_code=128
),
)
def test_push_source_tree_error(self, mock_push):
self.lpc._lp = LaunchpadImpl()
source_testdir = self.useFixture(TestDir())
source_testdir.create_file("foo")
repo_dir = source_testdir.path
self.assertRaises(
errors.LaunchpadGitPushError, self.lpc.push_source_tree, repo_dir
)
mock_push.assert_called_with(
"https://*****:*****@git.launchpad.net/~user/+git/id/",
"HEAD:master",
force=True,
)
class LaunchpadTestCase(unit.TestCase):
def setUp(self):
super().setUp()
self._project = self._make_snapcraft_project()
self.lp = LaunchpadImpl()
self.fake_login_with = fixtures.MockPatch(
"launchpadlib.launchpad.Launchpad.login_with",
return_value=self.lp)
self.useFixture(self.fake_login_with)
self.mock_git_class = mock.Mock(spec=Git)
self.mock_git_class.check_command_installed.return_value = True
self.lpc = LaunchpadClient(
project=self._project,
build_id="id",
architectures=[],
git_class=self.mock_git_class,
running_snapcraft_version="100.1234",
)
def test_login(self):
self.assertThat(self.lpc.user, Equals("user"))
self.fake_login_with.mock.assert_called_with(
"snapcraft remote-build {}".format(snapcraft.__version__),
"production",
mock.ANY,
credentials_file=mock.ANY,
version="devel",
)
def test_login_connection_issues(self):
self.fake_login_with.mock.side_effect = ConnectionRefusedError()
self.assertRaises(errors.LaunchpadHttpsError, self.lpc.login)
self.fake_login_with.mock.side_effect = TimeoutError()
self.assertRaises(errors.LaunchpadHttpsError, self.lpc.login)
def test_load_connection_refused(self):
# ConnectionRefusedError should surface.
self.fake_login_with.mock.reset_mock()
self.lpc._lp._load_mock.side_effect = ConnectionRefusedError
self.assertRaises(ConnectionRefusedError, self.lpc._lp_load_url, "foo")
self.fake_login_with.mock.assert_not_called()
def test_load_connection_reset_once(self):
# Load URL should work OK after single connection reset.
self.fake_login_with.mock.reset_mock()
self.lpc._lp._load_mock.side_effect = [ConnectionResetError, None]
self.lpc._lp_load_url(url="foo")
self.fake_login_with.mock.assert_called()
def test_load_connection_reset_twice(self):
# Load URL should fail with two connection resets.
self.fake_login_with.mock.reset_mock()
self.lpc._lp._load_mock.side_effect = [
ConnectionResetError,
ConnectionResetError,
]
self.assertRaises(ConnectionResetError, self.lpc._lp_load_url, "foo")
self.fake_login_with.mock.assert_called()
def test_create_snap(self):
self.lpc._create_snap()
self.lpc._lp.snaps.new_mock.assert_called_with(
auto_build=False,
auto_build_archive="/ubuntu/+archive/primary",
auto_build_pocket="Updates",
git_path="master",
git_repository_url="https://[email protected]/~user/+git/id/",
name="id",
owner="/~user",
)
def test_create_snap_with_archs(self):
self.lpc.architectures = ["arch1", "arch2"]
self.lpc._create_snap()
self.lpc._lp.snaps.new_mock.assert_called_with(
auto_build=False,
auto_build_archive="/ubuntu/+archive/primary",
auto_build_pocket="Updates",
git_path="master",
git_repository_url="https://[email protected]/~user/+git/id/",
name="id",
owner="/~user",
processors=["/+processors/arch1", "/+processors/arch2"],
)
def test_delete_snap(self):
self.lpc._delete_snap()
self.lpc._lp.snaps.getByName_mock.assert_called_with(name="id",
owner="/~user")
def test_start_build(self):
self.lpc.start_build()
@mock.patch(
"tests.unit.remote_build.test_launchpad.SnapImpl.requestBuilds",
return_value=SnapBuildReqImpl(
status="Failed", error_message="snapcraft.yaml not found..."),
)
def test_start_build_error(self, mock_rb):
raised = self.assertRaises(errors.RemoteBuilderError,
self.lpc.start_build)
self.assertThat(str(raised), Contains("snapcraft.yaml not found..."))
@mock.patch(
"tests.unit.remote_build.test_launchpad.SnapImpl.requestBuilds",
return_value=SnapBuildReqImpl(status="Pending", error_message=""),
)
@mock.patch("time.time", return_value=500)
def test_start_build_error_timeout(self, mock_time, mock_rb):
self.lpc.deadline = 499
raised = self.assertRaises(errors.RemoteBuildTimeoutError,
self.lpc.start_build)
self.assertThat(str(raised),
Equals("Remote build exceeded configured timeout."))
def test_issue_build_request_defaults(self):
fake_snap = mock.MagicMock()
self.lpc._issue_build_request(fake_snap)
self.assertThat(
fake_snap.mock_calls,
Equals([
mock.call.requestBuilds(
archive="main_archive",
channels={
"core18": "stable",
"snapcraft": "stable"
},
pocket="Updates",
)
]),
)
@mock.patch(
"snapcraft.internal.remote_build.LaunchpadClient._download_file")
def test_monitor_build(self, mock_download_file):
open("test_i386.txt", "w").close()
open("test_i386.1.txt", "w").close()
self.lpc.start_build()
self.lpc.monitor_build(interval=0)
self.assertThat(
mock_download_file.mock_calls,
Equals([
mock.call(dst="snap_file_i386.snap",
url="url_for/snap_file_i386.snap"),
mock.call(
dst="test_i386.2.txt",
gunzip=True,
url="url_for/build_log_file_1",
),
mock.call(dst="snap_file_amd64.snap",
url="url_for/snap_file_amd64.snap"),
mock.call(
dst="test_amd64.txt",
gunzip=True,
url="url_for/build_log_file_2",
),
mock.call(dst="snap_file_amd64.snap",
url="url_for/snap_file_amd64.snap"),
]),
)
@mock.patch(
"snapcraft.internal.remote_build.LaunchpadClient._download_file")
@mock.patch("tests.unit.remote_build.test_launchpad.BuildImpl.getFileUrls",
return_value=[])
@mock.patch("logging.Logger.error")
def test_monitor_build_error(self, mock_log, mock_urls,
mock_download_file):
self.lpc.start_build()
self.lpc.monitor_build(interval=0)
mock_download_file.assert_has_calls([
mock.call(url="url_for/build_log_file_2",
gunzip=True,
dst="test_amd64.txt")
])
self.assertThat(
mock_log.mock_calls,
Equals([
mock.call("Snap file not available for arch 'i386'."),
mock.call("Snap file not available for arch 'amd64'."),
mock.call("Build failed for arch 'amd64'."),
mock.call("Snap file not available for arch 'arm64'."),
mock.call("Build failed for arch 'arm64'."),
]),
)
@mock.patch(
"snapcraft.internal.remote_build.LaunchpadClient._download_file")
@mock.patch("time.time", return_value=500)
def test_monitor_build_error_timeout(self, mock_time, mock_rb):
self.lpc.deadline = 499
self.lpc.start_build()
raised = self.assertRaises(errors.RemoteBuildTimeoutError,
self.lpc.monitor_build,
interval=0)
self.assertThat(str(raised),
Equals("Remote build exceeded configured timeout."))
def test_get_build_status(self):
self.lpc.start_build()
build_status = self.lpc.get_build_status()
self.assertThat(
build_status,
Equals({
"amd64": "Failed to build",
"arm64": "Failed to build",
"i386": "Successfully built",
}),
)
def _make_snapcraft_project(self):
yaml = textwrap.dedent("""\
name: test
base: core18
version: "1.0"
summary: test
description: test
confinement: strict
grade: stable
""")
snapcraft_yaml_file_path = self.make_snapcraft_yaml(yaml)
project = snapcraft.project.Project(
snapcraft_yaml_file_path=snapcraft_yaml_file_path)
return project
def test_git_repository_creation(self):
source_testdir = self.useFixture(TestDir())
source_testdir.create_file("foo")
repo_dir = source_testdir.path
self.lpc._gitify_repository(repo_dir)
assert self.mock_git_class.mock_calls == [
mock.call.check_command_installed(),
mock.call(repo_dir, repo_dir, silent=True),
mock.call().init(),
mock.call().add("foo"),
mock.call().commit("committed by snapcraft version: 100.1234"),
]
def test_push_source_tree(self):
source_testdir = self.useFixture(TestDir())
repo_dir = source_testdir.path
self.lpc.push_source_tree(repo_dir)
self.mock_git_class.assert_has_calls == [
mock.call(
"https://*****:*****@git.launchpad.net/~user/+git/id/",
"HEAD:master",
force=True,
)
]
def test_push_source_tree_error(self):
self.mock_git_class.return_value.push.side_effect = (
SnapcraftPullError(
command="git push HEAD:master https://user:access-token@url",
exit_code=128,
), )
source_testdir = self.useFixture(TestDir())
repo_dir = source_testdir.path
self.assertRaises(errors.LaunchpadGitPushError,
self.lpc.push_source_tree, repo_dir)