Add tests for arena system