def test_when_query_and_insert_columns_are_different_insertion_and_query_uses_diferent_cols(self):
db = MongoDB()
db.query("labsk_test")
db.insert("to_delete")
db.saveThread({'id':'1'})
self.assertIsNone(db.find_one_by('id', '1'))
db.query("to_delete")
self.assertIsNotNone(db.find_one_by('id', '1'))
db.drop("to_delete")
class TestStatsWithMondoDB(unittest.TestCase):
def setUp(self):
self.db = MongoDB(col = 'stats_test')
self.stats = Statistics(self.db)
mock = MockDatetime()
self.expected_date = mock.now()
self.stats._datetime = mock
def test_save_stat(self):
json_doc = {'url':"url", 'datetime': "2014-01-01 14:10:00" }
self.stats.register_access_now("url")
result = self.db.find_one_by('url', "url")
self.assertEqual(result['url'], "url")
self.assertEqual(result['datetime'], str(self.expected_date))
self.db.drop('stats_test')
def test_all_visits(self):
self.stats.register_access_now("url")
visits = self.stats.all_visits()
self.assertEqual(len(visits), 1)
class TestMongoDB(unittest.TestCase):
def setUp(self):
self.db = MongoDB()
self.testcol = self.db.collection("labsk_test")
def test_merge_with_empty_collection_all_docs_merged(self):
db = self.db
db.query("empty_new_col")
db.insert("labsk_test")
merge_result = db.merge_insert_wih_query('link')
self.assertEqual(60, db.len("empty_new_col"))
self.assertEqual(60, db.len("labsk_test"))
db.drop("empty_new_col")
def test_merge_returns_new_and_update_threads(self):
db = self.db
db.query("empty_new_col")
db.insert("labsk_test")
merge_result = db.merge_insert_wih_query('link')
self.assertEqual(60, merge_result.new_threads)
self.assertEqual(0, merge_result.updated_threads)
db.drop("empty_new_col")
def test_merge_drops_insert_column_and_avoid_its_use(self):
db = self.db
db.copy("labsk_test", "query_delete")
db.copy("labsk_test", "insert_delete")
db.query("query_delete")
db.insert("insert_delete")
merge_result = db.merge('link')
self.assertIsNone(db.insert_col)
self.assertEqual(0, db.len("insert_delete"))
self.assertEqual(60, db.len("query_delete"))
db.drop("query_delete")
def test_find_all_by(self):
results = self.db.find_one_by('link', 'http://labsk.net/index.php?topic=119300.0')
self.assertEqual(results['link'], 'http://labsk.net/index.php?topic=119300.0')
self.assertEqual(results['title'], u'Fief - Feudo Ediciones MasQueOca editar\xe1 Fief en Espa\xf1ol')
def test_find_all_by_not_found(self):
results = self.db.find_one_by('link', 'no no')
self.assertIsNone(results)
def test_by_default_query_and_insert_columns_are_the_column_in_constructor(self):
db = MongoDB(col = "labsk_test")
self.assertIs(db.col, db.query_col)
self.assertIs(db.col, db.insert_col)
def test_when_query_and_insert_columns_are_different_insertion_and_query_uses_diferent_cols(self):
db = MongoDB()
db.query("labsk_test")
db.insert("to_delete")
db.saveThread({'id':'1'})
self.assertIsNone(db.find_one_by('id', '1'))
db.query("to_delete")
self.assertIsNotNone(db.find_one_by('id', '1'))
db.drop("to_delete")
def test_database_contains(self):
self.db = MongoDB(col = "labsk_merge")
result = self.db.find_one_by('link', "http://labsk.net/index.php?topic=129533.0")
self.assertEqual("http://labsk.net/index.php?topic=129533.0", result['link'])
