Dolosbattery: Repair broken tests

Repair broken tests

BUG=None
TEST=pytest dolosbattery
TEST=pytest doloscmd

Change-Id: I7c6f152610c9e338b1a5da2037d13ab971868af9
Reviewed-on: https://chromium-review.googlesource.com/c/chromiumos/platform/dolos/+/5855466
Commit-Queue: Keith Haddow <haddowk@chromium.org>
Tested-by: Brian Nemec <bnemec@google.com>
Reviewed-by: Keith Haddow <haddowk@chromium.org>
Auto-Submit: Brian Nemec <bnemec@google.com>