PictureUrlHelperTest.java

package org.xmx0632.deliciousfruit.util;

import static org.junit.Assert.assertEquals;

import org.junit.Before;
import org.junit.Test;
import org.mockito.Mock;
import org.mockito.Mockito;
import org.mockito.MockitoAnnotations;
import org.xmx0632.deliciousfruit.api.v1.bo.TerminalType;
import org.xmx0632.deliciousfruit.api.v1.helper.PictureUrlHelper;
import org.xmx0632.deliciousfruit.global.ConfigConstant;
import org.xmx0632.deliciousfruit.service.ConfigService;

public class PictureUrlHelperTest {
	@Mock
	private ConfigService configService;

	@Before
	public void setUp() {
		MockitoAnnotations.initMocks(this);

		Mockito.when(configService.getByName(TerminalType.IOS_NORMAL))
				.thenReturn("ios_normal");
		Mockito.when(configService.getByName(TerminalType.IOS_RETINA))
				.thenReturn("ios_retina");

		Mockito.when(configService.getByName(ConfigConstant.STATIC_ROOT))
				.thenReturn("c://static//upload//");

		Mockito.when(configService.getByName(ConfigConstant.CATEGORY_DIR))
				.thenReturn("category");

		Mockito.when(configService.getByName(ConfigConstant.FRUIT_STORY_DIR))
				.thenReturn("fruit_story");

		Mockito.when(configService.getByName(ConfigConstant.FRUIT_MENU_DIR))
				.thenReturn("fruit_menu");

		Mockito.when(
				configService.getByName(ConfigConstant.FRUIT_PROCEDURE_DIR))
				.thenReturn("fruit_procedure");
		Mockito.when(configService.getByName(ConfigConstant.WATER_MARK_SUFFIX))
				.thenReturn("_vm");
	}

	@Test
	public void testToWaterMarkFileName() {
		String picUrl1 = "fruit_story/111.jpg";
		String vmPicUrl1 = PictureUrlHelper.toWaterMarkFileName(picUrl1,
				configService);

		assertEquals("fruit_story/111_vm.jpg", vmPicUrl1);

		String picUrl2 = "http://127.0.0.1/deliciousfruit/ios_retina/fruit_story/111.jpg";
		String vmPicUrl2 = PictureUrlHelper.toWaterMarkFileName(picUrl2,
				configService);

		assertEquals(
				"http://127.0.0.1/deliciousfruit/ios_retina/fruit_story/111_vm.jpg",
				vmPicUrl2);

	}
}