[rpc][profiler] add a test case to verify record_function context manager works (#34511)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/34511
With https://github.com/pytorch/pytorch/pull/34122/files, issues
with using record_function context manager and profiling RPCs were fixed. This
adds a test case to verify that we can use RPC with the `record_function`
decorator.
ghstack-source-id: 100109932
Test Plan: Unit test change
Differential Revision: D20352242
fbshipit-source-id: d6429e4352ad3b8d874dc0f27b23ecb6202e6b2b
