瀏覽代碼

AMBARI-5604. Postgres should be option 4 in setup, not 2. (swagle)

Siddharth Wagle 11 年之前
父節點
當前提交
186db4e0cd
共有 2 個文件被更改,包括 15 次插入5 次删除
  1. 13 3
      ambari-server/src/main/python/ambari-server.py
  2. 2 2
      ambari-server/src/test/python/TestAmbariServer.py

+ 13 - 3
ambari-server/src/main/python/ambari-server.py

@@ -1185,7 +1185,7 @@ def prompt_db_properties(args):
 
       database_num = str(DATABASE_INDEX + 1)
       database_num = get_validated_string_input(
-        "[1] - PostgreSQL (Embedded)\n[2] - PostgreSQL (External)\n[3] - Oracle\n[4] - MySQL\n"
+        "[1] - PostgreSQL (Embedded)\n[2] - Oracle\n[3] - MySQL\n[4] - PostgreSQL (External)\n"
         "==============================================================================\n"
         "Enter choice (" + database_num + "): ",
         database_num,
@@ -1200,8 +1200,18 @@ def prompt_db_properties(args):
         DATABASE_INDEX = 0
       else:
         args.persistence_type = 'remote'
-        args.database_index = int(database_num) - 2
-        DATABASE_INDEX = int(database_num) - 2
+        selected_db_option = int(database_num)
+
+        if selected_db_option == 2:
+          args.database_index = 1
+        elif selected_db_option == 3:
+          args.database_index = 2
+        elif selected_db_option == 4:
+          args.database_index = 0
+        else:
+          print_info_msg('Unknown db option, default to embbeded postgres.')
+          args.database_index = 0
+      pass
 
       args.dbms = DATABASE_NAMES[args.database_index]
 

+ 2 - 2
ambari-server/src/test/python/TestAmbariServer.py

@@ -2102,7 +2102,7 @@ MIIFHjCCAwYCCQDpHKOBI+Lt0zANBgkqhkiG9w0BAQUFADBRMQswCQYDVQQGEwJV
   def test_prompt_db_properties_oracle_sname(self, gyni_mock, gvsi_mock, rp_mock, smk_mock):
     ambari_server.PROMPT_DATABASE_OPTIONS = True
     gyni_mock.return_value = True
-    list_of_return_values = ["ambari-server", "ambari", "1", "1521", "localhost", "3"]
+    list_of_return_values = ["ambari-server", "ambari", "1", "1521", "localhost", "2"]
 
     def side_effect(*args, **kwargs):
       return list_of_return_values.pop()
@@ -2128,7 +2128,7 @@ MIIFHjCCAwYCCQDpHKOBI+Lt0zANBgkqhkiG9w0BAQUFADBRMQswCQYDVQQGEwJV
   def test_prompt_db_properties_oracle_sid(self, gyni_mock, gvsi_mock, rp_mock, smk_mock):
     ambari_server.PROMPT_DATABASE_OPTIONS = True
     gyni_mock.return_value = True
-    list_of_return_values = ["ambari-server", "ambari", "2", "1521", "localhost", "3"]
+    list_of_return_values = ["ambari-server", "ambari", "2", "1521", "localhost", "2"]
 
     def side_effect(*args, **kwargs):
       return list_of_return_values.pop()