Java TestKit类代码示例

import akka.testkit.javadsl.TestKit; //导入依赖的package包/类
@Test
public void searchCourseOnReceiveTest() {

    TestKit probe = new TestKit(system);
    ActorRef subject = system.actorOf(props);

    Request reqObj = new Request();
    reqObj.setOperation(ActorOperations.SEARCH_COURSE.getValue());
    HashMap<String, Object> innerMap = new HashMap<>();
    innerMap.put(JsonKey.QUERY, "NTP course");
    Map<String, Object> map = new HashMap<>();
    List<String> fields = new ArrayList<String>();
    fields.add("noOfLecture");
    innerMap.put(JsonKey.FIELDS , fields);
    map.put(JsonKey.SEARCH, innerMap);
    reqObj.setRequest(map);
    subject.tell(reqObj, probe.getRef());
    Response res = probe.expectMsgClass(Response.class);
}
 

import akka.testkit.javadsl.TestKit; //导入依赖的package包/类
/**
 * Method to test delete Note when data is invalid.
 * Expected to throw exception
 */
@SuppressWarnings("deprecation")
@Test
public void test9DeleteNoteException(){
  TestKit probe = new TestKit(system);
  ActorRef subject = system.actorOf(props);

  Request actorMessage = new Request();
  Map<String,Object> request = new HashMap<>();
  request.put(JsonKey.REQUESTED_BY , userId);
  request.put(JsonKey.NOTE_ID , noteId+"invalid");
  actorMessage.setRequest(request);
  actorMessage.setOperation(ActorOperations.DELETE_NOTE.getValue());

  subject.tell(actorMessage, probe.getRef());
  ProjectCommonException res= probe.expectMsgClass(duration("10 second"),ProjectCommonException.class);
  if(null != res){
    Assert.assertEquals("You are not authorized.", res.getMessage());
  }
}
 

import akka.testkit.javadsl.TestKit; //导入依赖的package包/类
@Test
public void killAllOnCompensationComplete() throws Exception {
  new TestKit(actorSystem) {{
    CompletableFuture<SagaResponse> future = new CompletableFuture<>();

    ActorRef actor = actorSystem.actorOf(CompletionCallbackActor.props(future));

    actor.tell(context, noSender());
    actor.tell(new CompensateMessage(response), noSender());

    await().atMost(2, TimeUnit.SECONDS)
        .until(() -> actor1.isTerminated() && actor2.isTerminated() && actor.isTerminated());

    assertThat(future.get(), is(response));
  }};
}
 

import akka.testkit.javadsl.TestKit; //导入依赖的package包/类
@Test
public void Z18TestJoinUserOrganisation4(){
  TestKit probe = new TestKit(system);
  ActorRef subject = system.actorOf(props);
  Request reqObj = new Request();
  reqObj.setOperation(ActorOperations.JOIN_USER_ORGANISATION.getValue());
  Map<String, Object> innerMap = new HashMap<>();
  innerMap.put(JsonKey.USER_ID,null);
  innerMap.put(JsonKey.ORGANISATION_ID,null);
  reqObj.getRequest().put(JsonKey.REQUESTED_BY,userIdnew);
  Map<String, Object> request = new HashMap<String, Object>();
  request.put(JsonKey.USER_ORG, innerMap);
  reqObj.setRequest(request);
  subject.tell(reqObj, probe.getRef());
  probe.expectMsgClass(duration("200 second"), ProjectCommonException.class);
}
 

import akka.testkit.javadsl.TestKit; //导入依赖的package包/类
@Test
public void testabAddSkillAgain(){

  TestKit probe = new TestKit(system);
  ActorRef subject = system.actorOf(props);

  Request actorMessage = new Request();
  actorMessage.put(JsonKey.REQUESTED_BY , USER_ID);
  actorMessage.put(JsonKey.ENDORSED_USER_ID , ENDORSED_USER_ID);
  actorMessage.put(JsonKey.SKILL_NAME, skillsList);
  actorMessage.setOperation(ActorOperations.ADD_SKILL.getValue());

  subject.tell(actorMessage, probe.getRef());
  Response res= probe.expectMsgClass(duration("100 second"),Response.class);

}
 

import akka.testkit.javadsl.TestKit; //导入依赖的package包/类
@Test
public void TestACreateUserWithInvalidOrgId() {
  TestKit probe = new TestKit(system);
  ActorRef subject = system.actorOf(props);

  Request reqObj = new Request();
  reqObj.setOperation(ActorOperations.CREATE_USER.getValue());
  Map<String, Object> innerMap = new HashMap<>();
  innerMap.put(JsonKey.USERNAME, "sunbird_dummy_user_212121");
  innerMap.put(JsonKey.EMAIL, "[email protected]");
  innerMap.put(JsonKey.PASSWORD, "password");
  innerMap.put(JsonKey.PROVIDER, "BLR");
  innerMap.put(JsonKey.PHONE, "9874561230");
  innerMap.put(JsonKey.PHONE_VERIFIED, true);
  innerMap.put(JsonKey.EMAIL_VERIFIED, true);
  innerMap.put(JsonKey.REGISTERED_ORG_ID, (orgId+"13215665"));
  
  Map<String, Object> request = new HashMap<String, Object>();
  request.put(JsonKey.USER, innerMap);
  reqObj.setRequest(request);

  subject.tell(reqObj, probe.getRef());
  probe.expectMsgClass(duration("200 second"), ProjectCommonException.class);
}
 

import akka.testkit.javadsl.TestKit; //导入依赖的package包/类
@Test
public void Z17TestAssignRolesWithoutOrgId(){
  TestKit probe = new TestKit(system);
  ActorRef subject = system.actorOf(props);
  Request reqObj = new Request();
  reqObj.setOperation(ActorOperations.ASSIGN_ROLES.getValue());
  Map<String, Object> request = new HashMap<String, Object>();
  request.put(JsonKey.USERNAME,"sunbird_dummy_user_212121");
  request.put(JsonKey.EXTERNAL_ID, "EXT_ID_DUMMY");
  request.put(JsonKey.PROVIDER, "BLR");
  List<String> roles = new ArrayList<>();
  roles.add("CONTENT_REVIEWER");
  request.put(JsonKey.ROLES, roles);
  reqObj.setRequest(request);
  subject.tell(reqObj, probe.getRef());
  probe.expectMsgClass(duration("200 second"), Response.class);
  Map<String,Object> map = ElasticSearchUtil.getDataByIdentifier(ProjectUtil.EsIndex.sunbird.getIndexName(), ProjectUtil.EsType.user.getTypeName(), userId);
  System.out.println("Login Id "+map.get(JsonKey.LOGIN_ID));
  List<Map<String,Object>> usrOrgList = (List<Map<String, Object>>) map.get(JsonKey.ORGANISATIONS);
  for(Map<String,Object> usrOrg : usrOrgList){
    if(orgId.equalsIgnoreCase((String)usrOrg.get(JsonKey.ID))){
      assertTrue(((List<String>)map.get(JsonKey.ROLES)).contains("CONTENT_REVIEWER"));
    }
  }
}
 

import akka.testkit.javadsl.TestKit; //导入依赖的package包/类
public void testA1CreateBatchWithInvalidCourseId(){
  TestKit probe = new TestKit(system);
  ActorRef subject = system.actorOf(props);
  Request reqObj = new Request();
  reqObj.setOperation(ActorOperations.CREATE_BATCH.getValue());
  HashMap<String, Object> innerMap = new HashMap<>();
  innerMap.put(JsonKey.COURSE_ID, "12345");
  innerMap.put(JsonKey.NAME, "DUMMY_COURSE_NAME1");
  innerMap.put(JsonKey.ENROLLMENT_TYPE, "invite-only");
  innerMap.put(JsonKey.START_DATE , (String)format.format(new Date()));
  innerMap.put(JsonKey.HASHTAGID ,hashTagId );
  Calendar now =  Calendar.getInstance();
  now.add(Calendar.DAY_OF_MONTH, 5);
  Date after5Days = now.getTime();
  innerMap.put(JsonKey.END_DATE , (String)format.format(after5Days));
  reqObj.getRequest().put(JsonKey.BATCH, innerMap);
  subject.tell(reqObj, probe.getRef());
  probe.expectMsgClass(duration("1000 second"),ProjectCommonException.class);
}
 

import akka.testkit.javadsl.TestKit; //导入依赖的package包/类
@Test
public void tellNodeResponseToAllChildren() throws Exception {
  new TestKit(actorSystem) {{
    addChildren(getRef());

    ActorRef parent = someActor();
    context.addParent(requestId, parent);

    when(request.parents()).thenReturn(new String[] {parentRequestId1});
    when(task.commit(request, Operation.SUCCESSFUL_SAGA_RESPONSE)).thenReturn(response);

    ActorRef actorRef = actorSystem.actorOf(RequestActor.props(context, task, request));

    actorRef.tell(new TransactMessage(request1, Operation.SUCCESSFUL_SAGA_RESPONSE), parent);

    List<SagaResponse> responses = receiveN(2, duration("2 seconds")).stream()
        .map(o -> ((TransactMessage) o).response())
        .collect(Collectors.toList());

    assertThat(responses, containsInAnyOrder(response, response));

    verify(task).commit(request, Operation.SUCCESSFUL_SAGA_RESPONSE);
  }};
}
 

import akka.testkit.javadsl.TestKit; //导入依赖的package包/类
@Test
public void tellAllRelativesToAbortOnError() throws Exception {
  new TestKit(actorSystem) {{
    context.addChild(requestId, getRef());
    context.addActor(requestId, getRef());

    context.addParent(requestId, getRef());

    when(request.parents()).thenReturn(new String[] {parentRequestId1});
    when(task.commit(request, Operation.SUCCESSFUL_SAGA_RESPONSE)).thenThrow(exception);

    ActorRef actorRef = actorSystem.actorOf(RequestActor.props(context, task, request));

    actorRef.tell(new TransactMessage(request1, Operation.SUCCESSFUL_SAGA_RESPONSE), getRef());

    List<SagaResponse> responses = receiveN(2, duration("2 seconds")).stream()
        .map(o -> ((AbortMessage) o).response())
        .collect(Collectors.toList());

    assertThat(responses, containsInAnyOrder(instanceOf(FailedSagaResponse.class), instanceOf(FailedSagaResponse.class)));
  }};
}
 

import akka.testkit.javadsl.TestKit; //导入依赖的package包/类
@Test
public void tellAllRelativesExceptSenderToAbortOnAbort() throws Exception {
  new TestKit(actorSystem) {{
    context.addChild(requestId, getRef());
    context.addActor(requestId, getRef());

    context.addParent(requestId, getRef());

    when(request.parents()).thenReturn(new String[] {parentRequestId1});

    ActorRef actorRef = actorSystem.actorOf(RequestActor.props(context, task, request));

    actorRef.tell(new AbortMessage(exception), someActor());

    List<SagaResponse> responses = receiveN(2, duration("2 seconds")).stream()
        .map(o -> ((AbortMessage) o).response())
        .collect(Collectors.toList());

    assertThat(responses, containsInAnyOrder(instanceOf(FailedSagaResponse.class), instanceOf(FailedSagaResponse.class)));

    actorRef.tell(new AbortMessage(exception), someActor());
    expectNoMsg(duration("500 milliseconds"));
  }};
}
 

import akka.testkit.javadsl.TestKit; //导入依赖的package包/类
@Test
public void doNotCompensateIfTransactionIsNotCompleted() throws Exception {
  new TestKit(actorSystem) {{
    addChildren(someActor());
    context.addParent(requestId, getRef());

    when(request.parents()).thenReturn(new String[] {parentRequestId1});

    ActorRef actorRef = actorSystem.actorOf(RequestActor.props(context, task, request));

    actorRef.tell(new AbortMessage(exception), noSender());
    actorRef.tell(compensateMessage, getRef());
    actorRef.tell(compensateMessage, getRef());

    List<Object> responses = receiveN(2, duration("2 seconds"));
    assertThat(responses, contains(instanceOf(AbortMessage.class), instanceOf(CompensateMessage.class)));
    verify(task, never()).compensate(request);

    // no duplicate compensation
    reset(task);
    actorRef.tell(compensateMessage, getRef());
    actorRef.tell(compensateMessage, getRef());
    expectNoMsg(duration("200 milliseconds"));
    verify(task, never()).compensate(request);
  }};
}
 

import akka.testkit.javadsl.TestKit; //导入依赖的package包/类
@Test
public void tellTransactionResponseToChildrenOnRecovery() throws Exception {
  new TestKit(actorSystem) {{
    context.addChild(requestId, getRef());

    ActorRef parent = someActor();
    context.addParent(requestId, parent);

    when(request.parents()).thenReturn(new String[] {parentRequestId1});

    ActorRef actorRef = actorSystem.actorOf(RequestActor.props(context, task, request));

    actorRef.tell(new TransactionRecoveryMessage(response), noSender());
    actorRef.tell(new TransactMessage(request1, Operation.SUCCESSFUL_SAGA_RESPONSE), parent);

    List<SagaResponse> responses = receiveN(1, duration("2 seconds")).stream()
        .map(o -> ((TransactMessage) o).response())
        .collect(Collectors.toList());

    assertThat(responses, containsInAnyOrder(response));

    verify(task, never()).commit(request, Operation.SUCCESSFUL_SAGA_RESPONSE);
  }};
}
 

import akka.testkit.javadsl.TestKit; //导入依赖的package包/类
@Test
public void testCreateTanentPreferenceWithInvalidOrgId(){

  TestKit probe = new TestKit(system);
  ActorRef subject = system.actorOf(props);
  Request actorMessage = new Request();
  List<Map<String , Object>> reqList = new ArrayList<>();

  Map<String , Object> map = new HashMap<>();
  map.put(JsonKey.ROLE , "admin");
  reqList.add(map);

  actorMessage.getRequest().put(JsonKey.TENANT_PREFERENCE , reqList);
  actorMessage.getRequest().put(JsonKey.ROOT_ORG_ID , "");
  actorMessage.getRequest().put(JsonKey.REQUESTED_BY , USER_ID);
  actorMessage.setOperation(ActorOperations.CREATE_TENANT_PREFERENCE.getValue());

  subject.tell(actorMessage, probe.getRef());
  ProjectCommonException res= probe.expectMsgClass(duration("100 second"),ProjectCommonException.class);
}
 

import akka.testkit.javadsl.TestKit; //导入依赖的package包/类
@Test
public void Z18TestJoinUserOrganisation3(){
  TestKit probe = new TestKit(system);
  ActorRef subject = system.actorOf(props);
  Request reqObj = new Request();
  reqObj.setOperation(ActorOperations.JOIN_USER_ORGANISATION.getValue());
  Map<String, Object> innerMap = new HashMap<>();
  innerMap.put(JsonKey.USER_ID,userId);
  innerMap.put(JsonKey.ORGANISATION_ID,(orgId2+"456as"));
  reqObj.getRequest().put(JsonKey.REQUESTED_BY,userIdnew);
  Map<String, Object> request = new HashMap<String, Object>();
  request.put(JsonKey.USER_ORG, innerMap);
  reqObj.setRequest(request);
  subject.tell(reqObj, probe.getRef());
  probe.expectMsgClass(duration("200 second"), ProjectCommonException.class);
}