@@ -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