/* * Copyright (C) 2010 The Android Open Source Project * * Licensed 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 com.android.email.mail.transport; import android.test.MoreAsserts; import junit.framework.TestCase; public class DiscourseLoggerTest extends TestCase { /** Shortcut to create a byte array */ private static byte[] b(String chars) { byte[] ret = new byte[chars.length()]; for (int i = 0; i < chars.length(); i++) { ret[i] = (byte) chars.charAt(i); } return ret; } /** Shortcut to create a String array */ private static String[] s(String... strings) { return strings; } /** Shortcut to create an Object array */ private static Object[] o(Object... objects) { return objects; } public void testDiscourseLogger() { checkDiscourseStore(4, o(), s()); checkDiscourseStore(4, o( "command" ), s( "command" )); checkDiscourseStore(4, o( "1a", "2b", "3", "4dd" ), s( "1a", "2b", "3", "4dd" )); checkDiscourseStore(4, o( "1", "2", "3", "4", "5" ), s( "2", "3", "4", "5" )); checkDiscourseStore(4, o( b("A") ), s( "A" )); checkDiscourseStore(4, o( b("A\nB\nC") ), s( "A", "B", "C" )); checkDiscourseStore(4, o( b("A\nBCD\nC\nDEF\u0080\u0001G\r\n") ), s( "A", "BCD", "C", "DEF\\x80\\x01G" )); checkDiscourseStore(4, o( "1", b("2"), "3", b("4\n5\n"), "6 7 8", "7 a bbb ccc", b("* aaa8\n* bbb9\n7 ccc 10") ), s( "7 a bbb ccc", "* aaa8", "* bbb9", "7 ccc 10" )); } private void checkDiscourseStore(int storeSize, Object[] discource, String[] expected) { DiscourseLogger store = new DiscourseLogger(storeSize); for (Object o : discource) { if (o instanceof String) { store.addSentCommand((String) o); } else if (o instanceof byte[]) { for (byte b : (byte[]) o) { store.addReceivedByte(b); } } else { fail("Invalid argument. Test broken."); } } MoreAsserts.assertEquals(expected, store.getLines()); // logLastDiscourse should empty the buffer. store.logLastDiscourse(); assertEquals(0, store.getLines().length); } }