class DatabaseProductTestCase(unittest.TestCase):
product_list = [
'Cassandra', 'CouchDB', 'MariaDB', 'MongoDB', 'MySQL',
'PostgreSQL', 'Redis',
]
def setUp(self):
self.dbp = DatabaseProduct()
for name in self.product_list:
product = Product(name=name)
self.dbp.session.add(product)
self.dbp.session.commit()
def tearDown(self):
Base = declarative_base()
Base.metadata.drop_all(self.dbp.engine)
def test_vote(self):
redis = self.dbp.session.query(Product).filter_by(name='Redis').first()
before_vote = redis.vote
self.dbp.vote('Redis')
after_vote = redis.vote
self.assertEqual(before_vote, 0)
self.assertEqual(after_vote, 1)
def setUp(self):
self.dbp = DatabaseProduct()
for name in self.product_list:
product = Product(name=name)
self.dbp.session.add(product)
self.dbp.session.commit()
