def test_teamwise_azure_blob_storage_plugin(self, mock_get_secrets): parameters = {"storageConfigNames": ["STORAGE_AZURE_BLOB"]} mock_get_secrets.side_effect = self.get_secret command_generator = storage_command_generator.StorageCommandGenerator() storage_commands = command_generator.generate_plugin_commands( parameters) expect_commands = [ "umask 000", "apt-get update", "apt-get install --assume-yes wget curl lsb-release apt-transport-https", "valid_release=('14.04' '15.10' '16.04' '16.10' '17.04' '17.10' '18.04' '18.10' '19.04')", "release=`lsb_release -r | cut -f 2`", "if [[ ! ${valid_release[@]} =~ ${release} ]];" + " then echo \"Invalid OS version for Azureblob!\"; exit 1; fi", "wget https://packages.microsoft.com/config/ubuntu/${release}/packages-microsoft-prod.deb", "dpkg -i packages-microsoft-prod.deb", "apt-get update", "apt-get install --assume-yes blobfuse fuse", "mkdir --parents /mnt/resource/blobfusetmp/azure_blob_test", "echo \"accountName accountname\" >> /azure_blob_test.cfg", "echo \"accountKey key\" >> /azure_blob_test.cfg", "echo \"containerName containername\" >> /azure_blob_test.cfg", "chmod 600 /azure_blob_test.cfg", "mkdir --parents /tmp_azure_blob_test_root", "blobfuse /tmp_azure_blob_test_root --tmp-path=/mnt/resource/blobfusetmp/azure_blob_test" + " --config-file=/azure_blob_test.cfg -o attr_timeout=240" + " -o entry_timeout=240 -o negative_timeout=120", "mkdir --parents /mnt/data", "mkdir --parents /tmp_azure_blob_test_root/data", "rm -r /mnt/data", "ln -s /tmp_azure_blob_test_root/data /mnt/data" ] assert storage_commands == expect_commands
def test_default_storage_plugin(self, mock_get_secrets): mock_get_secrets.side_effect = self.get_secret command_generator = storage_command_generator.StorageCommandGenerator() default_storage_commands = command_generator.generate_plugin_commands( []) expect_commands = [ "umask 000", "mkdir --parents /tmp_SRV_BJ_root", "apt-get update;apt-get install --assume-yes nfs-common;", "mount -t nfs4 10.151.41.14:/data/share/drbdha /tmp_SRV_BJ_root", "mkdir --parents /mnt/data", "mkdir --parents /tmp_SRV_BJ_root/data", "mkdir --parents /mnt/home", "mkdir --parents /tmp_SRV_BJ_root/users/${PAI_USER_NAME}", "umount -l /tmp_SRV_BJ_root", "rm -r /tmp_SRV_BJ_root", "mount -t nfs4 10.151.41.14:/data/share/drbdha/data /mnt/data", "mount -t nfs4 10.151.41.14:/data/share/drbdha/users/${PAI_USER_NAME} /mnt/home" ] assert default_storage_commands == expect_commands
def test_teamwise_samba_storage_plugin(self, mock_get_secrets): parameters = {"storageConfigNames": ["STORAGE_SAMBA"]} mock_get_secrets.side_effect = self.get_secret command_generator = storage_command_generator.StorageCommandGenerator() storage_commands = command_generator.generate_plugin_commands( parameters) expect_commands = [ "umask 000", "mkdir --parents /tmp_samba_test_root", "apt-get update;apt-get install --assume-yes cifs-utils;", "mount -t cifs //10.151.41.14/data/share/drbdha /tmp_samba_test_root" + " -o vers=3.0,username=user,password=password,domain=domain", "mkdir --parents /mnt/data", "mkdir --parents /tmp_samba_test_root/data", "umount -l /tmp_samba_test_root", "rm -r /tmp_samba_test_root", "mount -t cifs //10.151.41.14/data/share/drbdha/data /mnt/data" + " -o vers=3.0,username=user,password=password,domain=domain" ] assert storage_commands == expect_commands
def test_teamwise_azure_file_storage_plugin(self, mock_get_secrets): parameters = {"storageConfigNames": ["STORAGE_AZURE_FILE"]} mock_get_secrets.side_effect = self.get_secret command_generator = storage_command_generator.StorageCommandGenerator() storage_commands = command_generator.generate_plugin_commands( parameters) expect_commands = [ "umask 000", "mkdir --parents /tmp_azure_file_test_root", "apt-get update;apt-get install --assume-yes cifs-utils sshpass;", "mount -t cifs //datastore/fileshare /tmp_azure_file_test_root" + " -o vers=3.0,username=accountname,password=key,dir_mode=0777,file_mode=0777,serverino", "mkdir --parents /mnt/data", "mkdir --parents /tmp_azure_file_test_root/data", "umount -l /tmp_azure_file_test_root", "rm -r /tmp_azure_file_test_root", "mount -t cifs //datastore/fileshare/data /mnt/data" + " -o vers=3.0,username=accountname,password=key,dir_mode=0777,file_mode=0777,serverino" ] assert storage_commands == expect_commands
def test_teamwise_nfs_storage_plugin(self, mock_get_secrets): mock_get_secrets.side_effect = self.get_secret parameters = {"storageConfigNames": ["STORAGE_NFS"]} command_generator = storage_command_generator.StorageCommandGenerator() storage_commands = command_generator.generate_plugin_commands( parameters) expect_commands = [ "umask 000", "mkdir --parents /tmp_SRV_BJ_root", "apt-get update;apt-get install --assume-yes nfs-common;", "mount -t nfs4 10.151.41.14:/data/share/drbdha /tmp_SRV_BJ_root", "mkdir --parents /mnt/data", "mkdir --parents /tmp_SRV_BJ_root/data", "mkdir --parents /mnt/home", "mkdir --parents /tmp_SRV_BJ_root/users/${PAI_USER_NAME}", "umount -l /tmp_SRV_BJ_root", "rm -r /tmp_SRV_BJ_root", "mount -t nfs4 10.151.41.14:/data/share/drbdha/data /mnt/data", "mount -t nfs4 10.151.41.14:/data/share/drbdha/users/${PAI_USER_NAME} /mnt/home" ] assert storage_commands == expect_commands