test: use updated object when calling update()

Author: einarf

extras/live_run.py

@@ -293,7 +293,7 @@ def test_incidents():
  component_status=enums.COMPONENT_STATUS_MAJOR_OUTAGE,
  )
  pprint(issue.attrs)
- issue.update()
+ issue = issue.update()
  issue.delete()
 
 

tests/test_components.py

@@ -76,7 +76,7 @@ def test_tags(self):
 
  comp.add_tag('Tag 1')
  comp.add_tag('Tag 2')
- comp = comp.update()
+ comp = comp = comp.update()
  self.assertTrue(comp.has_tag('Tag 1'))
  self.assertTrue(comp.has_tag('Tag 2'))
  self.assertTrue(comp.has_tag(slug='tag-1'))
@@ -88,7 +88,7 @@ def test_tags(self):
  self.assertEqual(len(comp.tags), 3)
  comp.del_tag(slug="tag-1")
  self.assertEqual(len(comp.tags), 2)
- comp.update()
+ comp = comp.update()
  self.assertFalse(comp.has_tag(slug="tag-1"))
  self.assertFalse(comp.has_tag(name="tag 1"))
  self.assertTrue(comp.has_tag(slug="tag-2"))

tests/test_incident_updates.py

@@ -64,7 +64,7 @@ def test_create(self):
  entry = updates[-1]
  entry.status = enums.INCIDENT_INVESTIGATING
  entry.message = "Lookin into it.."
- entry.update()
+ entry = entry.update()
 
  # Manually re-fetch
  updated_entry = self.client.incident_updates.get(entry.incident_id, entry.id)