def test_db(self):
#try:
self.wspace.create_tables()
from product import Product, Listing
from part import Part
from bom import BOM
tables = []
cur = self.wspace.memory.cursor()
for row in cur.execute("SELECT name FROM sqlite_master WHERE type='table' ORDER BY name"):
tables.append(row[0])
cur.close()
assert 'products' in tables
assert 'parts' in tables
assert 'part_attributes' in tables
assert 'listings' in tables
assert 'projects' in tables
assert 'pricebreaks' in tables
assert 'preferred_listings' in tables
self.test_BOM = BOM.new_project('dbtests', 'Databse Unit tests', '', self.wspace.memory)
self.wspace.projects = self.wspace.list_projects()
assert len(self.wspace.projects) == 1
assert 'dbtests' in self.wspace.projects
self.test_part = Part('C1', self.test_BOM, '1uF', 'C-USC0603', 'C0603', 'CAPACITOR, American symbol', self.test_product, self.part_attribs)
self.test_product.insert(self.wspace.memory)
self.test_listing_ct.insert(self.wspace.memory)
self.test_listing_tr.insert(self.wspace.memory)
self.test_listing_dr.insert(self.wspace.memory)
preferred_listing = self.test_product.get_preferred_listing(self.test_BOM, self.wspace.memory)
assert preferred_listing is None
self.test_product.set_preferred_listing(self.test_BOM, self.test_listing_ct, self.wspace.memory)
preferred_listing = self.test_product.get_preferred_listing(self.test_BOM, self.wspace.memory)
assert preferred_listing is not None
assert preferred_listing.key() == self.test_listing_ct.key()
self.test_product.set_preferred_listing(self.test_BOM, self.test_listing_dr, self.wspace.memory)
preferred_listing = self.test_product.get_preferred_listing(self.test_BOM, self.wspace.memory)
assert preferred_listing is not None
assert preferred_listing.key() == self.test_listing_dr.key()
self.test_part.insert(self.wspace.memory)
# Product.select_by_pn fetches listings for the product, and fetch_listings fetches the price dicts
ret_products = Product.select_by_pn(self.test_product.manufacturer_pn, self.wspace.memory)
assert self.test_product.is_in_db(self.wspace.memory)
# Should only return one result:
assert len(ret_products) == 1
# Product.equals() calls Listing.equals() as part of the check
assert len (Listing.select_by_vendor_pn(self.test_listing_ct.vendor_pn, self.wspace.memory)) == 1
assert len (Listing.select_by_vendor_pn(self.test_listing_tr.vendor_pn, self.wspace.memory)) == 1
assert len (Listing.select_by_vendor_pn(self.test_listing_dr.vendor_pn, self.wspace.memory)) == 1
assert len (Listing.select_by_manufacturer_pn(self.test_product.manufacturer_pn, self.wspace.memory)) == 3
assert self.test_product.equals(ret_products[0])
assert self.test_part.has_attribute('TOL', self.wspace.memory)
assert self.test_part.has_attribute('VOLT', self.wspace.memory)
assert self.test_part.has_attribute('TC', self.wspace.memory)
ret_parts = Part.select_by_name(self.test_part.name, self.wspace.memory, self.test_BOM)
try:
assert len(ret_parts) == 1
except AssertionError:
print 'Assertion failed: assert len(ret_parts) == 1'
print 'len(ret_parts): ', len(ret_parts)
raise AssertionError
assert self.test_part.equals(ret_parts[0])
assert self.test_part.is_in_db(self.wspace.memory)
ret_parts = self.test_BOM.select_parts_by_name(self.test_part.name, self.wspace.memory)
assert len(ret_parts) == 1
assert self.test_part.equals(ret_parts[0])
self.test_part.delete(self.wspace.memory)
assert len(self.test_BOM.select_parts_by_name(self.test_part.name, self.wspace.memory)) == 0
assert self.test_part.is_in_db(self.wspace.memory) == False
assert len (Listing.select_by_manufacturer_pn(self.test_product.manufacturer_pn, self.wspace.memory)) == 3
self.test_listing_ct.delete(self.wspace.memory)
assert len (Listing.select_by_manufacturer_pn(self.test_product.manufacturer_pn, self.wspace.memory)) == 2
assert len (Listing.select_by_vendor_pn(self.test_listing_ct.vendor_pn, self.wspace.memory)) == 0
self.test_listing_tr.delete(self.wspace.memory)
assert len (Listing.select_by_manufacturer_pn(self.test_product.manufacturer_pn, self.wspace.memory)) == 1
assert len (Listing.select_by_vendor_pn(self.test_listing_tr.vendor_pn, self.wspace.memory)) == 0
self.test_listing_dr.delete(self.wspace.memory)
assert len (Listing.select_by_manufacturer_pn(self.test_product.manufacturer_pn, self.wspace.memory)) == 0
assert len (Listing.select_by_vendor_pn(self.test_listing_dr.vendor_pn, self.wspace.memory)) == 0
assert len(Product.select_by_pn(self.test_product.manufacturer_pn, self.wspace.memory)) == 1
self.test_product.delete(self.wspace.memory)
assert len(Product.select_by_pn(self.test_product.manufacturer_pn, self.wspace.memory)) == 0
assert self.test_product.is_in_db(self.wspace.memory) == False
self.test_BOM.delete(self.wspace.memory)
self.wspace.projects = self.wspace.list_projects()
assert len(self.wspace.projects) == 0
assert 'dbtests' not in self.wspace.projects
tables = []
cur = self.wspace.memory.cursor()
cur.execute("SELECT name FROM sqlite_master WHERE type='table' ORDER BY name")
for row in cur.fetchall():
tables.append(row[0])
cur.close()
assert 'products' in tables
assert 'parts' in tables
assert 'part_attributes' in tables
assert 'listings' in tables
assert 'projects' in tables
assert 'pricebreaks' in tables
assert 'preferred_listings' in tables
assert 'dbtests' not in self.wspace.projects
