Skip to content

Commit ae88612

Browse files
committed
Merge pull request #1678 from tseaver/logging-system_tests-sink_reload
Add system test for 'Sink.reload'.
2 parents 9a7df6f + a7749e9 commit ae88612

File tree

1 file changed

+17
-3
lines changed

1 file changed

+17
-3
lines changed

system_tests/logging_.py

Lines changed: 17 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -154,7 +154,7 @@ def test_create_sink_storage_bucket(self):
154154
self.to_delete.append(sink)
155155
self.assertTrue(sink.exists())
156156

157-
def test_create_sink_bigquery_dataset(self):
157+
def _init_bigquery_dataset(self):
158158
from gcloud import bigquery
159159
from gcloud.bigquery.dataset import AccessGrant
160160
DATASET_URI = 'bigquery.googleapis.com/projects/%s/datasets/%s' % (
@@ -172,10 +172,24 @@ def test_create_sink_bigquery_dataset(self):
172172
'WRITER', 'groupByEmail', '[email protected]'))
173173
dataset.access_grants = grants
174174
dataset.update()
175+
return DATASET_URI
175176

176-
sink = Config.CLIENT.sink(
177-
DEFAULT_SINK_NAME, DEFAULT_FILTER, DATASET_URI)
177+
def test_create_sink_bigquery_dataset(self):
178+
uri = self._init_bigquery_dataset()
179+
sink = Config.CLIENT.sink(DEFAULT_SINK_NAME, DEFAULT_FILTER, uri)
178180
self.assertFalse(sink.exists())
179181
sink.create()
180182
self.to_delete.append(sink)
181183
self.assertTrue(sink.exists())
184+
185+
def test_reload_sink(self):
186+
uri = self._init_bigquery_dataset()
187+
sink = Config.CLIENT.sink(DEFAULT_SINK_NAME, DEFAULT_FILTER, uri)
188+
self.assertFalse(sink.exists())
189+
sink.create()
190+
self.to_delete.append(sink)
191+
sink.filter_ = 'BOGUS FILTER'
192+
sink.destination = 'BOGUS DESTINATION'
193+
sink.reload()
194+
self.assertEqual(sink.filter_, DEFAULT_FILTER)
195+
self.assertEqual(sink.destination, uri)

0 commit comments

Comments
 (0)