def test_existing(self, mock_get_collection):
mock_update = mock_get_collection.return_value.update
fake_id = bson.ObjectId()
call = ScheduledCall('PT1M', 'pulp.tasks.dosomething', id=fake_id)
call.save()
expected = call.as_dict()
del expected['_id']
mock_update.assert_called_once_with({'_id': fake_id}, expected)
def test_new(self, mock_get_collection):
mock_insert = mock_get_collection.return_value.insert
call = ScheduledCall('PT1M', 'pulp.tasks.dosomething')
call.save()
expected = call.as_dict()
expected['_id'] = bson.ObjectId(expected['_id'])
mock_insert.assert_called_once_with(expected, safe=True)
self.assertFalse(call._new)
def test_returns_dict(self):
call = ScheduledCall('PT1M', 'pulp.tasks.dosomething')
self.assertTrue(isinstance(call.as_dict(), dict))
