Add traffic duration support in test case