1 /* 2 * Copyright (C) 2011 The Libphonenumber Authors 3 * 4 * Licensed under the Apache License, Version 2.0 (the "License"); 5 * you may not use this file except in compliance with the License. 6 * You may obtain a copy of the License at 7 * 8 * http://www.apache.org/licenses/LICENSE-2.0 9 * 10 * Unless required by applicable law or agreed to in writing, software 11 * distributed under the License is distributed on an "AS IS" BASIS, 12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13 * See the License for the specific language governing permissions and 14 * limitations under the License. 15 */ 16 17 package com.android.i18n.phonenumbers.prefixmapper; 18 19 import junit.framework.TestCase; 20 21 import java.io.ByteArrayInputStream; 22 import java.io.ByteArrayOutputStream; 23 import java.io.IOException; 24 import java.io.ObjectInputStream; 25 import java.io.ObjectOutputStream; 26 import java.util.Collections; 27 import java.util.SortedMap; 28 import java.util.TreeMap; 29 30 /** 31 * Unittests for FlyweightMapStorage.java 32 * 33 * @author Philippe Liard 34 */ 35 public class FlyweightMapStorageTest extends TestCase { 36 private static final SortedMap<Integer, String> phonePrefixMap; 37 static { 38 SortedMap<Integer, String> tmpMap = new TreeMap<Integer, String>(); 39 tmpMap.put(331402, "Paris"); 40 tmpMap.put(331434, "Paris"); 41 tmpMap.put(334910, "Marseille"); 42 tmpMap.put(334911, "Marseille"); 43 tmpMap.put(334912, ""); 44 tmpMap.put(334913, ""); 45 phonePrefixMap = Collections.unmodifiableSortedMap(tmpMap); 46 } 47 48 private FlyweightMapStorage mapStorage; 49 50 @Override 51 protected void setUp() throws Exception { 52 mapStorage = new FlyweightMapStorage(); 53 mapStorage.readFromSortedMap(phonePrefixMap); 54 } 55 56 public void testReadFromSortedMap() { 57 assertEquals(331402, mapStorage.getPrefix(0)); 58 assertEquals(331434, mapStorage.getPrefix(1)); 59 assertEquals(334910, mapStorage.getPrefix(2)); 60 assertEquals(334911, mapStorage.getPrefix(3)); 61 62 assertEquals("Paris", mapStorage.getDescription(0)); 63 assertSame(mapStorage.getDescription(0), mapStorage.getDescription(1)); 64 65 assertEquals("Marseille", mapStorage.getDescription(2)); 66 assertSame(mapStorage.getDescription(2), mapStorage.getDescription(3)); 67 } 68 69 public void testReadFromSortedMapSupportsEmptyDescription() { 70 assertEquals(334912, mapStorage.getPrefix(4)); 71 assertEquals(334913, mapStorage.getPrefix(5)); 72 73 assertEquals("", mapStorage.getDescription(4)); 74 assertSame(mapStorage.getDescription(4), mapStorage.getDescription(5)); 75 } 76 77 public void testWriteAndReadExternal() throws IOException { 78 ByteArrayOutputStream byteArrayOutputStream = new ByteArrayOutputStream(); 79 ObjectOutputStream objectOutputStream = new ObjectOutputStream(byteArrayOutputStream); 80 mapStorage.writeExternal(objectOutputStream); 81 objectOutputStream.flush(); 82 83 FlyweightMapStorage newMapStorage = new FlyweightMapStorage(); 84 ObjectInputStream objectInputStream = 85 new ObjectInputStream(new ByteArrayInputStream(byteArrayOutputStream.toByteArray())); 86 newMapStorage.readExternal(objectInputStream); 87 88 String expected = mapStorage.toString(); 89 assertEquals(expected, newMapStorage.toString()); 90 } 91 92 public void testReadExternalThrowsIOExceptionWithMalformedData() throws IOException { 93 ByteArrayOutputStream byteArrayOutputStream = new ByteArrayOutputStream(); 94 ObjectOutputStream objectOutputStream = new ObjectOutputStream(byteArrayOutputStream); 95 objectOutputStream.writeUTF("hello"); 96 objectOutputStream.flush(); 97 ObjectInputStream objectInputStream = 98 new ObjectInputStream(new ByteArrayInputStream(byteArrayOutputStream.toByteArray())); 99 FlyweightMapStorage newMapStorage = new FlyweightMapStorage(); 100 try { 101 newMapStorage.readExternal(objectInputStream); 102 fail(); 103 } catch (IOException e) { 104 // Exception expected. 105 } 106 } 107 } 108