EST-410: Add tests