1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18 package org.apache.hadoop.chukwa.datacollection.controller;
19
20
21 import org.apache.hadoop.conf.Configuration;
22 import org.apache.hadoop.chukwa.datacollection.agent.ChukwaAgent;
23 import org.apache.hadoop.chukwa.datacollection.connector.ChunkCatcherConnector;
24 import org.apache.hadoop.chukwa.datacollection.connector.Connector;
25 import org.apache.hadoop.chukwa.datacollection.connector.http.HttpConnector;
26 import org.apache.hadoop.chukwa.datacollection.controller.ChukwaAgentController;
27 import java.io.IOException;
28 import java.util.Map;
29 import junit.framework.TestCase;
30
31 public class TestAgentClient extends TestCase {
32 Configuration config;
33 ChukwaAgent agent;
34 ChukwaAgentController c;
35 Connector connector;
36
37
38
39 protected void setUp() throws ChukwaAgent.AlreadyRunningException {
40 config = new Configuration();
41 agent = new ChukwaAgent(config);
42 c = new ChukwaAgentController();
43 connector = new ChunkCatcherConnector();
44 connector.start();
45 }
46
47 protected void tearDown() {
48 System.out.println("in tearDown()");
49 connector.shutdown();
50 }
51
52 public void testAddFile() {
53 String appType = "junit_addFileTest";
54 String params = "testFile";
55 try {
56
57 System.out.println("Adding adaptor with filename: " + params);
58 String adaptorID = c.addFile(appType, params);
59 System.out.println("Successfully added adaptor, id is:" + adaptorID);
60
61
62
63 Map<String, ChukwaAgentController.Adaptor> listResult = c.list();
64 assertTrue(listResult.containsKey(adaptorID));
65 } catch (IOException e) {
66 e.printStackTrace();
67 }
68 }
69
70 }