[Module Main] Check the correct implementation between Task and Context

Write tests to verify the correct behavior of the association between Task and Context.