Repository: ambari
Updated Branches:
  refs/heads/trunk c0efc80d8 -> fe8a8d1e1


http://git-wip-us.apache.org/repos/asf/ambari/blob/fe8a8d1e/ambari-server/src/test/python/stacks/2.3/common/test_stack_advisor.py
----------------------------------------------------------------------
diff --git 
a/ambari-server/src/test/python/stacks/2.3/common/test_stack_advisor.py 
b/ambari-server/src/test/python/stacks/2.3/common/test_stack_advisor.py
index aa7688b..22fcb1b 100644
--- a/ambari-server/src/test/python/stacks/2.3/common/test_stack_advisor.py
+++ b/ambari-server/src/test/python/stacks/2.3/common/test_stack_advisor.py
@@ -155,7 +155,7 @@ class TestHDP23StackAdvisor(TestCase):
   def 
test_createComponentLayoutRecommendations_pxf_co_locate_with_namenode_or_datanode(self):
     """ Test that PXF gets recommended on same host group which has NAMENODE 
or DATANODE"""
 
-    services = self.load_json("services-pxf-hdfs.json")
+    services = self.load_json("services-hawq-pxf-hdfs.json")
     hosts = self.load_json("hosts-3-hosts.json")
     recommendations = 
self.stackAdvisor.createComponentLayoutRecommendations(services, hosts)
 
@@ -165,6 +165,19 @@ class TestHDP23StackAdvisor(TestCase):
         self.assertTrue("PXF" in component_names)
 
 
+  def test_hawqsegmentDatanode(self):
+    """ Test that HAWQSegment gets recommended on same host group which has 
DATANODE"""
+
+    services = self.load_json("services-hawq-pxf-hdfs.json")
+    hosts = self.load_json("hosts-3-hosts.json")
+    recommendations = 
self.stackAdvisor.createComponentLayoutRecommendations(services, hosts)
+
+    for hostgroup in recommendations["blueprint"]["host_groups"]:
+      component_names = [component["name"] for component in 
hostgroup["components"]]
+      if 'DATANODE' in component_names:
+        self.assertTrue('HAWQSEGMENT' in component_names)
+
+
   def fqdn_mock_result(value=None):
       return 'c6401.ambari.apache.org' if value is None else value
 

Reply via email to