Where to allocate unit tests for this class?