[WIP] Add tests to identity