Add milestone service tests