Summary: TSIA

Test Plan: `bazel test //...`

Reviewers: vihang, michelle

Reviewed By: vihang

Differential Revision: https://phab.corp.pixielabs.ai/D9514

GitOrigin-RevId: 218f3cadb14c2a34286d8eae359c0a08928d3b0e