test_lambada.py 文件源码

def test_bouncer_env(self):
        """
        Validate the environment variable loader works as expected.
        """
        # Test it doesn't leak
        with patch.dict('lambada.os.environ', dict(BLAH='asdf'), clear=True):
            config = lambada.get_config_from_env()
            self.assertFalse(config)

        # Test it works with default
        with patch.dict(
            'lambada.os.environ', dict(BOUNCER_BLAH='asdf'), clear=True
        ):
            config = lambada.get_config_from_env()
            self.assertDictEqual(config, dict(blah='asdf'))

        # Test that we can override the prefix
        with patch.dict(
            'lambada.os.environ',
            dict(STUFF_BLAH='asdf', STUFFS_BLAH='jkl;'),
            clear=True
        ):
            config = lambada.get_config_from_env('STUFF_')
            self.assertDictEqual(config, dict(blah='asdf'))