class TestWeeklyChart(unittest.TestCase):
@vcr.use_cassette(FIXTURES_DIR + '/vcr_cassettes/weekly_chart.yaml')
def setUp(self):
self.bom = BOM(WEEKLY_CHART)
self.movies = self.bom.get_chart()
self.test_movie = self.movies.next() # get the first movie for testing
def test_weekly_chart(self):
"""
Tests if BOM holds appropriate Movie objects
"""
assert type(self.test_movie.rank) == str
assert type(self.test_movie.title) == str
assert type(self.test_movie.studio) == str
assert type(self.test_movie.gross) == str
assert "$" in self.test_movie.gross
@vcr.use_cassette(FIXTURES_DIR + '/vcr_cassettes/weekly_trend.yaml')
def test_weekly_trend(self):
"""
Tests for the weekend trend of a movie
"""
for movie in self.movies:
trend_data = movie.get_trend(WEEKLY_CHART)
assert len(trend_data) > 0
assert type(trend_data) == list
for data in trend_data:
assert type(data) == tuple
class TestDailyChart(unittest.TestCase):
@vcr.use_cassette(FIXTURES_DIR + '/vcr_cassettes/daily_chart.yaml')
def setUp(self):
self.bom = BOM()
self.movies = self.bom.get_chart()
self.test_movie = self.movies.next() # get the first movie for testing
def test_daily_chart(self):
"""
Tests if BOM holds appropriate Movie objects
"""
assert type(self.test_movie.rank) == str
assert type(self.test_movie.title) == str
assert type(self.test_movie.studio) == str
assert type(self.test_movie.gross) == str
assert "$" in self.test_movie.gross
temp_gross = self.test_movie.gross
self.test_movie.gross = '-'
assert type(self.test_movie.gross_val) == int
assert self.test_movie.gross_val == 0
self.test_movie.gross = temp_gross
@vcr.use_cassette(FIXTURES_DIR + '/vcr_cassettes/daily_trend.yaml')
def test_daily_trend(self):
"""
Tests for the daily trend of a movie
"""
for movie in self.movies:
trend_data = movie.get_trend()
assert len(trend_data) > 0
assert type(trend_data) == list
for data in trend_data:
assert type(data) == tuple
class TestBOM(unittest.TestCase):
@vcr.use_cassette(FIXTURES_DIR + '/vcr_cassettes/bom.yaml')
def setUp(self):
self.bom = BOM()
self.movies = self.bom.get_chart()
def test_bom(self):
assert type(self.bom.date) == str
class TestBOM(unittest.TestCase):
@vcr.use_cassette(FIXTURES_DIR + '/vcr_cassettes/bom.yaml')
def setUp(self):
self.bom = BOM()
self.movies = self.bom.get_chart()
def test_bom(self):
assert type(self.bom.date) == str
class TestMovie(unittest.TestCase):
#@vcr.use_cassette(FIXTURES_DIR + '/vcr_cassettes/movie.yaml')
def setUp(self):
self.bom = BOM()
self.movies = self.bom.get_chart()
self.test_movie = self.movies.next() # get the first movie for testing
def test_movie(self):
assert (urllib2.urlopen(self.test_movie.url))
assert type(self.test_movie.gross_val) == int
class TestMovie(unittest.TestCase):
# @vcr.use_cassette(FIXTURES_DIR + '/vcr_cassettes/movie.yaml')
def setUp(self):
self.bom = BOM()
self.movies = self.bom.get_chart()
self.test_movie = self.movies.next() # get the first movie for testing
def test_movie(self):
assert urllib2.urlopen(self.test_movie.url)
assert type(self.test_movie.gross_val) == int
def new_from_row(row, connection, known_project=None): ''' Given a part row from the DB, returns a Part object. ''' from bom import BOM #print 'new_from_row: row param: ', row if row[6] is None or row[6] == 'NULL' or row[6] == '': product = None #print 'new_from_row: setting no product' else: product = Product.select_by_pn(row[6], connection)[0] #print 'new_from_row: product results: ', product if row[1] is None or row[1] == 'NULL': project = None # TODO: Raise an exception here? This is a PK violation print 'row[1] is None/NULL!' else: if known_project is None: projects = BOM.read_from_db(row[1], connection) if len(projects) > 0: project = projects[0] else: project = known_project part = Part(row[0], project, row[2], row[3], row[4], row[5], product) part.fetch_attributes(connection) return part
def setUp(self):
self.bom = BOM()
self.movies = self.bom.get_chart()
self.test_movie = self.movies.next() # get the first movie for testing
def setUp(self):
self.bom = BOM(WEEKLY_CHART)
self.movies = self.bom.get_chart()
self.test_movie = self.movies.next() # get the first movie for testing
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
def test_csv(self):
try:
self.wspace.create_tables()
other_wspace = Workspace('DB Tests 2', os.path.join(os.getcwd(), 'testfixtures', 'dbtests2.sqlite'))
other_wspace.create_tables()
from product import Product, Listing
from part import Part
from bom import BOM
other_proj = BOM.new_project('other_proj', 'other', '', other_wspace.memory)
test_c5_prod = Product('TDK Corporation', 'C1005X5R1V104K', 'general_B11.pdf', 'CAP CER 0.1UF 35V 10% X5R 0402', '0402 (1005 Metric)')
test_c5_prod.scrape(other_wspace.memory) # Don't want to add this to the main test DB
test_c5 = Part('C5', other_proj, '0.1uF', 'C-USC0402', 'C0402', 'CAPACITOR, American symbol', test_c5_prod)
test1_csv = os.path.join(os.getcwd(), 'testfixtures', "test1.csv")
test2_csv = os.path.join(os.getcwd(), 'testfixtures', "test2.csv")
test3_csv = os.path.join(os.getcwd(), 'testfixtures', "test3.csv")
self.wspace.projects = self.wspace.list_projects()
assert len(self.wspace.projects) == 0
test1_bom = BOM.new_project('test1', 'Product column, no header', test1_csv, self.wspace.memory)
test2_bom = BOM.new_project('test2', 'No product column, no header', test2_csv, self.wspace.memory)
test3_bom = BOM.new_project('test3', 'Header row, no PN attribute', test3_csv, self.wspace.memory)
self.wspace.projects = self.wspace.list_projects()
assert len(self.wspace.projects) == 3
test1_bom.read_from_file(self.wspace.memory)
assert len(test1_bom.parts) == 6
test1_c5_query = test1_bom.select_parts_by_name('C5', self.wspace.memory)
assert len(test1_c5_query) == 1
test1_c5 = test1_c5_query[0]
assert test1_c5.equals(test_c5, True, True, False, True) == True
assert test1_c5.equals(test_c5, True, True, True, True) == False
assert test1_c5.product.equals(test_c5_prod)
test2_bom.read_from_file(self.wspace.memory)
assert len(test2_bom.parts) == 6
test2_c5_query = test2_bom.select_parts_by_name('C5', self.wspace.memory)
assert len(test2_c5_query) == 1
test2_c5 = test2_c5_query[0]
assert test2_c5.project is test2_bom
# Check: Attribs, Name, Proj, Prod
try:
assert test2_c5.equals(test_c5, False, True, False, True) == True
except AssertionError:
print 'Assertion failed: assert test2_c5.equals(test_c5, False, True, False, True) == True'
print 'Reference C5: ', test_c5
print 'Test 1 C5: ', test1_c5
print 'Test 2 C5: ', test2_c5
raise AssertionError
assert test2_c5.equals(test_c5, True, True, False, True) == True
assert test2_c5.equals(test_c5, True, True, True, True) == False
assert test2_c5.product.equals(test_c5_prod)
assert test1_bom.parts == test2_bom.parts
test3_bom.read_from_file(self.wspace.memory)
assert len(test3_bom.parts) == 382
test3_c5_query = test3_bom.select_parts_by_name('C5', self.wspace.memory)
assert len(test3_c5_query) == 1
test3_c5 = test3_c5_query[0]
#print 'test3_c5 1st check: \n', test3_c5.show()
assert test3_c5.equals(test_c5, True, True, False, True) == True
assert test3_c5.equals(test_c5, True, True, True, True) == False
assert test3_c5.product.equals(test_c5_prod)
test3_c11_query = test3_bom.select_parts_by_name('C11', self.wspace.memory)
assert len(test3_c11_query) == 1
test3_c11 = test3_c11_query[0]
assert test3_c5.project is test3_bom
assert test3_c11.product.equals(test_c5_prod)
c5_prod_query = Product.select_by_pn('C1005X5R1V104K', self.wspace.memory)
assert len(c5_prod_query) == 1
c5_prod = c5_prod_query[0]
assert c5_prod.equals(test_c5_prod)
test3_c63_query = test3_bom.select_parts_by_name('C63', self.wspace.memory)
# Mystery 'None' printout here... huh?
try:
#print 'Trying assertion: assert len(test3_c63_query) == 1'
assert len(test3_c63_query) == 1
except AssertionError:
print 'Assertion failed: assert len(test3_c63_query) == 1'
print 'len(test3_c63_query): ', len(test3_c63_query)
for p in test3_c63_query:
print 'test3_c63_query contains this part: ', p
raise AssertionError
test3_c63 = test3_c63_query[0]
#print '\ntest3_c63: \n', test3_c63.show() VOLT attribute is being preserved OK
# C63 has a VOLT = 25V attribute, which C5 does not.
# Therefore, C63's product should remain unset.
try:
#print 'Trying assertion: assert test3_c63.product is None'
assert test3_c63.product is None
except AssertionError:
print 'Assertion failed: assert test3_c63.product is None'
attribs = []
print test3_c63
print test3_c5
cur = self.wspace.memory.cursor()
cur.execute('SELECT * FROM part_attributes')
#params = ('C63', 'test3',)
#cur.execute('SELECT * FROM part_attributes WHERE part=? AND project=? ORDER BY id', params)
for row in cur.fetchall():
attribs.append((row[0], row[1], row[2], row[3], row[4]))
print 'ATTRIBUTES TABLE: \n', attribs
cur.close()
raise AssertionError
# Check: Attribs, Name, Proj, Prod
assert test3_c5.equals(test3_c63, True, False, True, False) == False
assert test3_c5.equals(test3_c63, False, False, False, False) == True
assert test3_c5.equals(test3_c63, False, False, False, True) == False
assert test3_c5.equals(test3_c63, False, False, True, False) == True
try:
assert test3_c63.equals(test3_c5, True, False, True, False) == False
except AssertionError:
print 'Assertion failed: assert test3_c63.equals(test3_c5, True, False, True, False) == False'
print test3_c63
print test3_c5
raise AssertionError
assert test3_c63.equals(test3_c5, False, False, True, False) == False
assert test3_c63.equals(test3_c5, True, False, False, False) == False
assert test3_c63.equals(test3_c5, False, False, False, False) == False
other_wspace.memory.close()
del other_wspace
finally:
self.wspace.memory.close()
def setUp(self):
self.bom = BOM()
self.movies = self.bom.get_chart()
def setUp(self):
self.bom = BOM()
self.movies = self.bom.get_chart()
