def test_simple_get(self):
''' test simple get
'''
db = 'tests/db/simple_get.sqlite'
uri = URI_TEMPLATE.format(db, 'product')
ds = dblite.Storage(Product, uri, autocommit=True)
ds.delete(_all=True)
ds.commit()
all_items = [
Product({
'name': 'product#%s' % i,
'price': i + 100
}) for i in range(10)
]
ds.put(all_items)
self.assertEqual(len(all_items), 10)
self.assertEqual(sum([1 for _ in ds.get()]), 10)
self.assertIsNotNone(ds.get_one({'name': 'product#2'}))
self.assertIsNotNone(ds.get_one({'price': 102}))
self.assertIsNotNone(
ds.get_one({'$and': {
'name': 'product#2',
'price': 102
}}))
def test_regexp_syntax(self):
''' test_regexp_syntax
'''
db = 'tests/db/regexp-request.sqlite'
if os.path.isfile(db):
os.remove(db)
uri = URI_TEMPLATE.format(db, 'product')
ds = dblite.Storage(Product, uri)
products = (Product({'name': 'Laptop'}), Product({'name': 'Desktop'}),
Product({'name': 'Nettop'}))
ds.put(products)
self.assertEqual([p for p in ds.get({'name': 'r/[Lap|Desk]top/'})],
[{
'_id': 1,
'catalog_url': None,
'name': u'Laptop',
'price': None,
'description': None
}, {
'_id': 2,
'catalog_url': None,
'name': u'Desktop',
'price': None,
'description': None
}])
ds.close()
def test_wrong_get(self):
''' test wrong get
'''
db = 'tests/db/wrong_get.sqlite'
uri = URI_TEMPLATE.format(db, 'product')
ds = dblite.Storage(Product, uri)
self.assertEqual([1 for _ in ds.get('name')], [])
def test_conditional_delete(self):
''' test conditional delete
'''
db = 'tests/db/cond-delete.sqlite'
if os.path.isfile(db):
os.remove(db)
uri = URI_TEMPLATE.format(db, 'product')
ds = dblite.Storage(Product, uri, autocommit=True)
# delete by name
ds.put(Product({'name': 'product_name'}))
self.assertEqual(len(ds), 1)
ds.delete({'name': 'product_name'})
self.assertEqual(len(ds), 0)
ds.commit()
# delete by _id
ds.put(Product({'name': 'product_name'}))
self.assertEqual(len(ds), 1)
for p in ds.get():
ds.delete({'_id': p['_id']})
self.assertEqual(len(ds), 0)
ds.commit()
# delete by Item
ds.put(Product({'name': 'product_name'}))
self.assertEqual(len(ds), 1)
for p in ds.get():
ds.delete(p)
self.assertEqual(len(ds), 0)
ds.commit()
ds.close()
def test_create_simple_db(self):
''' test_create_simple_db
'''
db = 'tests/db/simple_db.sqlite'
if os.path.isfile(db):
os.remove(db)
uri = URI_TEMPLATE.format(db, 'product')
ds = dblite.Storage(Product, uri)
self.assertEqual(type(ds), dblite.Storage)
ds.close()
def test_detect_db_fieldnames(self):
''' test detect db fieldnames
'''
db = 'tests/db/simple_db.sqlite'
uri = URI_TEMPLATE.format(db, 'product')
ds = dblite.Storage(Product, uri)
self.assertEqual(
set(ds.fieldnames),
set(['_id', 'name', 'price', 'catalog_url', 'description']))
ds.close()
def test_wrong_delete(self):
''' test wrong delete
'''
db = 'tests/db/wrong-delete.sqlite'
uri = URI_TEMPLATE.format(db, 'product')
ds = dblite.Storage(Product, uri)
self.assertRaises(
RuntimeError,
ds.delete,
)
def test_autocommit_as_counter(self):
''' test autocommit
'''
db = 'tests/db/db-autocommit.sqlite'
if os.path.isfile(db):
os.remove(db)
uri = URI_TEMPLATE.format(db, 'product')
ds = dblite.Storage(Product, uri, autocommit=50)
for i in xrange(105):
ds.put(Product({'name': 'product#%d' % i}))
self.assertEqual(len(ds), 105)
def test_non_exists_item(self):
''' test_non_exists_item
'''
db = 'tests/db/limited-get.sqlite'
if os.path.isfile(db):
os.remove(db)
uri = URI_TEMPLATE.format(db, 'product')
ds = dblite.Storage(Product, uri, autocommit=True)
ds.put(Product({'name': 'product', 'price': 100}))
self.assertIsNone(ds.get_one({'name': 'Product'}))
ds.close()
def test_put_duplicate_item(self):
''' test_put_duplicate_item
'''
db = 'tests/db/db-put-and-get.sqlite'
if os.path.isfile(db):
os.remove(db)
uri = URI_TEMPLATE.format(db, 'product')
ds = dblite.Storage(Product, uri)
products = [
Product({
'catalog_url': 'http://catalog/1',
}) for _ in range(2)
]
self.assertRaises(dblite.DuplicateItem, ds.put, products)
def test_wrong_put(self):
''' test_wrong_put
'''
db = 'tests/db/wrong-put.sqlite'
if os.path.isfile(db):
os.remove(db)
uri = URI_TEMPLATE.format(db, 'product')
ds = dblite.Storage(Product, uri)
self.assertRaises(RuntimeError, ds.put, ['product#0', 1000])
self.assertRaises(RuntimeError, ds.put, {
'name': 'product#0',
'price': 1000
})
ds.close()
def test_update_item(self):
''' test_update_item
'''
db = 'tests/db/update_item.sqlite'
if os.path.isfile(db):
os.remove(db)
uri = URI_TEMPLATE.format(db, 'product')
ds = dblite.Storage(Product, uri, autocommit=True)
product = Product({'name': 'old product'})
ds.put(product)
self.assertEqual(len([p for p in ds.get({'name': 'old product'})]), 1)
for p in ds.get():
p['name'] = 'new product'
ds.put(p)
self.assertEqual(len([p for p in ds.get({'name': 'old product'})]), 0)
self.assertEqual(len([p for p in ds.get({'name': 'new product'})]), 1)
ds.close()
def test_sql(self):
''' test_sql
'''
db = 'tests/db/sql.sqlite'
uri = URI_TEMPLATE.format(db, 'product')
ds = dblite.Storage(Product, uri)
self.assertIsNone(
ds.sql(
'INSERT INTO product (name, price, catalog_url) VALUES (?, ?, ?);',
('Laptop', 100, 'http://catalog/1')))
self.assertRaises(
dblite.DuplicateItem, ds.sql,
'INSERT INTO product (name, price, catalog_url) VALUES (?, ?, ?);',
('Laptop', 100, 'http://catalog/1'))
self.assertRaises(dblite.SQLError, ds.sql, 'SELECT rowid, * FROM p;')
self.assertEqual([p for p in ds.sql('SELECT name FROM product;')], [
Product({'name': 'Laptop'}),
])
self.assertIsNone(ds.sql('DELETE FROM product WHERE name = "Laptop"'))
self.assertEqual([p for p in ds.sql('SELECT name FROM product;')], [])
def test_limited_get(self):
''' test_limited_get
'''
db = 'tests/db/limited-get.sqlite'
if os.path.isfile(db):
os.remove(db)
uri = URI_TEMPLATE.format(db, 'product')
ds = dblite.Storage(Product, uri, autocommit=True)
all_items = [
Product({
'name': 'product#%s' % i,
'price': i + 100
}) for i in range(10)
]
ds.put(all_items)
self.assertEqual(len(all_items), 10)
self.assertEqual(sum([1 for _ in ds.get(limit=5)]), 5)
ds.close()
def test_like_syntax(self):
''' test_like_syntax
'''
db = 'tests/db/like-request.sqlite'
if os.path.isfile(db):
os.remove(db)
uri = URI_TEMPLATE.format(db, 'product')
ds = dblite.Storage(Product, uri)
products = (Product({'name': 'Laptop'}), Product({'name': 'Desktop'}),
Product({'name': 'Nettop'}))
ds.put(products)
self.assertEqual(ds.get_one({'name': '/ptop/'}), None)
self.assertEqual(
ds.get_one({'name': '/%aptop%/'}), {
'_id': 1,
'catalog_url': None,
'name': u'Laptop',
'price': None,
'description': None
})
ds.close()
def test_put_many(self):
''' test put many dicts to database
'''
db = 'tests/db/db-put-many.sqlite'
if os.path.isfile(db):
os.remove(db)
uri = URI_TEMPLATE.format(db, 'product')
ds = dblite.Storage(Product, uri)
products = [
Product({
'name': 'product#%d' % i,
'price': i,
}) for i in range(10)
]
ds.put(products)
ds.commit()
self.assertEqual(len(ds), 10)
for product in products:
self.assertEqual(
len([p for p in ds.get({'name': product['name']})]), 1)
ds.close()
def test_put_get_delete(self):
''' test put & get & delete dicts to/from database
'''
db = 'tests/db/db-put-and-get.sqlite'
if os.path.isfile(db):
os.remove(db)
uri = URI_TEMPLATE.format(db, 'product')
ds = dblite.Storage(Product, uri)
products = list()
for i in range(10):
p = Product({
'name': 'product#%d' % i,
'price': i,
'description': 'product description'
})
ds.put(p)
ds.commit()
self.assertEqual(len(ds), 10)
for i in range(10):
self.assertEqual(
ds.get_one({'name': 'product#%d' % i}), {
'_id': i + 1,
'name': 'product#%d' % i,
'catalog_url': None,
'price': i,
'description': 'product description'
})
ds.delete(_all=True)
ds.commit()
self.assertEqual(len(ds), 0)
ds.close()