Skip to content

Commit

Permalink
Merge pull request #1598 from tseaver/logging-sink_reload
Browse files Browse the repository at this point in the history
Add 'Sink.reload' API wrapper.
  • Loading branch information
tseaver committed Mar 11, 2016
2 parents 2c94675 + 629418b commit 427f96d
Show file tree
Hide file tree
Showing 2 changed files with 62 additions and 2 deletions.
15 changes: 15 additions & 0 deletions gcloud/logging/sink.py
Original file line number Diff line number Diff line change
Expand Up @@ -111,3 +111,18 @@ def exists(self, client=None):
return False
else:
return True

def reload(self, client=None):
"""API call: sync local sink configuration via a GET request
See
https://cloud.google.com/logging/docs/api/ref_v2beta1/rest/v2beta1/projects.sinks/get
:type client: :class:`gcloud.logging.client.Client` or ``NoneType``
:param client: the client to use. If not passed, falls back to the
``client`` stored on the current sink.
"""
client = self._require_client(client)
data = client.connection.api_request(method='GET', path=self.path)
self.filter_ = data['filter']
self.destination = data['destination']
49 changes: 47 additions & 2 deletions gcloud/logging/test_sink.py
Original file line number Diff line number Diff line change
Expand Up @@ -68,7 +68,7 @@ def test_create_w_alternate_client(self):
'filter': self.FILTER,
'destination': self.DESTINATION_URI,
}
conn1 = _Connection({'name': FULL})
conn1 = _Connection()
client1 = _Client(project=self.PROJECT, connection=conn1)
conn2 = _Connection({'name': FULL})
client2 = _Client(project=self.PROJECT, connection=conn2)
Expand Down Expand Up @@ -96,7 +96,7 @@ def test_exists_miss_w_bound_client(self):

def test_exists_hit_w_alternate_client(self):
FULL = 'projects/%s/sinks/%s' % (self.PROJECT, self.SINK_NAME)
conn1 = _Connection({'name': FULL})
conn1 = _Connection()
CLIENT1 = _Client(project=self.PROJECT, connection=conn1)
conn2 = _Connection({'name': FULL})
CLIENT2 = _Client(project=self.PROJECT, connection=conn2)
Expand All @@ -109,6 +109,51 @@ def test_exists_hit_w_alternate_client(self):
self.assertEqual(req['method'], 'GET')
self.assertEqual(req['path'], '/%s' % FULL)

def test_reload_w_bound_client(self):
FULL = 'projects/%s/sinks/%s' % (self.PROJECT, self.SINK_NAME)
NEW_FILTER = 'logName:syslog AND severity>=INFO'
NEW_DESTINATION_URI = 'faux.googleapis.com/other'
RESOURCE = {
'name': self.SINK_NAME,
'filter': NEW_FILTER,
'destination': NEW_DESTINATION_URI,
}
conn = _Connection(RESOURCE)
CLIENT = _Client(project=self.PROJECT, connection=conn)
sink = self._makeOne(self.SINK_NAME, self.FILTER, self.DESTINATION_URI,
client=CLIENT)
sink.reload()
self.assertEqual(sink.filter_, NEW_FILTER)
self.assertEqual(sink.destination, NEW_DESTINATION_URI)
self.assertEqual(len(conn._requested), 1)
req = conn._requested[0]
self.assertEqual(req['method'], 'GET')
self.assertEqual(req['path'], '/%s' % FULL)

def test_reload_w_alternate_client(self):
FULL = 'projects/%s/sinks/%s' % (self.PROJECT, self.SINK_NAME)
NEW_FILTER = 'logName:syslog AND severity>=INFO'
NEW_DESTINATION_URI = 'faux.googleapis.com/other'
RESOURCE = {
'name': self.SINK_NAME,
'filter': NEW_FILTER,
'destination': NEW_DESTINATION_URI,
}
conn1 = _Connection()
CLIENT1 = _Client(project=self.PROJECT, connection=conn1)
conn2 = _Connection(RESOURCE)
CLIENT2 = _Client(project=self.PROJECT, connection=conn2)
sink = self._makeOne(self.SINK_NAME, self.FILTER, self.DESTINATION_URI,
client=CLIENT1)
sink.reload(client=CLIENT2)
self.assertEqual(sink.filter_, NEW_FILTER)
self.assertEqual(sink.destination, NEW_DESTINATION_URI)
self.assertEqual(len(conn1._requested), 0)
self.assertEqual(len(conn2._requested), 1)
req = conn2._requested[0]
self.assertEqual(req['method'], 'GET')
self.assertEqual(req['path'], '/%s' % FULL)


class _Connection(object):

Expand Down

0 comments on commit 427f96d

Please sign in to comment.