allocationRegressionTest.java

@Test
public void allocationRegressionTest(){
  long limit = 1<<20;
  ByteWatcher bw = new ByteWatcher();
  bw.onByteWatch((t, size) ->
      //Allocation breached - cause an assertion to happen
      assertTrue(String.format("%s exceeded limit: %d using: %d%n",
            t.getName(), limit, size), false)
      , limit);
  //Run your program here
  //.....
  //.....
  //If the bytes limit is breached ByteWatcher will call back the method
  //onByteWatch and an assertion will be triggered. 
}