def test_getitem_returns_collection(self, mock_collection, sqlite):
sqlite.connect.return_value = sqlite
mock_collection.return_value = mock_collection
conn = nosqlite.Connection()
assert 'foo' not in conn._collections
assert conn['foo'] == mock_collection
def test_if_deleted():
conn = nosqlite.Connection('dump.nosqlite')
dump_nosqlite = conn['dump_nosqlite']
for i in dump_nosqlite.find({'deleted': 0}):
status = requests.get("https://twitter.com/" + i['username'] +
"/status/" + i['tweetId']).status_code
if status == 404:
alert_deleted_tweet(i)
i['deleted'] = 1
dump_nosqlite.insert(i)
conn.close()
def test_getattr_returns_collection(self, sqlite):
conn = nosqlite.Connection()
foo = conn.__getattr__('foo')
assert foo not in conn.__dict__.values()
assert isinstance(foo, nosqlite.Collection)
def test_getattr_returns_attribute(self, sqlite):
conn = nosqlite.Connection()
assert conn.__getattr__('db') in conn.__dict__.values()
def test_drop_collection(self, sqlite):
conn = nosqlite.Connection()
conn.drop_collection('foo')
conn.db.execute.assert_called_with('drop table if exists foo')
def test_getitem_returns_cached_collection(self, sqlite):
conn = nosqlite.Connection()
conn._collections['foo'] = 'bar'
assert conn['foo'] == 'bar'
def test_context_manager_closes_connection(self, sqlite):
with nosqlite.Connection() as conn:
pass
assert conn.db.close.called
def test_connect(self):
conn = nosqlite.Connection(':memory:')
assert conn.db.isolation_level is None
def alert_deleted_tweet(deleted):
print("Deleted Tweet: \n\n" + str(deleted) + "\n\n")
conn2 = nosqlite.Connection('deleted_tweets.nosqlite')
deleted_tweets = conn2['deleted_tweets']
deleted_tweets.insert(deleted)
conn2.close()
def save_tweet_to_db(tweet):
conn = nosqlite.Connection('dump.nosqlite')
dump_nosqlite = conn['dump_nosqlite']
if len(dump_nosqlite.find({'tweetId': tweet['tweetId']})) < 1:
dump_nosqlite.insert(tweet)
conn.close()
