old_contents: # -*- coding: utf-8 -*-
  
  from unittest import TestCase
  
  from byceps.application import create_app
  from byceps.blueprints.brand.models import Brand
  from byceps.blueprints.party.models import Party
  from byceps.database import db
  
  
  class AbstractAppTestCase(TestCase):
  
      def setUp(self):
          self.app = create_app('test', initialize=False)
  
          self.db = db
          db.app = self.app
  
          db.drop_all()
          db.create_all()
  
          self.create_brand_and_party()
  
          self.client = self.app.test_client()
  
      def create_brand_and_party(self):
          self.brand = Brand(id='acme', title='ACME')
          db.session.add(self.brand)
  
          self.party = Party(id='acme-2014', brand=self.brand, title='ACME 2014')
          db.session.add(self.party)
  
          db.session.commit()
  
      def tearDown(self):
          db.session.remove()
          db.drop_all()
  
new_contents: # -*- coding: utf-8 -*-
  
  from unittest import TestCase
  
  from byceps.application import create_app
  from byceps.blueprints.brand.models import Brand
  from byceps.blueprints.party.models import Party
  from byceps.database import db
  
  
  class AbstractAppTestCase(TestCase):
  
      def setUp(self, env='test'):
          self.app = create_app(env, initialize=False)
  
          self.db = db
          db.app = self.app
  
          db.drop_all()
          db.create_all()
  
          self.create_brand_and_party()
  
          self.client = self.app.test_client()
  
      def create_brand_and_party(self):
          self.brand = Brand(id='acme', title='ACME')
          db.session.add(self.brand)
  
          self.party = Party(id='acme-2014', brand=self.brand, title='ACME 2014')
          db.session.add(self.party)
  
          db.session.commit()
  
      def tearDown(self):
          db.session.remove()
          db.drop_all()
  
subject: Allow use of a custom environment for tests.