Commit f18e454
committed
More specific GaxError assertion.
1 parent 903c29e commit f18e454
1 file changed
Lines changed: 3 additions & 1 deletion
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
@@ -773,11 +773,13 @@ def test_subscription_pull_deadline_exceeded(self): | |||
| 773 | 773 | self.assertEqual(result, []) | |
| 774 | 774 | ||
| 775 | 775 | def test_subscription_pull_deadline_exceeded_return_immediately(self): | |
| 776 | + from google.gax.errors import GaxError | ||
| 777 | + | ||
| 776 | 778 | client = _Client(self.PROJECT) | |
| 777 | 779 | gax_api = _GAXSubscriberAPI(_deadline_exceeded_gax_error=True) | |
| 778 | 780 | api = self._make_one(gax_api, client) | |
| 779 | 781 | ||
| 780 | - with self.assertRaises(Exception): | ||
| 782 | + with self.assertRaises(GaxError): | ||
| 781 | 783 | api.subscription_pull(self.SUB_PATH, return_immediately=True) | |
| 782 | 784 | ||
| 783 | 785 | def test_subscription_acknowledge_hit(self): | |
0 commit comments