Added battle state unit test