tests: Modify test_set_get_goal to also test custom goals

Change-Id: I75e2bdadcc46ffc42fc01f3ec6e08b508383c42c