Source Code for Module cm_api_tests.test_clusters

 1  # Licensed to Cloudera, Inc. under one 
 2  # or more contributor license agreements.  See the NOTICE file 
 3  # distributed with this work for additional information 
 4  # regarding copyright ownership.  Cloudera, Inc. licenses this file 
 5  # to you under the Apache License, Version 2.0 (the 
 6  # "License"); you may not use this file except in compliance 
 7  # with the License.  You may obtain a copy of the License at 
 8  # 
 9  #     http://www.apache.org/licenses/LICENSE-2.0 
10  # 
11  # Unless required by applicable law or agreed to in writing, software 
12  # distributed under the License is distributed on an "AS IS" BASIS, 
13  # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 
14  # See the License for the specific language governing permissions and 
15  # limitations under the License. 
16   
17  import datetime 
18  import json 
19  import unittest 
20  from cm_api.endpoints.clusters import * 
21  from cm_api.endpoints.types import * 
22  from cm_api_tests import utils 
23   
24 -class TestCluster(unittest.TestCase): 
25   
26 -  def test_add_hosts(self): 
27      resource = utils.MockResource(self) 
28      cluster = ApiCluster(resource, name="foo") 
29   
30      data = ApiList([ ApiHostRef(resource, hostId='foo') ]) 
31   
32      resource.expect("POST", "/clusters/foo/hosts", 
33          data=json.dumps(data.to_json_dict()), 
34          retdata={ 'items' : [ { 'hostId' : 'foo' } ] }) 
35      cluster.add_hosts(['foo']) 
36