From 714edb3cde92215e81a0cd3f94789cf1019f5681 Mon Sep 17 00:00:00 2001
From: Elijah Ahianyo <elijahahianyo@gmail.com>
Date: Mon, 20 Nov 2023 23:21:17 +0000
Subject: [PATCH] REF-1202: Upgrade bun version if it differs from reflex set
 version (#2206)

---
 reflex/utils/prerequisites.py |  4 +++-
 tests/utils/test_utils.py     | 26 ++++++++++++++++++++++++++
 2 files changed, 29 insertions(+), 1 deletion(-)

diff --git a/reflex/utils/prerequisites.py b/reflex/utils/prerequisites.py
index 52e8f7317..4cc811054 100644
--- a/reflex/utils/prerequisites.py
+++ b/reflex/utils/prerequisites.py
@@ -487,7 +487,9 @@ def install_bun():
         return
 
     # Skip if bun is already installed.
-    if os.path.exists(get_config().bun_path):
+    if os.path.exists(get_config().bun_path) and get_bun_version() == version.parse(
+        constants.Bun.VERSION
+    ):
         console.debug("Skipping bun installation as it is already installed.")
         return
 
diff --git a/tests/utils/test_utils.py b/tests/utils/test_utils.py
index 364c6e719..72c1d0f5e 100644
--- a/tests/utils/test_utils.py
+++ b/tests/utils/test_utils.py
@@ -437,6 +437,32 @@ def test_bun_install_without_unzip(mocker):
         prerequisites.install_bun()
 
 
+@pytest.mark.parametrize("bun_version", [constants.Bun.VERSION, "1.0.0"])
+def test_bun_install_version(mocker, bun_version):
+    """Test that bun is downloaded when the host version(installed by reflex)
+    different from the current version set in reflex.
+
+    Args:
+        mocker: Pytest mocker object.
+        bun_version: the host bun version
+
+    """
+    mocker.patch("reflex.utils.prerequisites.constants.IS_WINDOWS", False)
+    mocker.patch("os.path.exists", return_value=True)
+    mocker.patch(
+        "reflex.utils.prerequisites.get_bun_version",
+        return_value=version.parse(bun_version),
+    )
+    mocker.patch("reflex.utils.path_ops.which")
+    mock = mocker.patch("reflex.utils.prerequisites.download_and_run")
+
+    prerequisites.install_bun()
+    if bun_version == constants.Bun.VERSION:
+        mock.assert_not_called()
+    else:
+        mock.assert_called_once()
+
+
 @pytest.mark.parametrize("is_windows", [True, False])
 def test_create_reflex_dir(mocker, is_windows):
     """Test that a reflex directory is created on initializing frontend