def test_get_site_name_blank(self): with mock.patch('settings_composer.environment.os') as _os: _os.environ = {} self.assertEqual( environment.get_site_name(), '' )
def test_get_site_name(self): with mock.patch('settings_composer.environment.os') as _os: _os.environ = { constants.SITE_VARIABLE_NAME: 'mysite' } self.assertEqual( environment.get_site_name(), 'mysite' )