事件背景:为了测试数据提交后,需要在另一个环境的多个测试节点下去验证测试数据是否添加成功,找了一大堆放法,用了比较笨的方法实现了。不多废话思路如下:
/** * 获取文件全部内容 * @param fileName * @return */ public String readToString(String fileName) { String encoding = "UTF-8"; File file = new File(fileName); Long filelength = file.length(); byte[] filecontent = new byte[filelength.intValue()]; try { FileInputStream in = FileInputStream(file); in.read(filecontent); in.close(); } catch (FileNotFoundException e) { e.printStackTrace(); } (IOException e) { e.printStackTrace(); } { return String(filecontent,encoding); } (UnsupportedEncodingException e) { logger.error("The OS does not support " + encoding); e.printStackTrace(); null; } }
* 清空文本内容 * fileName public void clearInfoForFile(String fileName) { File file = File(fileName); if(!file.exists()) { file.createNewFile(); } FileWriter fileWriter = FileWriter(file); fileWriter.write(""); fileWriter.flush(); fileWriter.close(); } (IOException e) { e.printStackTrace(); } }
3.追加一行hosts,也可以多个,视情况写入
* 在已有的文件后面追加信息 * info appendInfoToFile(String fileName,String info) { File file =file.exists()){ file.createNewFile(); } FileWriter fileWriter =new FileWriter(file,true); fileWriter.write(info); fileWriter.flush(); fileWriter.close(); } (IOException e) { e.printStackTrace(); } }
4.打开hosts绑定网站,这里我们以百度为例,提示无法连接。
5.在清空hosts文件,将备份的原内容写入即可完成还原hosts文件操作
6.测试代码如下:
@Test run(){ String hosts =readToString("C:/Windows/System32/drivers/etc/hosts"); clearInfoForFile("C:/Windows/System32/drivers/etc/hosts"); appendInfoToFile("C:/Windows/System32/drivers/etc/hosts","127.0.0.1 www.baidu.com"); System.setProperty("webdriver.chrome.driver","tools/chromedriver.exe"); WebDriver driver= ChromeDriver(); driver.get("https://www.baidu.com/"); { Thread.sleep(5); } (InterruptedException e) { // TODO Auto-generated catch block e.printStackTrace(); } clearInfoForFile("C:/Windows/System32/drivers/etc/hosts"); appendInfoToFile("C:/Windows/System32/drivers/etc/hosts",hosts); }
笨方法实现,先Mark下。