1 /* 2 * Licensed to the Apache Software Foundation (ASF) under one or more 3 * contributor license agreements. See the NOTICE file distributed with 4 * this work for additional information regarding copyright ownership. 5 * The ASF licenses this file to You under the Apache License, Version 2.0 6 * (the "License"); you may not use this file except in compliance with 7 * the License. You may obtain a copy of the License at 8 * 9 * http://www.apache.org/licenses/LICENSE-2.0 10 * 11 * Unless required by applicable law or agreed to in writing, software 12 * distributed under the License is distributed on an "AS IS" BASIS, 13 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 14 * See the License for the specific language governing permissions and 15 * limitations under the License. 16 */ 17 package org.apache.commons.mail.resolver; 18 19 import java.io.IOException; 20 import java.io.InputStream; 21 22 import javax.activation.DataSource; 23 24 import org.apache.commons.io.IOUtils; 25 26 public abstract class AbstractDataSourceResolverTest { 27 28 protected final int IMG_SIZE = 5866; 29 30 protected byte[] toByteArray(final DataSource dataSource) throws IOException 31 { 32 if(dataSource != null) 33 { 34 final InputStream is = dataSource.getInputStream(); 35 try { 36 return IOUtils.toByteArray(is); 37 } finally { 38 is.close(); 39 } 40 } 41 return null; 42 } 43 44 }