Author: norman
Date: Thu Jul 27 06:05:27 2006
New Revision: 426056

URL: http://svn.apache.org/viewvc?rev=426056&view=rev
Log:
Junit test for TarpitHandler

Added:
    
james/server/trunk/src/test/org/apache/james/smtpserver/TarpitHandlerTest.java

Added: 
james/server/trunk/src/test/org/apache/james/smtpserver/TarpitHandlerTest.java
URL: 
http://svn.apache.org/viewvc/james/server/trunk/src/test/org/apache/james/smtpserver/TarpitHandlerTest.java?rev=426056&view=auto
==============================================================================
--- 
james/server/trunk/src/test/org/apache/james/smtpserver/TarpitHandlerTest.java 
(added)
+++ 
james/server/trunk/src/test/org/apache/james/smtpserver/TarpitHandlerTest.java 
Thu Jul 27 06:05:27 2006
@@ -0,0 +1,65 @@
+/****************************************************************
+ * Licensed to the Apache Software Foundation (ASF) under one   *
+ * or more contributor license agreements.  See the NOTICE file *
+ * distributed with this work for additional information        *
+ * regarding copyright ownership.  The ASF licenses this file   *
+ * to you under the Apache License, Version 2.0 (the            *
+ * "License"); you may not use this file except in compliance   *
+ * with the License.  You may obtain a copy of the License at   *
+ *                                                              *
+ *   http://www.apache.org/licenses/LICENSE-2.0                 *
+ *                                                              *
+ * Unless required by applicable law or agreed to in writing,   *
+ * software distributed under the License is distributed on an  *
+ * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY       *
+ * KIND, either express or implied.  See the License for the    *
+ * specific language governing permissions and limitations      *
+ * under the License.                                           *
+ ****************************************************************/
+package org.apache.james.smtpserver;
+
+import org.apache.avalon.framework.container.ContainerUtil;
+import org.apache.james.smtpserver.core.filter.fastfail.TarpitHandler;
+import org.apache.james.test.mock.avalon.MockLogger;
+
+import junit.framework.TestCase;
+
+public class TarpitHandlerTest extends TestCase {
+
+    private SMTPSession session;
+
+    private SMTPSession setupMockedSession(final int rcptCount) {
+        session = new AbstractSMTPSession() {
+
+            public int getRcptCount() {
+                return rcptCount;
+            }
+
+        };
+
+        return session;
+    }
+
+    public void testTarpit() {
+        long tarpitTime = 1000;
+        long startTime;
+        TarpitHandler handler = new TarpitHandler();
+
+        ContainerUtil.enableLogging(handler, new MockLogger());
+
+        handler.setTarpitRcptCount(2);
+        handler.setTarpitSleepTime(tarpitTime);
+
+        // no tarpit used
+        startTime = System.currentTimeMillis();
+        handler.onCommand(setupMockedSession(0));
+        assertTrue("No tarpit",
+                (System.currentTimeMillis() - startTime) < tarpitTime);
+
+        // tarpit used
+        startTime = System.currentTimeMillis();
+        handler.onCommand(setupMockedSession(3));
+        assertTrue("tarpit",
+                (System.currentTimeMillis() - startTime) >= tarpitTime);
+    }
+}



---------------------------------------------------------------------
To unsubscribe, e-mail: [EMAIL PROTECTED]
For additional commands, e-mail: [EMAIL PROTECTED]

Reply via email to