Fix up unit test for changes since 6114d0f9b
Fix up unit test for changes since 6114d0f9b

Change-Id: I6e6cdf73444f45917ebbbe49f4cbe0d22a667660
(cherry picked from commit b73162a56abf067b7e15d795b480a011d8a3542c)

Linaro/lava-server,Linaro/lava-server,Linaro/lava-server,Linaro/lava-server