[FLAKY-TEST-FIX][STREAMING][TEST] Make sure StreamingContexts are shutdown after test

[tdas]

No description provided.

[tdas]

@zsxwing Please take a look. Most of the change is in indentation.

[SparkQA]

Test build #47128 has finished for PR 10124 at commit a66723e.

• This patch passes all tests.
• This patch merges cleanly.
• This patch adds no public classes.

[tdas]

@zsxwing Please merge this to older branches if possible.

[SparkQA]

Test build #2156 has finished for PR 10124 at commit a66723e.

• This patch passes all tests.
• This patch merges cleanly.
• This patch adds no public classes.

[SparkQA]

Test build #2157 has finished for PR 10124 at commit a66723e.

• This patch passes all tests.
• This patch merges cleanly.
• This patch adds no public classes.

[JoshRosen]

Diff minus whitespace changes: https://github.com/apache/spark/pull/10124/files?w=0

[zsxwing]

This one doesn't fix the flaky test. Instead, it just makes sure StreamingContexts will be shutdown if the tests fail. Right?

[tdas]

It is kindof. One test failure, would cause many subsequent tests to fail. So it is a fix for flakiness of the subsequent tests in the suite :)

[zsxwing]

LGTM