class TestSolrDB(unittest.TestCase):
def setUp(self):
self.dbFactory = DBFactory()
self.dbManager = self.dbFactory.get_specific_db_manager_and_schema('SOLR','test_core')
def test_insertion(self):
print ('> Insertion test')
self.previous_count = len(self.dbManager.get_all_data())
# print ('previous count: %i' % (self.previous_count))
self.dbManager.insert_data({'title':'test title','field':'test-field'})
self.new_count = len(self.dbManager.get_all_data())
# print ('new count: %i' % (self.new_count) )
self.assertEqual(self.previous_count+1, self.new_count)
def test_all_deletion(self):
print ('> All deletion test')
self.dbManager.insert_data({'title':'test title','field':'test-field'})
self.dbManager.delete_all_data()
self.new_count = len(self.dbManager.get_all_data())
# print ('data count after deletion: %i' % (self.new_count) )
self.assertEqual(0, self.new_count)
def test_delete(self):
print ('> Select test')
self.dbManager.delete_all_data()
self.dbManager.insert_data({'title':'first test title','field':'test-field'})
self.dbManager.insert_data({'title':'second test title','field':'test-field'})
self.dbManager.insert_data({'title':'third test title','field':'test-field'})
self.previous_count = len(self.dbManager.get_all_data())
# print ('previous count: %i' % (self.previous_count))
self.assertEqual(3, self.previous_count)
self.dbManager.delete_data_by_conditions([['EQ','title','first test title']])
self.new_count = len(self.dbManager.get_all_data())
# print ('new count: %i' % (self.new_count) )
self.assertEqual(2, self.new_count)
self.dbManager.delete_data_by_conditions(
[
['OR',
[
['AND',[
['EQ','field','test-field'],
['EQ','title','second test title']
]
],
['AND',[
['EQ','field','test-field'],
['EQ','title','first test title']
]
]
]
]
])
self.new_count = len(self.dbManager.get_all_data())
# print ('new count: %i' % (self.new_count) )
self.assertEqual(1, self.new_count)
def test_select(self):
print ('> Select test')
self.dbManager.delete_all_data()
self.dbManager.insert_data({'title':'first test title','field':'test-field'})
self.dbManager.insert_data({'title':'second test title','field':'test-field'})
self.dbManager.insert_data({'title':'third test title','field':'test-field'})
self.previous_count = len(self.dbManager.get_all_data())
# print ('previous count: %i' % (self.previous_count))
self.assertEqual(3, self.previous_count)
self.new_count = len(self.dbManager.get_data_by_conditions_full([['EQ','title','first test title']], None, None))
# print ('new count: %i' % (self.new_count) )
self.assertEqual(1, self.new_count)
self.new_count = len(self.dbManager.get_data_by_conditions_full([['EQ','field','test-field']], None, None))
# print ('new count: %i' % (self.new_count) )
self.assertEqual(3, self.new_count)
self.new_count = len(self.dbManager.get_data_by_conditions_full([['NO','title','second test title']], None, None))
# print ('new count: %i' % (self.new_count) )
self.assertEqual(2, self.new_count)
self.new_count = len(self.dbManager.get_data_by_conditions_full([['EQ','field','test-field'], ['EQ','title','second test title']], None, None))
# print ('new count: %i' % (self.new_count) )
self.assertEqual(1, self.new_count)
self.new_count = len(self.dbManager.get_data_by_conditions_full([['EQ','field','test-field'], ['NO','title','second test title']], None, None))
# print ('new count: %i' % (self.new_count) )
self.assertEqual(2, self.new_count)
self.new_count = len(self.dbManager.get_data_by_conditions_full([['EQ','field','test-field'], ['EQ','title','second test title']], None, None))
# print ('new count: %i' % (self.new_count) )
self.assertEqual(1, self.new_count)
self.new_count = len(self.dbManager.get_data_by_conditions_full(
[
['AND',[
['EQ','field','test-field'],
['EQ','title','second test title']
]
]
], None, None))
# print ('new count: %i' % (self.new_count) )
self.assertEqual(1, self.new_count)
self.new_count = len(self.dbManager.get_data_by_conditions_full(
[
['OR',
[
['EQ','field','test-field'],
['EQ','title','second test title']
]
]
], None, None))
# print ('new count: %i' % (self.new_count) )
self.assertEqual(3, self.new_count)
self.new_count = len(self.dbManager.get_data_by_conditions_full(
[
['OR',
[
['AND',[
['EQ','field','test-field'],
['EQ','title','second test title']
]
],
['AND',[
['EQ','field','test-field'],
['EQ','title','first test title']
]
]
]
]
], None, None))
# print ('new count: %i' % (self.new_count) )
self.assertEqual(2, self.new_count)
self.new_count = len(self.dbManager.get_data_by_conditions_full(
[
['OR',
[
['NO',
[
['AND',[
['EQ','field','test-field'],
['EQ','title','second test title']
]
]
]
],
['AND',[
['EQ','field','test-field'],
['EQ','title','first test title']
]
]
]
]
], None, None))
# print ('new count: %i' % (self.new_count) )
self.assertEqual(1, self.new_count)
# Testing sorting rules...
sorted_results = self.dbManager.get_data_by_conditions_full([['EQ','field','test-field']], [['title','ASC']], None)
first_sorted_result = sorted_results[0]
self.assertEqual(3, len(sorted_results))
self.assertEqual('first test title', first_sorted_result['title'])
sorted_results = self.dbManager.get_data_by_conditions_full([['EQ','field','test-field']], [['title','DESC']], None)
first_sorted_result = sorted_results[0]
self.assertEqual(3, len(sorted_results))
self.assertEqual('third test title', first_sorted_result['title'])
class TestSolrDB(unittest.TestCase):
def setUp(self):
self.dbFactory = DBFactory()
self.dbManager = self.dbFactory.get_specific_db_manager_and_schema(
'SOLR', 'test_core')
def test_insertion(self):
print('> Insertion test')
self.previous_count = len(self.dbManager.get_all_data())
# print ('previous count: %i' % (self.previous_count))
self.dbManager.insert_data({
'title': 'test title',
'field': 'test-field'
})
self.new_count = len(self.dbManager.get_all_data())
# print ('new count: %i' % (self.new_count) )
self.assertEqual(self.previous_count + 1, self.new_count)
def test_all_deletion(self):
print('> All deletion test')
self.dbManager.insert_data({
'title': 'test title',
'field': 'test-field'
})
self.dbManager.delete_all_data()
self.new_count = len(self.dbManager.get_all_data())
# print ('data count after deletion: %i' % (self.new_count) )
self.assertEqual(0, self.new_count)
def test_delete(self):
print('> Select test')
self.dbManager.delete_all_data()
self.dbManager.insert_data({
'title': 'first test title',
'field': 'test-field'
})
self.dbManager.insert_data({
'title': 'second test title',
'field': 'test-field'
})
self.dbManager.insert_data({
'title': 'third test title',
'field': 'test-field'
})
self.previous_count = len(self.dbManager.get_all_data())
# print ('previous count: %i' % (self.previous_count))
self.assertEqual(3, self.previous_count)
self.dbManager.delete_data_by_conditions(
[['EQ', 'title', 'first test title']])
self.new_count = len(self.dbManager.get_all_data())
# print ('new count: %i' % (self.new_count) )
self.assertEqual(2, self.new_count)
self.dbManager.delete_data_by_conditions([[
'OR',
[[
'AND',
[['EQ', 'field', 'test-field'],
['EQ', 'title', 'second test title']]
],
[
'AND',
[['EQ', 'field', 'test-field'],
['EQ', 'title', 'first test title']]
]]
]])
self.new_count = len(self.dbManager.get_all_data())
# print ('new count: %i' % (self.new_count) )
self.assertEqual(1, self.new_count)
def test_select(self):
print('> Select test')
self.dbManager.delete_all_data()
self.dbManager.insert_data({
'title': 'first test title',
'field': 'test-field'
})
self.dbManager.insert_data({
'title': 'second test title',
'field': 'test-field'
})
self.dbManager.insert_data({
'title': 'third test title',
'field': 'test-field'
})
self.previous_count = len(self.dbManager.get_all_data())
# print ('previous count: %i' % (self.previous_count))
self.assertEqual(3, self.previous_count)
self.new_count = len(
self.dbManager.get_data_by_conditions_full(
[['EQ', 'title', 'first test title']], None, None))
# print ('new count: %i' % (self.new_count) )
self.assertEqual(1, self.new_count)
self.new_count = len(
self.dbManager.get_data_by_conditions_full(
[['EQ', 'field', 'test-field']], None, None))
# print ('new count: %i' % (self.new_count) )
self.assertEqual(3, self.new_count)
self.new_count = len(
self.dbManager.get_data_by_conditions_full(
[['NO', 'title', 'second test title']], None, None))
# print ('new count: %i' % (self.new_count) )
self.assertEqual(2, self.new_count)
self.new_count = len(
self.dbManager.get_data_by_conditions_full(
[['EQ', 'field', 'test-field'],
['EQ', 'title', 'second test title']], None, None))
# print ('new count: %i' % (self.new_count) )
self.assertEqual(1, self.new_count)
self.new_count = len(
self.dbManager.get_data_by_conditions_full(
[['EQ', 'field', 'test-field'],
['NO', 'title', 'second test title']], None, None))
# print ('new count: %i' % (self.new_count) )
self.assertEqual(2, self.new_count)
self.new_count = len(
self.dbManager.get_data_by_conditions_full(
[['EQ', 'field', 'test-field'],
['EQ', 'title', 'second test title']], None, None))
# print ('new count: %i' % (self.new_count) )
self.assertEqual(1, self.new_count)
self.new_count = len(
self.dbManager.get_data_by_conditions_full([[
'AND',
[['EQ', 'field', 'test-field'],
['EQ', 'title', 'second test title']]
]], None, None))
# print ('new count: %i' % (self.new_count) )
self.assertEqual(1, self.new_count)
self.new_count = len(
self.dbManager.get_data_by_conditions_full([[
'OR',
[['EQ', 'field', 'test-field'],
['EQ', 'title', 'second test title']]
]], None, None))
# print ('new count: %i' % (self.new_count) )
self.assertEqual(3, self.new_count)
self.new_count = len(
self.dbManager.get_data_by_conditions_full([[
'OR',
[[
'AND',
[['EQ', 'field', 'test-field'],
['EQ', 'title', 'second test title']]
],
[
'AND',
[['EQ', 'field', 'test-field'],
['EQ', 'title', 'first test title']]
]]
]], None, None))
# print ('new count: %i' % (self.new_count) )
self.assertEqual(2, self.new_count)
self.new_count = len(
self.dbManager.get_data_by_conditions_full([[
'OR',
[[
'NO',
[[
'AND',
[['EQ', 'field', 'test-field'],
['EQ', 'title', 'second test title']]
]]
],
[
'AND',
[['EQ', 'field', 'test-field'],
['EQ', 'title', 'first test title']]
]]
]], None, None))
# print ('new count: %i' % (self.new_count) )
self.assertEqual(1, self.new_count)
# Testing sorting rules...
sorted_results = self.dbManager.get_data_by_conditions_full(
[['EQ', 'field', 'test-field']], [['title', 'ASC']], None)
first_sorted_result = sorted_results[0]
self.assertEqual(3, len(sorted_results))
self.assertEqual('first test title', first_sorted_result['title'])
sorted_results = self.dbManager.get_data_by_conditions_full(
[['EQ', 'field', 'test-field']], [['title', 'DESC']], None)
first_sorted_result = sorted_results[0]
self.assertEqual(3, len(sorted_results))
self.assertEqual('third test title', first_sorted_result['title'])
