Add unit tests for progress bar