Aiml机器人程序

/* Copyleft (C) 2005 Hélio Perroni Filho xperroni@bol.com.br ICQ: 2490863 This file is part of ChatterBean. ChatterBean is free software; you can redistribute it and/or modify it under the terms of the GNU General Public License as published by the Free Software Foundation; either version 2 of the License, or (at your option) any later version. ChatterBean is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for more details. You should have received a copy of the GNU General Public License along with ChatterBean (look at the Documents/ directory); if not, either write to the Free Software Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA, or visit (http://www.gnu.org/licenses/gpl.txt). */ package bitoflife.chatterbean.aiml; import bitoflife.chatterbean.AliceBotMother; import junit.framework.TestCase; import org.xml.sax.helpers.AttributesImpl; public class AIMLHandlerTest extends TestCase { /* Attribute Section */ private AIMLHandler handler; private AIMLStack stack; private AliceBotMother aliceBot = new AliceBotMother(); /* Event Section */ protected void setUp() throws Exception { handler = new AIMLHandler(); stack = handler.stack; } protected void tearDown() { handler = null; } /* Method Section */ private char[] toCharArray(String string) { int n = string.length(); char[] chars = new char[n]; string.getChars(0, n, chars, 0); return chars; } public void testCharacters() throws Exception { char[] text; AttributesImpl attributes = new AttributesImpl(); handler.startElement(null, null, "template", attributes); handler.characters(text = toCharArray("Hello."), 0, text.length); handler.endElement(null, null, "template"); Text expected = new Text("Hello."); Text actual = (Text) ((Template) stack.peek()).children().get(0); assertEquals(expected, actual); } public void testAiml() throws Exception { char[] text; AttributesImpl attributes = new AttributesImpl(); AttributesImpl aimlAtts = new AttributesImpl(); aimlAtts.addAttribute(null, "version", null, "String", "1.0.1"); handler.startElement(null, null, "aiml", aimlAtts); handler.startElement(null, null, "category", attributes); handler.startElement(null, null, "pattern", attributes); handler.characters(text = toCharArray("HELLO ALICE I AM *"), 0, text.length); handler.endElement(null, null, "pattern"); handler.startElement(null, null, "template", attributes); handler.characters(text = toCharArray("Hello "), 0, text.length); handler.startElement(null, null, "star", attributes); handler.characters(text = toCharArray(", nice to meet you."), 0, text.length); handler.endElement(null, null, "template"); handler.endElement(null, null, "category"); handler.endElement(null, null, "aiml"); Aiml actual = (Aiml) stack.peek(); Aiml expected = new Aiml(new Category(new Pattern("HELLO ALICE I AM *"), new Template("Hello ", new Star(1), ", nice to meet you."))); assertEquals(expected, actual); assertEquals("1.0.1", actual.getVersion()); } public void testBot() throws Exception { AttributesImpl attributes = new AttributesImpl(); attributes.addAttribute(null, "name", null, "String", "bot.predicate1"); handler.startElement(null, null, "bot", attributes); Bot expected = new Bot("bot.predicate1"); Bot actual = (Bot) stack.peek(); assertEquals(expected, actual); Bot bot2 = new Bot("bot.predicate2"); assertFalse(bot2.equals(actual)); } public void testCategory() throws Exception { char[] text; AttributesImpl attributes = new AttributesImpl(); handler.startElement(null, null, "category", attributes); handler.startElement(null, null, "pattern", attributes); handler.characters(text = toCharArray("HELLO ALICE I AM *"), 0, text.length); handler.endElement(null, null, "pattern"); handler.startElement(null, null, "that", attributes); handler.characters(text = toCharArray("TEST"), 0, text.length); handler.endElement(null, null, "that"); handler.startElement(null, null, "template", attributes); handler.characters(text = toCharArray("Hello "), 0, text.length); handler.startElement(null, null, "star", attributes); handler.characters(text = toCharArray(", nice to meet you."), 0, text.length); handler.endElement(null, null, "template"); handler.endElement(null, null, "category"); Category actual = (Category) stack.peek(); Category expected = new Category(new Pattern("HELLO ALICE I AM *"), new That("TEST"), new Template("Hello ", new Star(1), ", nice to meet you.")); assertEquals(expected, actual); } public void testCondition() throws Exception { char[] text; AttributesImpl attributes = new AttributesImpl(); handler.startElement(null, null, "condition", attributes); attributes = new AttributesImpl(); attributes.addAttribute(null, null, "name", "String", "condition1"); attributes.addAttribute(null, null, "value", "String", "test value1"); handler.startElement(null, null, "li", attributes); handler.characters(text = toCharArray("Conditioned output 1"), 0, text.length); handler.endElement(null, null, "li"); attributes = new AttributesImpl(); handler.startElement(null, null, "li", attributes); handler.characters(text = toCharArray("Else output"), 0, text.length); handler.endElement(null, null, "li"); handler.endElement(null, null, "condition"); Condition actual = (Condition) stack.peek(); Condition expected = new Condition(null, null, new Li("condition1", "test value1", "Conditioned output 1"), new Li(null, null, "Else output")); assertEquals(expected, actual); attributes = new AttributesImpl(); attributes.addAttribute(null, null, "name", "String", "condition"); handler.startElement(null, null, "condition", attributes); attributes = new AttributesImpl(); attributes.addAttribute(null, null, "value", "String", "test value1"); handler.startElement(null, null, "li", attributes); handler.characters(text = toCharArray("Conditioned output 1"), 0, text.length); handler.endElement(null, null, "li"); attributes = new AttributesImpl(); handler.startElement(null, null, "li", attributes); handler.characters(text = toCharArray("Else output"), 0, text.length); handler.endElement(null, null, "li"); handler.endElement(null, null, "condition"); actual = (Condition) stack.peek(); expected = new Condition("condition1", null, new Li(null, "test value1", "Conditioned output 1"), new Li(null, null, "Else output")); assertEquals(expected, actual); attributes = new AttributesImpl(); attributes.addAttribute(null, null, "name", "String", "condition"); attributes.addAttribute(null, null, "value", "String", "test value1"); handler.startElement(null, null, "condition", attributes); handler.characters(text = toCharArray("Conditioned output"), 0, text.length); handler.endElement(null, null, "condition"); expected = new Condition("condition1", "test value1", "Conditioned output"); actual = (Condition) stack.peek(); assertEquals(expected, actual); } public void testDate() throws Exception { AttributesImpl attributes = new AttributesIm