import os

import unittest

import retrying

class ClearEnvTest(unittest.TestCase):

'''

This class clears the environment variables that the

library uses for clean testing.

'''

def setUp(self):

# if you have these specified in the environment it will break tests

to_save = [

'STACKIFY_APPLICATION',

'STACKIFY_ENVIRONMENT',

'STACKIFY_API_KEY',

'STACKIFY_API_URL',

]

self.saved = {}

for key in to_save:

if key in os.environ:

self.saved[key] = os.environ[key]

del os.environ[key]

def tearDown(self):

# restore deleted environment variables

for key, item in self.saved.items():

os.environ[key] = item

del self.saved