def main(): parser = argparse.ArgumentParser(description='Manage your environment.') parser.add_argument( '-f', '--file', metavar='<FILE>', type=str, action='store', help='File to load variables from') parser.add_argument( '-d', '--directory', metavar='<DIR>', type=str, action='store', help='Directory to load variables from, works just like `envparse\'') subparsers = parser.add_subparsers(title='subcommands', metavar='COMMAND') get_parser = subparsers.add_parser( 'get', help='Get and print out the value of the variable <VAR>') get_parser.add_argument('cmd_get') get_uri_parser = subparsers.add_parser( 'get-uri', help='Exposes the URI parser API') get_uri_parser.add_argument('cmd_get_uri', nargs=2) args = parser.parse_args() if args.file: env = Environment.from_file(args.file) if args.directory: env = Environment.from_folder(args.directory) if hasattr(args, 'cmd_get'): return env.get(args.cmd_get) if hasattr(args, 'cmd_get_uri'): part, variable = args.cmd_get_uri return getattr(env.get_uri(variable), part)
def test_milieu_environment_from_directory_set(_io): # Given that I load variables to my env from a folder env = Environment.from_folder( os.path.join(os.path.dirname(__file__), './fixtures/env')) # When I set some stuff env.set('CITY', 'NEW-YORK') # Then I see that we always try to write the file _io.open.return_value.write.assert_called_once_with('NEW-YORK')
def test_milieu_environment_from_directory_del(_os, _io): # Given that I have a folder environment with an item `CITY` env = Environment.from_folder('./path') env.set('CITY', 'NEW-YORK') # We need the path.join function over there, so we need to restore it # manually _os.path.join.side_effect = os.path.join # When I remove that item del env['CITY'] # Then I can see that the unlink function was called properly _os.unlink.assert_called_once_with('./path/CITY')
def test_merge_folder_environment(): "Environment.update() Should update a folder based environment with data from another one" # Given that I load variables to my env from a folder path = os.path.join(os.path.dirname(__file__), './fixtures/env') environment = Environment.from_folder(path) # When I merge another environment into the current one that *does # not* have the variable CITY environment.update(Environment(storage={'CITY': 'Porto Alegre'})) # Then I see that the old value was overwritten environment.get('CITY').should.equal('Porto Alegre') # And then I just cleanup the file I created above del environment['CITY']
def test_milieu_environment_from_directory_set(): # Given that I load variables to my env from a folder path = os.path.join(os.path.dirname(__file__), './fixtures/env') env = Environment.from_folder(path) # When I set a new variable to my folder env env.set('CITY', 'NEW-YORK') # Then I see the file was created with the right content target = os.path.join(path, 'CITY') os.path.exists(target).should.be.true # And then I see that the value is also right open(target).read().should.equal('NEW-YORK') # And then I see that after removing the item, the file will also go away del env['CITY'] os.path.exists(target).should.be.false
def test_milieu_environment_from_directory(): # Given that I load variables to my env from a folder env = Environment.from_folder( os.path.join(os.path.dirname(__file__), './fixtures/env')) # When I try to list all the variables inside of that folder sorted(env.items(), key=lambda x: x[0]).should.equal([ ('ENABLE_SOMETHING', u''), ('PI', u'3.14'), ('SERVER_URI', u'smtp://[email protected]:[email protected]:25'), ]) # When I try to find the variables, then I see they're there correctly env.get_bool('ENABLE_SOMETHING').should.be.false env.get_bool('ENABLE_SOMETHING_ELSE', True).should.be.true env.get_float('PI').should.equal(3.14) env.get_uri('SERVER_URI').host.should.equal('mserver.com') env.get_uri('SERVER_URI').user.should.equal('*****@*****.**')
def test_milieu_environment_from_directory_items(_os, _io): # Given that I load variables to my env from a folder env = Environment.from_folder( os.path.join(os.path.dirname(__file__), './fixtures/env')) _os.listdir.return_value = ['ENABLE_SOMETHING', 'PI', 'SERVER_URI'] _io.open.return_value.read.side_effect = [ '', '3.14', 'smtp://[email protected]:[email protected]:25', ] # When I try to list all the variables inside of that folder sorted(env.items(), key=lambda x: x[0]).should.equal([ ('ENABLE_SOMETHING', u''), ('PI', u'3.14'), ('SERVER_URI', u'smtp://[email protected]:[email protected]:25'), ])
def test_milieu_environment_from_directory_get(_os, _io): # Given that I load variables to my env from a folder env = Environment.from_folder( os.path.join(os.path.dirname(__file__), './fixtures/env')) _os.listdir.return_value = ['ENABLE_SOMETHING', 'PI', 'SERVER_URI'] _io.open.return_value.read.side_effect = [ '', IOError, '3.14', 'smtp://[email protected]:[email protected]:25', 'smtp://[email protected]:[email protected]:25', ] # When I try to find the variables, then I see they're there correctly env.get_bool('ENABLE_SOMETHING').should.be.false env.get_bool('ENABLE_SOMETHING_ELSE', True).should.be.true env.get_float('PI').should.equal(3.14) env.get_uri('SERVER_URI').host.should.equal('mserver.com') env.get_uri('SERVER_URI').user.should.equal('*****@*****.**')