11d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert/* 21d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert * Copyright (C) 2008 The Guava Authors 31d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert * 41d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert * Licensed under the Apache License, Version 2.0 (the "License"); 51d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert * you may not use this file except in compliance with the License. 61d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert * You may obtain a copy of the License at 71d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert * 81d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert * http://www.apache.org/licenses/LICENSE-2.0 91d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert * 101d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert * Unless required by applicable law or agreed to in writing, software 111d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert * distributed under the License is distributed on an "AS IS" BASIS, 121d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 131d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert * See the License for the specific language governing permissions and 141d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert * limitations under the License. 151d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert */ 161d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert 171d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringertpackage com.google.common.collect.testing; 181d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert 191d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringertimport java.io.ByteArrayInputStream; 201d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringertimport java.io.ByteArrayOutputStream; 211d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringertimport java.io.IOException; 221d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringertimport java.io.ObjectInputStream; 231d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringertimport java.io.ObjectOutputStream; 241d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringertimport java.util.Collection; 251d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringertimport java.util.List; 261d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert 271d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert/** 281d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert * Reserializes the sets created by another test set generator. 291d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert * 301d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert * TODO: make CollectionTestSuiteBuilder test reserialized collections 311d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert * 321d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert * @author Jesse Wilson 331d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert */ 341d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringertpublic class ReserializingTestCollectionGenerator<E> 351d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert implements TestCollectionGenerator<E> { 361d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert private final TestCollectionGenerator<E> delegate; 371d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert 381d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert ReserializingTestCollectionGenerator(TestCollectionGenerator<E> delegate) { 391d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert this.delegate = delegate; 401d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert } 411d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert 421d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert public static <E> ReserializingTestCollectionGenerator<E> newInstance( 431d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert TestCollectionGenerator<E> delegate) { 441d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert return new ReserializingTestCollectionGenerator<E>(delegate); 451d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert } 461d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert 471d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert @Override 481d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert public Collection<E> create(Object... elements) { 491d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert return reserialize(delegate.create(elements)); 501d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert } 511d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert 521d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert @SuppressWarnings("unchecked") 531d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert static <T> T reserialize(T object) { 541d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert try { 551d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert ByteArrayOutputStream bytes = new ByteArrayOutputStream(); 561d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert ObjectOutputStream out = new ObjectOutputStream(bytes); 571d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert out.writeObject(object); 581d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert ObjectInputStream in = new ObjectInputStream( 591d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert new ByteArrayInputStream(bytes.toByteArray())); 601d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert return (T) in.readObject(); 611d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert } catch (IOException e) { 621d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert Helpers.fail(e, e.getMessage()); 631d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert } catch (ClassNotFoundException e) { 641d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert Helpers.fail(e, e.getMessage()); 651d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert } 661d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert throw new AssertionError("not reachable"); 671d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert } 681d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert 691d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert @Override 701d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert public SampleElements<E> samples() { 711d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert return delegate.samples(); 721d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert } 731d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert 741d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert @Override 751d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert public E[] createArray(int length) { 761d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert return delegate.createArray(length); 771d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert } 781d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert 791d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert @Override 801d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert public Iterable<E> order(List<E> insertionOrder) { 811d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert return delegate.order(insertionOrder); 821d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert } 831d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert}