1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
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 }