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
