Add tests infrastructure