def create_master_data_directory():
"""Create the master data directory, append relevant environment variable to admin user."""
import params
Directory(params.master_data_directory,
action="create",
recursive=True,
owner=params.admin_user)
utilities.append_bash_profile(
params.admin_user, 'export MASTER_DATA_DIRECTORY="%s";' %
params.master_data_segment_directory)
def add_psql_variables(user=None):
import params
if user == None:
user = params.admin_user
utilities.append_bash_profile(
user, "source %s;" %
path.join(params.absolute_installation_path, 'greenplum_path.sh'))
utilities.append_bash_profile(user,
'export PGPORT="%s";' % params.master_port)
utilities.append_bash_profile(
user, 'export PGDATABASE="%s";' % params.database_name)
def test_append_to_profile_run(self, append_to_file_mock, execute_mock):
with Environment('/'):
utilities.append_bash_profile('testuser', 'testcommand', True)
self.assertTrue(append_to_file_mock.called)
self.assertTrue(execute_mock.called)
def test_append_to_profile(self, append_to_file_mock):
utilities.append_bash_profile('testuser', 'testcommand')
self.assertTrue(append_to_file_mock.called)