Convert to junit test