Add tests for file extension check