From 7ab32b27fe52644d96a1ec07e9dd79be37db26ec Mon Sep 17 00:00:00 2001
From: Alberto Contreras <alberto.contreras@canonical.com>
Date: Mon, 16 Oct 2023 13:29:26 +0200
Subject: [PATCH] test: add test for apt.update_sources_list

---
 uaclient/tests/test_apt.py | 34 ++++++++++++++++++++++++++++++++++
 1 file changed, 34 insertions(+)

diff --git a/uaclient/tests/test_apt.py b/uaclient/tests/test_apt.py
index 1048afe84c..df05a08dcb 100644
--- a/uaclient/tests/test_apt.py
+++ b/uaclient/tests/test_apt.py
@@ -42,6 +42,7 @@
     run_apt_update_command,
     setup_apt_proxy,
     update_esm_caches,
+    update_sources_list,
 )
 from uaclient.entitlements.base import UAEntitlement
 from uaclient.entitlements.entitlement_status import ApplicationStatus
@@ -1453,3 +1454,36 @@ def test_get_alternative_versions_excluding_origin(self):
                 )
             )
         ]
+
+
+class TestUpdateSourcesList:
+    @mock.patch("uaclient.apt.os.path.join")
+    @mock.patch("apt_pkg.FileLock")
+    @mock.patch("uaclient.apt.AcquireProgress")
+    @mock.patch("apt_pkg.SourceList")
+    @mock.patch("apt_pkg.config")
+    @mock.patch("uaclient.apt.PreserveAptCfg")
+    def test_update_sources_list(
+        self,
+        m_preserve_apt_cfg,
+        m_apt_pkg_config,
+        m_apt_pkg_source_list,
+        m_acquire_progress,
+        _m_file_lock,
+        _m_join,
+    ):
+        update_sources_list("/sources.list")
+        assert (
+            mock.call("Dir::Etc::sourcelist", "/sources.list")
+            in m_apt_pkg_config.set.call_args_list
+        )
+        assert (
+            [
+                mock.call(
+                    m_acquire_progress.return_value,
+                    m_apt_pkg_source_list.return_value,
+                    0,
+                )
+            ]
+            == m_preserve_apt_cfg.return_value.__enter__.return_value.update.call_args_list  # noqa: E501
+        )