View Javadoc

1   /*
2    * Licensed to the Apache Software Foundation (ASF) under one
3    * or more contributor license agreements.  See the NOTICE file
4    * distributed with this work for additional information
5    * regarding copyright ownership.  The ASF licenses this file
6    * to you under the Apache License, Version 2.0 (the
7    * "License"); you may not use this file except in compliance
8    * with the License.  You may obtain a copy of the License at
9    *
10   *     http://www.apache.org/licenses/LICENSE-2.0
11   *
12   * Unless required by applicable law or agreed to in writing, software
13   * distributed under the License is distributed on an "AS IS" BASIS,
14   * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
15   * See the License for the specific language governing permissions and
16   * limitations under the License.
17   */
18  package org.apache.hadoop.chukwa.rest.resource;
19  
20  import org.json.simple.JSONObject;
21  
22  import org.apache.hadoop.chukwa.rest.bean.ReturnCodeBean;
23  import org.apache.hadoop.chukwa.rest.bean.UserBean;
24  import org.apache.hadoop.chukwa.util.ExceptionUtil;
25  import org.json.simple.JSONArray;
26  
27  import com.sun.jersey.api.client.Client;
28  
29  public class TestUserResource extends SetupTestEnv {
30    public void testUserSave() {
31      try {
32      UserBean user = new UserBean();
33      user.setId("admin");
34      user.setProperty("testKey", "testValue");
35      JSONArray ja = new JSONArray();
36      user.setViews(ja);
37      client = Client.create();
38      resource = client.resource("http://localhost:"+restPort);
39      ReturnCodeBean result = resource.path("/hicc/v1/user").
40      header("Content-Type","application/json").
41      header("Authorization", authorization).
42      put(ReturnCodeBean.class, user);
43      assertEquals(1, result.getCode());
44      } catch (Exception e) {
45        fail(ExceptionUtil.getStackTrace(e));
46      }
47    }
48    
49    public void testUserLoad() {
50      client = Client.create();
51      resource = client.resource("http://localhost:"+restPort);
52      UserBean user = resource.path("/hicc/v1/user/uid/admin").header("Authorization", authorization).get(UserBean.class);
53      try {
54        assertEquals("admin", user.getId());
55        assertEquals("testValue", user.getPropertyValue("testKey"));
56      } catch (Exception e) {
57        fail(ExceptionUtil.getStackTrace(e));
58      }
59    }
60  }