1   /*
2    * Licensed to the Apache Software Foundation (ASF) under one
3    * or more contributor license agreements.  See the NOTICE file
4    * distributed with this work for additional information
5    * regarding copyright ownership.  The ASF licenses this file
6    * to you under the Apache License, Version 2.0 (the
7    * "License"); you may not use this file except in compliance
8    * with the License.  You may obtain a copy of the License at
9    *
10   *   http://www.apache.org/licenses/LICENSE-2.0
11   *
12   * Unless required by applicable law or agreed to in writing,
13   * software distributed under the License is distributed on an
14   * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
15   * KIND, either express or implied.  See the License for the
16   * specific language governing permissions and limitations
17   * under the License.
18   */
19  package org.apache.mina.example.haiku;
20  
21  import org.apache.mina.core.service.IoHandler;
22  import org.apache.mina.core.session.IoSession;
23  import org.jmock.Mock;
24  import org.jmock.MockObjectTestCase;
25  
26  /**
27   * @author Apache Mina Project (dev@mina.apache.org)
28   * @version $Rev: $, $Date:  $
29   */
30  public class HaikuValidatorIoHandlerTest extends MockObjectTestCase {
31      private IoHandler handler;
32  
33      @Override
34      protected void setUp() throws Exception {
35          super.setUp();
36  
37          handler = new HaikuValidatorIoHandler();
38      }
39  
40      public void testValidHaiku() throws Exception {
41          Mock session = mock(IoSession.class);
42          session.expects(once()).method("write").with(eq("HAIKU!"));
43          IoSession sessionProxy = (IoSession) session.proxy();
44  
45          handler.messageReceived(sessionProxy, new Haiku(
46                  "Oh, I drank too much.", "Why, oh why did I sign up",
47                  "For an eight thirty?"));
48      }
49  
50      public void testInvalidHaiku() throws Exception {
51          Mock session = mock(IoSession.class);
52          session.expects(once()).method("write").with(
53                  eq("NOT A HAIKU: phrase 1, 'foo' had 1 syllables, not 5"));
54          IoSession sessionProxy = (IoSession) session.proxy();
55  
56          handler.messageReceived(sessionProxy,
57                  new Haiku("foo", "a haiku", "poo"));
58      }
59  }