7
7
import os
8
8
import socket
9
9
import re
10
- import importlib
11
10
import subprocess
12
11
import unittest
13
12
import pytest
20
19
21
20
logging .basicConfig (level = logging .DEBUG )
22
21
23
- def setup_function ():
24
- utils ._ip_for_cloud = None
25
- # Mock memoize method and reload the module under test later with imp
26
- mock .patch ('crmsh.utils.memoize' , lambda x : x ).start ()
27
- importlib .reload (utils )
28
-
29
22
30
23
@mock .patch ("crmsh.utils.get_stdout_stderr" )
31
24
def test_print_cluster_nodes (mock_run ):
@@ -547,7 +540,7 @@ def test_detect_cloud_no_cmd(mock_is_program):
547
540
def test_detect_cloud_aws (mock_is_program , mock_aws ):
548
541
mock_is_program .return_value = True
549
542
mock_aws .return_value = True
550
- assert utils .detect_cloud () == constants .CLOUD_AWS
543
+ assert utils .detect_cloud . __wrapped__ () == constants .CLOUD_AWS
551
544
mock_is_program .assert_called_once_with ("dmidecode" )
552
545
mock_aws .assert_called_once_with ()
553
546
@@ -558,7 +551,7 @@ def test_detect_cloud_azure(mock_is_program, mock_aws, mock_azure):
558
551
mock_is_program .return_value = True
559
552
mock_aws .return_value = False
560
553
mock_azure .return_value = True
561
- assert utils .detect_cloud () == constants .CLOUD_AZURE
554
+ assert utils .detect_cloud . __wrapped__ () == constants .CLOUD_AZURE
562
555
mock_is_program .assert_called_once_with ("dmidecode" )
563
556
mock_aws .assert_called_once_with ()
564
557
mock_azure .assert_called_once_with ()
@@ -572,7 +565,7 @@ def test_detect_cloud_gcp(mock_is_program, mock_aws, mock_azure, mock_gcp):
572
565
mock_aws .return_value = False
573
566
mock_azure .return_value = False
574
567
mock_gcp .return_value = True
575
- assert utils .detect_cloud () == constants .CLOUD_GCP
568
+ assert utils .detect_cloud . __wrapped__ () == constants .CLOUD_GCP
576
569
mock_is_program .assert_called_once_with ("dmidecode" )
577
570
mock_aws .assert_called_once_with ()
578
571
mock_azure .assert_called_once_with ()
@@ -1681,6 +1674,7 @@ def test_handle_role_for_ocf_1_1(mock_support, mock_warn):
1681
1674
def test_handle_role_for_ocf_1_1_convert_new (mock_support , mock_info ):
1682
1675
config .core .OCF_1_1_SUPPORT = True
1683
1676
mock_support .return_value = True
1677
+ utils .auto_convert_role = True
1684
1678
assert utils .handle_role_for_ocf_1_1 ("Master" ) == "Promoted"
1685
1679
mock_support .assert_called_once_with ()
1686
1680
mock_info .assert_called_once_with ('Convert deprecated "%s" to "%s"' , "Master" , "Promoted" )
0 commit comments