def test_get_recomputation(): log = logging.getLogger() mock_collection = mock.MagicMock() expected_recomputation = json.loads(FOO_RECOMPUTATION_STR) mock_collection.find_one.return_value = expected_recomputation results = recompute.get_recomputation(mock_collection, "551bdd701c8a97e78635a911", log) # Assert with a valid Object Id mock_collection.find_one.assert_called_with({'_id': ObjectId('551bdd701c8a97e78635a911')}) assert results == expected_recomputation # 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)
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)
def test_get_recomputation(): log = logging.getLogger() mock_collection = mock.MagicMock() expected_recomputation = json.loads(FOO_RECOMPUTATION_STR) mock_collection.find_one.return_value = expected_recomputation results = recompute.get_recomputation(mock_collection, "551bdd701c8a97e78635a911", log) # Assert with a valid Object Id mock_collection.find_one.assert_called_with({"id": "551bdd701c8a97e78635a911"}) assert results == expected_recomputation