Updated Branches: refs/heads/trunk 399a63b8e -> b2d057065
http://git-wip-us.apache.org/repos/asf/incubator-ambari/blob/b2d05706/ambari-client/src/test/python/utils/HttpClientInvoker.py ---------------------------------------------------------------------- diff --git a/ambari-client/src/test/python/utils/HttpClientInvoker.py b/ambari-client/src/test/python/utils/HttpClientInvoker.py index a8e34ec..88eefa9 100644 --- a/ambari-client/src/test/python/utils/HttpClientInvoker.py +++ b/ambari-client/src/test/python/utils/HttpClientInvoker.py @@ -3,65 +3,92 @@ class HttpClientInvoker(): def http_client_invoke_side_effects(*args, **kwargs): print locals() - if args[0] != "GET" and args[0] != "DELETE": - return "", 201 , "text/plain" + http_method = args[0] + url = args[1] mocked_code = 200 mocked_content = "text/plain" - if args[1] == "//clusters/test1": - mocked_response = open('json/clustermodel_get_cluster.json', 'r').read() - return mocked_response, mocked_code , mocked_content - elif args[1] == "//clusters/test1/hosts/myhost": - mocked_response = open('json/clustermodel_get_host.json', 'r').read() - return mocked_response, mocked_code , mocked_content - elif args[1] == "//clusters/test1/hosts?fields=*": - mocked_response = open('json/clustermodel_get_all_hosts.json', 'r').read() - return mocked_response, mocked_code , mocked_content - elif args[1] == "//clusters/test1/configurations?type=global&tag=version1": - mocked_response = open('json/clustermodel_get_global_config.json', 'r').read() - return mocked_response, mocked_code , mocked_content - elif args[1] == "//clusters/test1/configurations?type=core-site&tag=version1": - mocked_response = open('json/clustermodel_get_core_site_config.json', 'r').read() - return mocked_response, mocked_code , mocked_content - elif args[1] == "//clusters/test1/configurations?type=hdfs-site&tag=version1": - mocked_response = open('json/clustermodel_get_hdfs_site_config.json', 'r').read() - return mocked_response, mocked_code , mocked_content - elif args[1] == "//clusters/test1/configurations?type=mapred-site&tag=version1": - mocked_response = open('json/clustermodel_get_mapred_site_config.json', 'r').read() - return mocked_response, mocked_code , mocked_content - elif args[1] == "//clusters/test1/services/GANGLIA": - mocked_response = open('json/clustermodel_get_service.json', 'r').read() - return mocked_response, mocked_code , mocked_content - elif args[1] == "//clusters/test1/services?fields=*": - mocked_response = open('json/clustermodel_get_all_services.json', 'r').read() - return mocked_response, mocked_code , mocked_content - elif args[1] == "//stacks2/HDP/versions/2.0.5/stackServices/HDFS/serviceComponents?fields=*": - mocked_response = open('json/get_components_from_stack.json', 'r').read() - return mocked_response, mocked_code , mocked_content - elif args[1] == "//hosts": - mocked_response = open('json/get_all_hosts.json', 'r').read() - return mocked_response, mocked_code , mocked_content - elif args[1] == "//clusters/test6/services/GANGLIA": - mocked_response = open('json/get_cluster_service.json', 'r').read() - return mocked_response, mocked_code , mocked_content - elif args[1] == "//clusters/test6/hosts/r01wn01/host_components/NAMENODE": - mocked_response = open('json/get_host_component.json', 'r').read() - return mocked_response, mocked_code , mocked_content - elif args[1] == "//clusters/test6/hosts/r01wn01/host_components?ServiceComponentInfo": - mocked_response = open('json/get_host_components.json', 'r').read() - return mocked_response, mocked_code , mocked_content - elif args[1] == "//clusters/test6/services/GANGLIA/components/GANGLIA_MONITOR": - mocked_response = open('json/get_service_component.json', 'r').read() - return mocked_response, mocked_code , mocked_content - elif args[1] == "//clusters/test6/services/GANGLIA/components?fields=*": - mocked_response = open('json/get_service_components.json', 'r').read() - return mocked_response, mocked_code , mocked_content - elif args[1] == "//clusters/test1/hosts/deleted_nonexistant_cluster": - mocked_response = open('json/error_deleting_host.json', 'r').read() - return mocked_response, mocked_code , mocked_content - elif args[1] == "//clusters/test1/hosts/hostname01": - mocked_response = None - return mocked_response, mocked_code , mocked_content - else: - print "Unknown url: %s" % args[1] + + if http_method == "GET": + # ClusterModel mocking + if url == "//clusters/test1": + mocked_response = open('json/clustermodel_get_cluster.json', 'r').read() + return mocked_response, mocked_code , mocked_content + elif url == "//clusters/test1/hosts/myhost": + mocked_response = open('json/clustermodel_get_host.json', 'r').read() + return mocked_response, mocked_code , mocked_content + elif url == "//clusters/test1/hosts?fields=*": + mocked_response = open('json/clustermodel_get_all_hosts.json', 'r').read() + return mocked_response, mocked_code , mocked_content + elif url == "//clusters/test1/configurations?type=global&tag=version1": + mocked_response = open('json/clustermodel_get_global_config.json', 'r').read() + return mocked_response, mocked_code , mocked_content + elif url == "//clusters/test1/configurations?type=core-site&tag=version1": + mocked_response = open('json/clustermodel_get_core_site_config.json', 'r').read() + return mocked_response, mocked_code , mocked_content + elif url == "//clusters/test1/configurations?type=hdfs-site&tag=version1": + mocked_response = open('json/clustermodel_get_hdfs_site_config.json', 'r').read() + return mocked_response, mocked_code , mocked_content + elif url == "//clusters/test1/configurations?type=mapred-site&tag=version1": + mocked_response = open('json/clustermodel_get_mapred_site_config.json', 'r').read() + return mocked_response, mocked_code , mocked_content + # AmbariClient mocking + elif url == "//clusters": + mocked_response = open('json/ambariclient_get_all_clusters.json', 'r').read() + return mocked_response, mocked_code , mocked_content + elif url == "//hosts": + mocked_response = open('json/ambariclient_get_all_hosts.json', 'r').read() + return mocked_response, mocked_code , mocked_content + elif url == "//hosts/dev06.hortonworks.com": + mocked_response = open('json/ambariclient_get_host.json', 'r').read() + return mocked_response, mocked_code , mocked_content + elif url == "//stacks2/HDP/versions/1.3.0/stackServices/HDFS/configurations?fields=*": + mocked_response = open('json/ambariclient_get_config.json', 'r').read() + return mocked_response, mocked_code , mocked_content + elif url == "//stacks2/HDP/versions/1.3.0/stackServices/HDFS/serviceComponents?fields=*": + mocked_response = open('json/ambariclient_get_components.json', 'r').read() + return mocked_response, mocked_code , mocked_content + # others + elif url == "//clusters/test1/services/GANGLIA": + mocked_response = open('json/clustermodel_get_service.json', 'r').read() + return mocked_response, mocked_code , mocked_content + elif url == "//clusters/test1/services?fields=*": + mocked_response = open('json/clustermodel_get_all_services.json', 'r').read() + return mocked_response, mocked_code , mocked_content + elif url == "//stacks2/HDP/versions/2.0.5/stackServices/HDFS/serviceComponents?fields=*": + mocked_response = open('json/get_components_from_stack.json', 'r').read() + return mocked_response, mocked_code , mocked_content + elif url == "//clusters/test6/services/GANGLIA": + mocked_response = open('json/get_cluster_service.json', 'r').read() + return mocked_response, mocked_code , mocked_content + elif url == "//clusters/test6/hosts/r01wn01/host_components/NAMENODE": + mocked_response = open('json/get_host_component.json', 'r').read() + return mocked_response, mocked_code , mocked_content + elif url == "//clusters/test6/hosts/r01wn01/host_components?ServiceComponentInfo": + mocked_response = open('json/get_host_components.json', 'r').read() + return mocked_response, mocked_code , mocked_content + elif url == "//clusters/test6/services/GANGLIA/components/GANGLIA_MONITOR": + mocked_response = open('json/get_service_component.json', 'r').read() + return mocked_response, mocked_code , mocked_content + elif url == "//clusters/test6/services/GANGLIA/components?fields=*": + mocked_response = open('json/get_service_components.json', 'r').read() + return mocked_response, mocked_code , mocked_content + else: + print "Unknown get request on url: %s" % url + elif http_method == "DELETE": + # ClusterModel + if url == "//clusters/test1/hosts/deleted_nonexistant_cluster": + mocked_response = open('json/clustermodel_error_deleting_host.json', 'r').read() + return mocked_response, mocked_code , mocked_content + else: # DELETE (generally does not require any response) + return "", mocked_code , mocked_content + elif http_method == "POST": + # AmbariClient + if url == "//bootstrap": + mocked_response = open('json/ambariclient_bootstrap_hosts.json', 'r').read() + return mocked_response, mocked_code , mocked_content + else: # POST (generally does not require any response) + return "", mocked_code , mocked_content + else: # PUT (generally does not require any response) + return "", mocked_code , mocked_content \ No newline at end of file
