def setUp(self):
self.storage = Nors_LocalStorage_DAO('nors_test_db', 'mongodb://localhost:27017/')
self.storage.dropCollection('TestCollection')
self.test_data1 = {'field1': 100, 'field2': 'lucky', 'date': '2016-04-06 06:40:41.758893'}
self.test_data2 = {'field1': 200, 'field2': 'almost', 'date': '2016-04-06 06:42:41.758893'}
self.test_data3 = {'field1': 300, 'field2': 'sunnny', 'date': '2016-04-06 06:44:41.758893'}
self.test_data4 = {'field1': 400, 'field2': 'yards', 'date': '2016-04-06 06:46:41.758893'}
self.storage.insert('TestCollection', self.test_data1)
self.storage.insert('TestCollection', self.test_data2)
self.storage.insert('TestCollection', self.test_data3)
self.storage.insert('TestCollection', self.test_data4)
class Nors_LocalStorage_DAO_Test(unittest.TestCase):
def setUp(self):
self.storage = Nors_LocalStorage_DAO('nors_test_db', 'mongodb://localhost:27017/')
self.storage.dropCollection('TestCollection')
self.test_data1 = {'field1': 100, 'field2': 'lucky', 'date': '2016-04-06 06:40:41.758893'}
self.test_data2 = {'field1': 200, 'field2': 'almost', 'date': '2016-04-06 06:42:41.758893'}
self.test_data3 = {'field1': 300, 'field2': 'sunnny', 'date': '2016-04-06 06:44:41.758893'}
self.test_data4 = {'field1': 400, 'field2': 'yards', 'date': '2016-04-06 06:46:41.758893'}
self.storage.insert('TestCollection', self.test_data1)
self.storage.insert('TestCollection', self.test_data2)
self.storage.insert('TestCollection', self.test_data3)
self.storage.insert('TestCollection', self.test_data4)
def test_db_find_compare_with_string(self):
findstring = {'field1' : 200}
findresult = self.storage.find('TestCollection', findstring)
self.assertEqual(self.test_data2, findresult[0])
def test_db_find_failure(self):
findstring = {'field1' : 250}
findresult = self.storage.find('TestCollection', findstring)
self.assertEqual(len(findresult), 0 )
def test_db_find_one(self):
findstring = {'field1' : 400}
findresult = self.storage.find('TestCollection', findstring)
self.assertEqual(len(findresult), 1)
def test_db_find_two(self):
findstring = { 'field1': { '$gt': 250 } }
findresult = self.storage.find('TestCollection', findstring)
self.assertEqual(len(findresult), 2)
def test_db_find_two_limit(self):
findstring = { 'field1': { '$gt': 0 } }
findresult = self.storage.find('TestCollection', findstring, 2)
self.assertEqual(len(findresult), 2)
def test_db_find_first_n(self):
findresult = self.storage.get_first_n_itens('TestCollection', 3)
self.assertEqual(self.test_data1, findresult[0])
self.assertEqual(self.test_data2, findresult[1])
self.assertEqual(self.test_data3, findresult[2])
self.assertEqual(len(findresult), 3)
def test_db_id(self):
findresult = self.storage.get_first_n_itens('TestCollection', 1)
test_id = findresult[0]['_id']
# print test_id
findresult = self.storage.get_item_by_id('TestCollection', test_id)
# print findresult
self.assertEqual(self.test_data1, findresult[0])
def test_db_get_date_in_range_all(self):
findresult = self.storage.get_itens_from_date('TestCollection', '2016-04-06 06:39:41.758893', 2)
self.assertEqual(len(findresult), 2)
def test_db_get_date_in_range_few(self):
findresult = self.storage.get_itens_from_date('TestCollection', '2016-04-06 06:45:41.758893', 1)
self.assertEqual(len(findresult), 1)
def test_db_get_date_in_range_none(self):
findresult = self.storage.get_itens_from_date('TestCollection', '2016-04-06 06:47:41.758893', 1)
self.assertEqual(len(findresult), 0)
def test_db_update_existent(self):
find_string = {'field1' : 100, 'field2': 'new_value'}
findresult = self.storage.find('TestCollection', find_string)
self.assertEqual(len(findresult), 0)
query_string = {'field1': 100}
update_string = {'field2': 'new_value'}
update_result = self.storage.update('TestCollection', query_string, update_string)
find_string = {'field1' : 100, 'field2': 'new_value'}
findresult = self.storage.find('TestCollection', find_string)
self.assertEqual(len(findresult), 1)
self.assertEqual(update_result.matched_count, 1)
# self.assertEqual(update_result.modified_count, 1) # i dont know why this fail
def test_delete_by_id(self):
findresult = self.storage.get_first_n_itens('TestCollection', 1)
test_id = findresult[0]['_id']
# print test_id
findresult = self.storage.delete_by_id('TestCollection', test_id)
# print findresult
self.assertEqual(1, findresult)
findresult = self.storage.get_item_by_id('TestCollection', test_id)
print findresult
self.assertEqual(len(findresult), 0)
def tearDown(self):
self.storage.dropCollection('TestCollection')
self.storage.dropDB()
def tearDown(self):
storage = Nors_LocalStorage_DAO('testDB', 'mongodb://localhost:27017/')
storage.dropCollection('testDB')
storage.dropDB()
