union-1016578: Unit test fixes

Signed-off-by: Joe Walker <jwalker@mozilla.com>