Add missing import for unittest.

Two changes went in at once: One that removed all usages of the unittest
module, and one that added a new usage. Since we don't run Chromite unit
tests on the CQ currently, both went in and we have broken unit tests.
Fix it.

BUG=chromium:280620
TEST=Run cbuildbot_stages_unittest.py and verify it works now.

Change-Id: I93405477413cdb3aef5a12d7e58d01dd666ee6d0
Reviewed-on: https://chromium-review.googlesource.com/170240
Tested-by: David James <davidjames@chromium.org>
Reviewed-by: Matt Tennant <mtennant@chromium.org>
Commit-Queue: David James <davidjames@chromium.org>
1 file changed