package chanjarster.weixin.api; import java.io.IOException; import java.io.InputStream; import java.util.ArrayList; import java.util.List; import javax.xml.bind.JAXBException; import javax.xml.bind.annotation.XmlAccessType; import javax.xml.bind.annotation.XmlAccessorType; import javax.xml.bind.annotation.XmlRootElement; import org.apache.commons.lang3.StringUtils; import org.testng.Assert; import org.testng.annotations.BeforeTest; import org.testng.annotations.DataProvider; import org.testng.annotations.Test; import chanjarster.weixin.bean.WxCustomMessage; import chanjarster.weixin.bean.WxMenu; import chanjarster.weixin.bean.WxMenu.WxMenuButton; import chanjarster.weixin.bean.result.WxUploadResult; import chanjarster.weixin.exception.WxErrorException; import chanjarster.weixin.util.xml.XmlTransformer; public class WxServiceTest { private WxServiceImpl wxService; private List media_ids = new ArrayList(); @BeforeTest public void prepare() throws JAXBException { this.wxService = null; InputStream is1 = ClassLoader.getSystemResourceAsStream("test-config.xml"); WxXmlConfigStorage config = XmlTransformer.fromXml(WxXmlConfigStorage.class, is1); this.wxService = new WxServiceImpl(); this.wxService.setWxConfigStorage(config); } @Test() public void testRefreshAccessToken() throws WxErrorException { WxConfigStorage configStorage = wxService.wxConfigStorage; String before = configStorage.getAccessToken(); wxService.refreshAccessToken(); String after = configStorage.getAccessToken(); Assert.assertNotEquals(before, after); Assert.assertTrue(StringUtils.isNotBlank(after)); } @Test(dependsOnMethods = "testRefreshAccessToken", enabled = true) public void sendCustomMessage() throws WxErrorException { WxXmlConfigStorage configProvider = (WxXmlConfigStorage) wxService.wxConfigStorage; WxCustomMessage message = new WxCustomMessage(); message.setMsgtype(WxConsts.MSG_TEXT); message.setTouser(configProvider.getOpenId()); message.setContent("欢迎欢迎,热烈欢迎\n换行测试\n超链接:Hello World"); wxService.sendCustomMessage(message); } @Test(dataProvider = "menu", dependsOnMethods = "testRefreshAccessToken", enabled = true) public void testCreateMenu(WxMenu wxMenu) throws WxErrorException { wxService.createMenu(wxMenu); } @Test(dependsOnMethods = { "testRefreshAccessToken" , "testCreateMenu"}, enabled = true) public void testGetMenu() throws WxErrorException { Assert.assertNotNull(wxService.getMenu()); } @Test(dependsOnMethods = { "testRefreshAccessToken", "testGetMenu"}, enabled = true) public void testDeleteMenu() throws WxErrorException { wxService.deleteMenu(); } @Test(dependsOnMethods = { "testRefreshAccessToken" }, dataProvider="uploadMedia", enabled = true) public void testUploadMedia1(String mediaType, String fileType, String fileName) throws WxErrorException, IOException { InputStream inputStream = ClassLoader.getSystemResourceAsStream(fileName); WxUploadResult res = wxService.uploadMedia(mediaType, fileType, inputStream); Assert.assertNotNull(res.getType()); Assert.assertNotNull(res.getCreated_at()); Assert.assertTrue(res.getMedia_id() != null || res.getThumb_media_id() != null); if (res.getMedia_id() != null) { media_ids.add(res.getMedia_id()); } if (res.getThumb_media_id() != null) { media_ids.add(res.getThumb_media_id()); } } @DataProvider public Object[][] uploadMedia() { return new Object[][] { new Object[] { WxConsts.MEDIA_IMAGE, WxConsts.FILE_JPG, "mm.jpeg" }, new Object[] { WxConsts.MEDIA_VOICE, WxConsts.FILE_MP3, "mm.mp3" }, new Object[] { WxConsts.MEDIA_VIDEO, WxConsts.FILE_MP4, "mm.mp4" }, new Object[] { WxConsts.MEDIA_THUMB, WxConsts.FILE_JPG, "mm.jpeg" } }; } @Test(dependsOnMethods = { "testUploadMedia1" }, dataProvider="downloadMedia") public void testDownloadMedia(String media_id) throws WxErrorException { wxService.downloadMedia(media_id); } @DataProvider public Object[][] downloadMedia() { Object[][] params = new Object[this.media_ids.size()][]; for (int i = 0; i < this.media_ids.size(); i++) { params[i] = new Object[] { this.media_ids.get(i) }; } return params; } @Test(enabled = true) public void testCheckSignature() throws WxErrorException { String timestamp = "23234235423246"; String nonce = "y7didfkcmvnbd90sdofjkiefhsd"; String signature = "77b6651628dfb9a64bfb0d3432ee053ac566a459"; Assert.assertTrue(wxService.checkSignature(timestamp, nonce, signature)); } @DataProvider(name="menu") public Object[][] getMenu() throws JAXBException { WxMenu menu = new WxMenu(); WxMenuButton button1 = new WxMenuButton(); button1.setType("click"); button1.setName("今日歌曲"); button1.setKey("V1001_TODAY_MUSIC"); WxMenuButton button2 = new WxMenuButton(); button2.setType("click"); button2.setName("歌手简介"); button2.setKey("V1001_TODAY_SINGER"); WxMenuButton button3 = new WxMenuButton(); button3.setName("菜单"); menu.getButton().add(button1); menu.getButton().add(button2); menu.getButton().add(button3); WxMenuButton button31 = new WxMenuButton(); button31.setType("view"); button31.setName("搜索"); button31.setUrl("http://www.soso.com/"); WxMenuButton button32 = new WxMenuButton(); button32.setType("view"); button32.setName("视频"); button32.setUrl("http://v.qq.com/"); WxMenuButton button33 = new WxMenuButton(); button33.setType("click"); button33.setName("赞一下我们"); button33.setKey("V1001_GOOD"); button3.getSub_button().add(button31); button3.getSub_button().add(button32); button3.getSub_button().add(button33); return new Object[][] { new Object[] { menu } }; } @XmlRootElement(name = "xml") @XmlAccessorType(XmlAccessType.FIELD) public static class WxXmlConfigStorage extends WxInMemoryConfigStorage { protected String openId; public String getOpenId() { return openId; } public void setOpenId(String openId) { this.openId = openId; } @Override public String toString() { return "SimpleWxConfigProvider [appId=" + appId + ", secret=" + secret + ", accessToken=" + accessToken + ", expiresIn=" + expiresIn + ", token=" + token + ", openId=" + openId + "]"; } } }