(String[][] data)
| 291 | } |
| 292 | |
| 293 | public void storeFunction(String[][] data) throws Throwable { |
| 294 | |
| 295 | File tmpFile=TestHelper.createTempFile(data) ; |
| 296 | |
| 297 | //Load, Execute and Store query |
| 298 | String query = "foreach (load '" |
| 299 | + Util.generateURI(tmpFile.toString(), pig.getPigContext()) |
| 300 | + "') generate $0,$1;"; |
| 301 | System.out.println(query); |
| 302 | pig.registerQuery("asdf_id = " + query); |
| 303 | try { |
| 304 | pig.deleteFile("frog"); |
| 305 | } catch(Exception e) {} |
| 306 | pig.store("asdf_id", "frog", MyStorage.class.getName()+"()"); |
| 307 | |
| 308 | |
| 309 | InputStream is = FileLocalizer.open("frog", pig.getPigContext()); |
| 310 | BufferedReader br = new BufferedReader(new InputStreamReader(is)); |
| 311 | String line; |
| 312 | |
| 313 | //verify query |
| 314 | int i= 0; |
| 315 | while((line = br.readLine()) != null) { |
| 316 | |
| 317 | assertEquals( data[i][0] + '-' + data[i][1], line ); |
| 318 | i++; |
| 319 | } |
| 320 | |
| 321 | br.close(); |
| 322 | |
| 323 | try { |
| 324 | pig.deleteFile("frog"); |
| 325 | } catch(Exception e) {} |
| 326 | tmpFile.delete(); |
| 327 | |
| 328 | } |
| 329 | |
| 330 | |
| 331 | @Test |
no test coverage detected