1818
1919package org .apache .flink .client ;
2020
21+ import static org .junit .Assert .assertEquals ;
2122import static org .junit .Assert .assertTrue ;
2223import static org .junit .Assert .fail ;
2324
25+ import akka .actor .ActorRef ;
26+ import akka .actor .ActorSystem ;
27+ import akka .actor .Props ;
28+ import akka .actor .UntypedActor ;
29+ import akka .testkit .JavaTestKit ;
30+ import org .apache .commons .cli .CommandLine ;
31+ import org .apache .flink .runtime .event .job .RecentJobEvent ;
2432import org .apache .flink .runtime .jobgraph .JobID ;
33+ import org .apache .flink .runtime .messages .EventCollectorMessages ;
34+ import org .apache .flink .runtime .messages .JobManagerMessages ;
35+ import org .apache .flink .runtime .messages .JobResult ;
36+ import org .junit .AfterClass ;
2537import org .junit .BeforeClass ;
2638import org .junit .Test ;
2739
40+ import java .util .ArrayList ;
41+
2842//TODO: Update test case
2943public class CliFrontendListCancelTest {
44+
45+ private static ActorSystem actorSystem ;
46+
47+ @ BeforeClass
48+ public static void setup (){
49+ actorSystem = ActorSystem .create ("TestingActorSystem" );
50+ }
51+
52+ @ AfterClass
53+ public static void teardown (){
54+ JavaTestKit .shutdownActorSystem (actorSystem );
55+ actorSystem = null ;
56+ }
3057
3158 @ BeforeClass
3259 public static void init () {
@@ -57,9 +84,11 @@ public void testCancel() {
5784 {
5885 JobID jid = new JobID ();
5986 String jidString = jid .toString ();
87+
88+ final ActorRef jm = actorSystem .actorOf (Props .create (CliJobManager .class , jid ));
6089
6190 String [] parameters = {"-i" , jidString };
62- InfoListTestCliFrontend testFrontend = new InfoListTestCliFrontend ();
91+ InfoListTestCliFrontend testFrontend = new InfoListTestCliFrontend (jm );
6392 int retCode = testFrontend .cancel (parameters );
6493 assertTrue (retCode == 0 );
6594 }
@@ -75,6 +104,8 @@ public void testCancel() {
75104 @ Test
76105 public void testList () {
77106 try {
107+ final ActorRef jm = actorSystem .actorOf (Props .create (CliJobManager .class , (Object )null ));
108+
78109 // test unrecognized option
79110 {
80111 String [] parameters = {"-v" , "-k" };
@@ -94,7 +125,7 @@ public void testList() {
94125 // test list properly
95126 {
96127 String [] parameters = {"-r" , "-s" };
97- InfoListTestCliFrontend testFrontend = new InfoListTestCliFrontend ();
128+ InfoListTestCliFrontend testFrontend = new InfoListTestCliFrontend (jm );
98129 int retCode = testFrontend .list (parameters );
99130 assertTrue (retCode == 0 );
100131 }
@@ -108,20 +139,36 @@ public void testList() {
108139
109140
110141 protected static final class InfoListTestCliFrontend extends CliFrontendTestUtils .TestingCliFrontend {
142+ private ActorRef jobmanager ;
111143
112- public InfoListTestCliFrontend () {
144+ public InfoListTestCliFrontend (ActorRef jobmanager ){
145+ this .jobmanager = jobmanager ;
113146 }
114147
115- public int getTotalNumberOfRegisteredSlots () {
116- return 1 ;
148+ @ Override
149+ public ActorRef getJobManager (CommandLine line ){
150+ return jobmanager ;
117151 }
152+ }
118153
119- @ Override
120- public int getNumberOfSlotsAvailableToScheduler () throws IOException {
154+ protected static final class CliJobManager extends UntypedActor {
155+ private final JobID jobID ;
156+
157+ public CliJobManager (final JobID jobID ){
158+ this .jobID = jobID ;
159+ }
121160
122161 @ Override
123- public int getBlobServerPort () {
124- throw new UnsupportedOperationException ();
162+ public void onReceive (Object message ) throws Exception {
163+ if (message instanceof JobManagerMessages .RequestAvailableSlots$ ){
164+ getSender ().tell (1 , getSelf ());
165+ }else if (message instanceof EventCollectorMessages .RequestRecentJobEvents$ ) {
166+ getSender ().tell (new EventCollectorMessages .RecentJobs (new ArrayList <RecentJobEvent >()), getSelf ());
167+ }else if (message instanceof JobManagerMessages .CancelJob ){
168+ JobManagerMessages .CancelJob cancelJob = (JobManagerMessages .CancelJob ) message ;
169+ assertEquals (jobID , cancelJob .jobID ());
170+ getSender ().tell (new JobResult .JobCancelResult (JobResult .SUCCESS (), null ), getSelf ());
171+ }
125172 }
126173 }
127174}
0 commit comments