add menu support

src/test/java/chanjarster/weixin/api/WxServiceTest.java

@@ -9,68 +9,117 @@ 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.api.WxConsts;
-import chanjarster.weixin.api.WxMemoryConfigProvider;
-import chanjarster.weixin.api.WxService;
-import chanjarster.weixin.api.WxServiceImpl;
 import chanjarster.weixin.bean.WxCustomMessage;
+import chanjarster.weixin.bean.WxMenu;
+import chanjarster.weixin.bean.WxMenu.WxMenuButton;
 import chanjarster.weixin.exception.WxErrorException;
 import chanjarster.weixin.util.XmlTransformer;
 
 public class WxServiceTest {
 
-  @Test(dataProvider = "configs")
-  public void testRefreshAccessToken(WxTestConfigProvider config) throws WxErrorException {
-    String before = config.getAccessToken();
-    
-    WxService wxService = new WxServiceImpl();
-    wxService.setWxConfigProvider(config);
+  private WxServiceImpl wxService;
+  
+  @BeforeTest
+  public void prepare() throws JAXBException {
+    InputStream is1 = ClassLoader.getSystemResourceAsStream("test-config.xml");
+    WxXmlConfigStorage config1 = XmlTransformer.fromXml(WxXmlConfigStorage.class, is1);
+    this.wxService = new WxServiceImpl();
+    this.wxService.setWxConfigProvider(config1);
+  }
+  
+  @Test
+  public void testRefreshAccessToken() throws WxErrorException {
+    WxConfigStorage configProvider = wxService.wxConfigProvider;
+    String before = configProvider.getAccessToken();
     wxService.refreshAccessToken();
     
-    String after = config.getAccessToken();
+    String after = configProvider.getAccessToken();
     
     Assert.assertNotEquals(before, after);
     Assert.assertTrue(StringUtils.isNotBlank(after));
   }
   
-  @Test(dataProvider = "configs")
-  public void sendCustomMessage(WxTestConfigProvider config) throws WxErrorException {
-    WxService wxService = new WxServiceImpl();
-    wxService.setWxConfigProvider(config);
-    
+  @Test(dependsOnMethods = "testRefreshAccessToken", enabled = false)
+  public void sendCustomMessage() throws WxErrorException {
+    WxXmlConfigStorage configProvider = (WxXmlConfigStorage) wxService.wxConfigProvider;
     WxCustomMessage message = new WxCustomMessage();
-    message.setMsgtype(WxConsts.TEXT);
-    message.setTouser(config.getOpenId());
-    message.setContent("欢迎使用教务系统微信公众号\n下面\n<a href=\"http://192.168.1.249:9180/eams-rc/login.action\">Hello World</a>");
+    message.setMsgtype(WxConsts.MSG_TEXT);
+    message.setTouser(configProvider.getOpenId());
+    message.setContent("欢迎使用教务系统微信公众号\n下面\n<a href=\"http://www.baidu.com\">Hello World</a>");
 
     wxService.sendCustomMessage(message);
   }
-  /**
-   * 返回新的access_token
-   * @return
-   * @throws JAXBException 
-   */
-  @DataProvider(name = "configs")
-  public Object[][] getConfig() throws JAXBException {
-    /**
-     * 将 src/test/resources/test-config.sample.xml 改成 test-config.xml 并设置appId, secret, 一个过期的accessToken
-     */
-    // 没有access_token
-    InputStream is1 = ClassLoader.getSystemResourceAsStream("test-config.xml");
-    WxTestConfigProvider config1 = XmlTransformer.fromXml(WxTestConfigProvider.class, is1);
+  
+  @Test(dataProvider = "menu", enabled = true, dependsOnMethods = "testRefreshAccessToken")
+  public void testCreateMenu(WxMenu wxMenu) throws WxErrorException {
+    wxService.createMenu(wxMenu);
+  }
+  
+  @Test(dependsOnMethods = { "testRefreshAccessToken" , "testCreateMenu"})
+  public void testGetMenu() throws WxErrorException {
+    Assert.assertNotNull(wxService.getMenu());
+  }
+  
+  @Test(dependsOnMethods = { "testRefreshAccessToken", "testGetMenu" }, enabled = false)
+  public void testDeleteMenu() throws WxErrorException {
+    wxService.deleteMenu();
+  }
+  
+  @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[] {
-            config1
+            menu
         }
     };
+  
   }
   
+  
   @XmlRootElement(name = "xml")
   @XmlAccessorType(XmlAccessType.FIELD)
-  public static class WxTestConfigProvider extends WxMemoryConfigProvider {
+  public static class WxXmlConfigStorage extends WxInMemoryConfigStorage {
     
     protected String openId;