Python with_db_ids Examples

Example #1

File: test_store.py Project: marklar/massiu

def test_put_repeated(repeated_tweets, tweets):
    drop()
    store.put(COLLECTION_NAME, repeated_tweets)
    with_ids = store.with_db_ids(tweets)
    assert list(store.get_all(COLLECTION_NAME)) == with_ids
    drop()

Example #2

File: test_store.py Project: marklar/massiu

def test_with_db_ids(tweets):
    expected_ids = ['1234', '2345']
    new_ids = [t['_id'] for t in store.with_db_ids(tweets)]
    assert new_ids == expected_ids