[Rust] tokio: test

#Rust

비동기 함수가 하나 있고, 그걸 테스트해야 한다고 생각해보자.

여기서 문제는, 일반 유닛테스트 컨텍스트는 async가 아니라서 awaiting을 제대로 할 수가 없단 것이다.

이걸 동작하게 하려면 직접 tokio 컨텍스트 생성해서 block_on 걸고 하면 되겠지만, 너무 손이 많이 가고 복잡하다.

그래서 그런걸 해주는게 tokio::test라는 숏컷 매크로다.

스레드 몇개 쓸지 정도만 정의해서 대신 박아주면 잘 동작한다.

스레드 개수나 flavor를 지정하지 않으면 current_thread로 동작한다. 그냥 테스트마다 적당히 별도의 스레드로 뜨도록 하는 것이다.

Sleep 처리 대응: start_paused

비동기 코드들은 속도 제어를 위해서 Sleep 등을 통해서 명시적인 대기를 사용할 때가 많다.
문제는, 그렇게 붙이다보면 함수 호출 하나하나가 매우 오래 걸릴 수도 있다는 것이다.

이렇게만 해도 한번 돌때마다 10초씩 걸린다.

시간 대기가 단위테스트에 있어서 유의미한 요소가 아니라면, start_paused 옵션을 사용해서 테스트 속도를 끌어올릴 수 있다.

이게 뭐냐면, 테스트 컨텍스트에 한해서는 sleep 동작을 씹도록 하는 것이다.

내부적으로는 런타임 생성시에 저 플래그를 넣어주는 것인데

저걸 넣으면 paused 동작, 정확히는 tokio::time::sleep과 tokio::time::Interval::tick가 무시되도록 한다.
std:🧵sleep이나 다른 함수들에 대해서는 동작하지 않는다. tokio만을 쓴다는 전제 하에서만 유효하다.

아무튼 딱 이런 테스트 컨텍스트에만 넣을 수 있는 플래그다.