library(massdataset)
library(testthat)
expression_data <-
  as.data.frame(matrix(1:20, nrow = 5, ncol = 4))

colnames(expression_data) <-
  paste0("sample", 1:4)

rownames(expression_data) <-
  paste0("variable", 1:5)

sample_info <-
  data.frame(
    sample_id = colnames(expression_data),
    injection.order = 1:4,
    class = "QC",
    group = "case"
  )

variable_info <-
  data.frame(
    variable_id = rownames(expression_data),
    mz = 1:5,
    rt = 2:6
  )

object <-
  create_mass_dataset(
    expression_data = expression_data,
    sample_info = sample_info,
    variable_info = variable_info
  )

test_that(desc = "select",
          code = {
            ###expression_data
            object1 <-
              activate_mass_dataset(object, what = "expression_data")
            object1 <-
              select(.data = object1, -sample2)
            result1 <-
              check_mass_dataset(
                expression_data = object1@expression_data,
                sample_info = object1@sample_info,
                variable_info = object1@variable_info,
                sample_info_note = object1@sample_info_note,
                variable_info_note = object1@variable_info_note
              )
            testthat::expect_equal(object = result1, "all good.")
            
            ###sample_info
            object2 <-
              activate_mass_dataset(object, what = "sample_info")
            object2 <-
              select(.data = object2, -group)
            result2 <-
              check_mass_dataset(
                expression_data = object2@expression_data,
                sample_info = object2@sample_info,
                variable_info = object2@variable_info,
                sample_info_note = object2@sample_info_note,
                variable_info_note = object2@variable_info_note
              )
            testthat::expect_equal(object = result2, "all good.")
            
            ###sample_info2
            object2 <-
              activate_mass_dataset(object, what = "sample_info")
            object2 <-
              select(.data = object2, -c(group, injection.order, class))
            testthat::expect_equal(ncol(object2@sample_info), 1)
            
            ###sample_info3
            object3 <-
              activate_mass_dataset(object, what = "sample_info")
            testthat::expect_error(object3 <-
                                     select(.data = object3, -sample_id))
            
            ###variable_info
            object3 <-
              activate_mass_dataset(object, what = "variable_info")
            object3 <-
              select(.data = object3, -rt)
            result3 <-
              check_mass_dataset(
                expression_data = object3@expression_data,
                sample_info = object3@sample_info,
                variable_info = object3@variable_info,
                sample_info_note = object3@sample_info_note,
                variable_info_note = object3@variable_info_note
              )
            testthat::expect_equal(object = result3, "all good.")
            
            ###variable_info2
            object3 <-
              activate_mass_dataset(object, what = "variable_info")
            testthat::expect_error(object3 <-
                                     select(.data = object3, -variable_id))
          })