tests/testthat/test-09_file_transfer.R

In adamryczkowski/clustertools:

library(clustertools)

library(testthat)

source("remote_host.R")

context(paste0('Transfer files to ', remote_host))

prepare_file<-function(n=1000) {
  tmp<-list(payload=rnorm(n), name="tempobj")
  loctmpdir1<-base::tempdir()
  saveRDS(tmp, file = file.path(loctmpdir1, 'file.rds'), compress=FALSE)
  return(list(dir=loctmpdir1, file='file.rds'))
}

test_that(paste0("Simple transfer of file to ", remote_host), {
  gc()
  options(warn=2)
  srv_loc<-RemoteServer$new(remote_host)
  fileinfo<-prepare_file()
  expect_true(is.numeric(srv_loc$send_file(local_path =file.path(fileinfo$dir, fileinfo$file), remote_path = file.path(fileinfo$dir, "file2.rds"), timeout = 0)))

  srv_loc$finalize()
})

test_that(paste0("Simple transfer of file from ", remote_host), {
  gc()
  options(warn=2)
  srv_loc<-RemoteServer$new(remote_host)
  srv_loc$execute_job({
    tmp<-list(payload=rnorm(1000), name="tempobj")
    filename<-file.path(base::tempdir(), 'file.rds')
    saveRDS(tmp, file = filename, compress=FALSE)
  }, job_name = "prepare_file")

#  debug(srv_loc$execute_job)
  remote_file<-srv_loc$execute_job(filename)

  local_file<-file.path(base::tempdir(), 'file2.rds')
  out<-srv_loc$receive_file(local_path =local_file, remote_path = remote_file)


  expect_equal(local_file, out)
  expect_equal(tools::md5sum(local_file), tools::md5sum(out))
  srv_loc$finalize()
})

test_that(paste0("Transfer of file that requires new directory to ", remote_host), {
  gc()
  options(warn=2)
  srv_loc<-RemoteServer$new(remote_host)
  fileinfo<-prepare_file()
  expect_true(is.numeric(srv_loc$send_file(local_path =file.path(fileinfo$dir, fileinfo$file),
                                          remote_path = file.path(paste0(fileinfo$dir, "_remote"), fileinfo$file), timeout = 0)))

  srv_loc$finalize()
})