88"""
99
1010import os
11- import platform
12- import subprocess
1311import tempfile
1412import unittest
1513from pathlib import Path
16- from unittest .mock import MagicMock , Mock , mock_open , patch
14+ from unittest .mock import Mock , mock_open , patch
1715
16+ from azext_aks_agent ._consts import HELM_VERSION
1817from azext_aks_agent .agent .k8s .helm_manager import HelmManager
1918
2019
@@ -23,7 +22,7 @@ class TestHelmManager(unittest.TestCase):
2322
2423 def setUp (self ):
2524 """Set up test fixtures."""
26- self .test_helm_version = "3.14.0"
25+ self .test_helm_version = HELM_VERSION
2726 self .test_kubeconfig = "/mock/kubeconfig"
2827 self .test_bin_dir = "/mock/bin"
2928
@@ -35,7 +34,7 @@ def test_init_default_values(self, mock_ensure_helm, mock_mkdir):
3534
3635 manager = HelmManager ()
3736
38- self .assertEqual (manager .helm_version , "3.14 .0" )
37+ self .assertEqual (manager .helm_version , "3.16 .0" )
3938 self .assertIsNone (manager .kubeconfig_path )
4039 self .assertIsNotNone (manager .local_bin_dir )
4140 mock_mkdir .assert_called_once ()
@@ -171,15 +170,15 @@ def test_get_version(self, mock_ensure_helm, mock_mkdir, mock_run):
171170 mock_ensure_helm .return_value = "/mock/helm"
172171 mock_run .return_value = Mock (
173172 returncode = 0 ,
174- stdout = "version.BuildInfo{Version:\" v3.14 .0\" }" ,
173+ stdout = "version.BuildInfo{Version:\" v3.16 .0\" }" ,
175174 stderr = ""
176175 )
177176
178177 manager = HelmManager ()
179178 version = manager .get_version ()
180179
181180 self .assertIsNotNone (version )
182- self .assertIn ("3.14 .0" , version )
181+ self .assertIn ("3.16 .0" , version )
183182
184183 @patch ('azext_aks_agent.agent.k8s.helm_manager.subprocess.run' )
185184 @patch ('azext_aks_agent.agent.k8s.helm_manager.Path.mkdir' )
@@ -227,7 +226,7 @@ class TestHelmManagerRealDownload(unittest.TestCase):
227226
228227 def setUp (self ):
229228 """Set up test fixtures."""
230- self .test_helm_version = "3.14.0"
229+ self .test_helm_version = HELM_VERSION
231230 self .temp_dir = None
232231
233232 def tearDown (self ):
@@ -249,7 +248,7 @@ def test_real_download_all_platforms(self):
249248 ("darwin" , "amd64" ),
250249 ("darwin" , "arm64" ),
251250 ("windows" , "amd64" ),
252- # ("windows", "arm64"), # The blob does not exist for this platform
251+ ("windows" , "arm64" ),
253252 ]
254253
255254 for os_name , arch in platforms :
0 commit comments