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'])