def test_mget_all(self):
'''Can get a list of all environments
'''
environment.create('stage')
environment.create('prod')
actual = environment.mget_all(offset=0, limit=10)
eq_([env['environment_name'] for env in actual],
[self.environment_name, 'stage', 'prod'])
def setup_class(cls):
'''load data & set up expected results
'''
with gus.config.get_db_conn().cursor() as c:
c.execute("TRUNCATE TABLE environments CASCADE")
cls.expected_id_list = []
cls.expected_id_list.append(environment.create('production'))
cls.expected_id_list.append(environment.create('dev'))
cls.expected_id_list.append(environment.create('testing'))
cls.expected_object_list = [environment.get_by_id(eid)
for eid in cls.expected_id_list]
def setup(self):
'''Clear out the database & create a test project
'''
# TODO: refactor database cleanup
self.db_conn = gus.config.get_db_conn()
with self.db_conn.cursor() as c:
c.execute("TRUNCATE TABLE environments CASCADE")
self.environment_name = 'dev'
self.environment_id = environment.create(self.environment_name)