DisconnectClientTaskTestCase.java revision ea1068c292e9b341af6d6b563cd8988a96be20a9
/*
* CDDL HEADER START
*
* The contents of this file are subject to the terms of the
* Common Development and Distribution License, Version 1.0 only
* (the "License"). You may not use this file except in compliance
* with the License.
*
* You can obtain a copy of the license at legal-notices/CDDLv1_0.txt
* or http://forgerock.org/license/CDDLv1.0.html.
* See the License for the specific language governing permissions
* and limitations under the License.
*
* When distributing Covered Code, include this CDDL HEADER in each
* file and include the License file at legal-notices/CDDLv1_0.txt.
* If applicable, add the following below this CDDL HEADER, with the
* fields enclosed by brackets "[]" replaced with your own identifying
* information:
* Portions Copyright [yyyy] [name of copyright owner]
*
* CDDL HEADER END
*
*
* Copyright 2008 Sun Microsystems, Inc.
* Portions Copyright 2014-2015 ForgeRock AS
*/
package org.opends.server.tasks;
import java.net.Socket;
import org.testng.annotations.Test;
import org.testng.annotations.BeforeClass;
import org.opends.server.TestCaseUtils;
import org.forgerock.i18n.LocalizableMessage;
import org.opends.server.backends.task.Task;
import org.opends.server.backends.task.TaskState;
import org.opends.server.extensions.GetConnectionIDExtendedOperation;
import org.opends.server.protocols.ldap.*;
import org.opends.server.types.DN;
import org.opends.server.util.StaticUtils;
import org.forgerock.opendj.ldap.ByteString;
import static org.testng.Assert.*;
import static org.opends.server.util.ServerConstants.*;
/**
* Tests the disconnect client task.
*/
public class DisconnectClientTaskTestCase
extends TasksTestCase
{
/**
* Make sure that the Directory Server is running.
*
* @throws Exception If an unexpected problem occurs.
*/
@BeforeClass
public void startServer()
throws Exception
{
TestCaseUtils.startServer();
}
/**
* Tests the ability of the server to disconnect an arbitrary client
* connection with a notice of disconnection.
*
* @throws Exception If an unexpected problem occurs.
*/
@Test
public void testDisconnectWithNotification()
throws Exception
{
// Establish a connection to the server, bind, and get the connection ID.
Socket s = new Socket("127.0.0.1", TestCaseUtils.getServerLdapPort());
org.opends.server.tools.LDAPReader r =
new org.opends.server.tools.LDAPReader(s);
org.opends.server.tools.LDAPWriter w =
new org.opends.server.tools.LDAPWriter(s);
BindRequestProtocolOp bindRequest =
new BindRequestProtocolOp(ByteString.valueOf("cn=Directory Manager"),
3, ByteString.valueOf("password"));
LDAPMessage message = new LDAPMessage(1, bindRequest);
w.writeMessage(message);
message = r.readMessage();
BindResponseProtocolOp bindResponse = message.getBindResponseProtocolOp();
assertEquals(bindResponse.getResultCode(), LDAPResultCode.SUCCESS);
ExtendedRequestProtocolOp extendedRequest =
new ExtendedRequestProtocolOp(OID_GET_CONNECTION_ID_EXTOP);
message = new LDAPMessage(2, extendedRequest);
w.writeMessage(message);
message = r.readMessage();
ExtendedResponseProtocolOp extendedResponse =
message.getExtendedResponseProtocolOp();
assertEquals(extendedResponse.getResultCode(), LDAPResultCode.SUCCESS);
assertEquals(extendedResponse.getOID(), OID_GET_CONNECTION_ID_EXTOP);
long connectionID = GetConnectionIDExtendedOperation.decodeResponseValue(
extendedResponse.getValue());
// Invoke the disconnect client task.
String taskID = "Disconnect Client " + connectionID;
LocalizableMessage disconnectMessage = LocalizableMessage.raw("testDisconnectWithNotification");
DN taskDN = DN.valueOf("ds-task-id=" + taskID +
",cn=Scheduled Tasks,cn=Tasks");
TestCaseUtils.addEntry(
"dn: " + taskDN.toString(),
"objectClass: top",
"objectClass: ds-task",
"objectClass: ds-task-disconnect",
"ds-task-id: " + taskID,
"ds-task-class-name: org.opends.server.tasks.DisconnectClientTask",
"ds-task-disconnect-connection-id: " + connectionID,
"ds-task-disconnect-notify-client: true",
"ds-task-disconnect-message: " + disconnectMessage);
Task task = getCompletedTask(taskDN);
assertNotNull(task);
assertEquals(task.getTaskState(), TaskState.COMPLETED_SUCCESSFULLY);
// Make sure that we get a notice of disconnection on the initial
// connection.
message = r.readMessage();
extendedResponse = message.getExtendedResponseProtocolOp();
assertEquals(extendedResponse.getOID(),
LDAPConstants.OID_NOTICE_OF_DISCONNECTION);
assertEquals(extendedResponse.getErrorMessage(), disconnectMessage);
StaticUtils.close(s);
}
/**
* Tests the ability of the server to disconnect an arbitrary client
* connection without a notice of disconnection.
*
* @throws Exception If an unexpected problem occurs.
*/
@Test
public void testDisconnectWithoutNotification()
throws Exception
{
// Establish a connection to the server, bind, and get the connection ID.
Socket s = new Socket("127.0.0.1", TestCaseUtils.getServerLdapPort());
org.opends.server.tools.LDAPReader r =
new org.opends.server.tools.LDAPReader(s);
org.opends.server.tools.LDAPWriter w =
new org.opends.server.tools.LDAPWriter(s);
BindRequestProtocolOp bindRequest =
new BindRequestProtocolOp(ByteString.valueOf("cn=Directory Manager"),
3, ByteString.valueOf("password"));
LDAPMessage message = new LDAPMessage(1, bindRequest);
w.writeMessage(message);
message = r.readMessage();
BindResponseProtocolOp bindResponse = message.getBindResponseProtocolOp();
assertEquals(bindResponse.getResultCode(), LDAPResultCode.SUCCESS);
ExtendedRequestProtocolOp extendedRequest =
new ExtendedRequestProtocolOp(OID_GET_CONNECTION_ID_EXTOP);
message = new LDAPMessage(2, extendedRequest);
w.writeMessage(message);
message = r.readMessage();
ExtendedResponseProtocolOp extendedResponse =
message.getExtendedResponseProtocolOp();
assertEquals(extendedResponse.getResultCode(), LDAPResultCode.SUCCESS);
assertEquals(extendedResponse.getOID(), OID_GET_CONNECTION_ID_EXTOP);
long connectionID = GetConnectionIDExtendedOperation.decodeResponseValue(
extendedResponse.getValue());
// Invoke the disconnect client task.
String taskID = "Disconnect Client " + connectionID;
DN taskDN = DN.valueOf("ds-task-id=" + taskID +
",cn=Scheduled Tasks,cn=Tasks");
TestCaseUtils.addEntry(
"dn: " + taskDN.toString(),
"objectClass: top",
"objectClass: ds-task",
"objectClass: ds-task-disconnect",
"ds-task-id: " + taskID,
"ds-task-class-name: org.opends.server.tasks.DisconnectClientTask",
"ds-task-disconnect-connection-id: " + connectionID,
"ds-task-disconnect-notify-client: false");
Task task = getCompletedTask(taskDN);
assertNotNull(task);
assertEquals(task.getTaskState(), TaskState.COMPLETED_SUCCESSFULLY);
// Make sure that the client connection has been closed with no notice of
// disconnection.
assertNull(r.readMessage());
StaticUtils.close(s);
}
}