def test_update_status(): log = logging.getLogger() mock_collection = mock.MagicMock() timestamp = datetime.now() recompute.update_status(mock_collection, "551bdd701c8a97e78635a911", "FOO", timestamp, log) query_id = {"id": "551bdd701c8a97e78635a911"} query_update = {"$set": {"status": "FOO", "timestamp": timestamp.strftime("%Y-%m-%d %H:%M:%S")}} mock_collection.update_one.assert_called_with(query_id, query_update)
def test_update_status(): log = logging.getLogger() mock_collection = mock.MagicMock() timestamp = datetime.now() recompute.update_status(mock_collection, "551bdd701c8a97e78635a911", "FOO", timestamp, log) query_id = {'_id': ObjectId('551bdd701c8a97e78635a911')} query_update = {'$set': {'s': 'FOO'}, '$push': {'history': {'status': 'FOO', 'ts': timestamp}}} # Assert with a valid Object Id mock_collection.update.assert_called_with(query_id, query_update) # Assert with invalid ObjectId and raised exception with pytest.raises(ValueError) as excinfo: recompute.get_recomputation(mock_collection, "33", log) assert 'Invalid Object Id used' in str(excinfo.value)