[FLINK-1208] Update test cases