Pass through context for DiagHandler in VFS

This allows the unit tests to not use global state when checking
diagnostics.