diff --git a/tests/testthat/test-sort.R b/tests/testthat/test-sort.R
index b074516..96717e1 100644
--- a/tests/testthat/test-sort.R
+++ b/tests/testthat/test-sort.R
@@ -50,3 +50,57 @@ test_that("sort_by_key sorts data by key variables from metacore spec", {
 
   expect_equal(result, data)
 })
+
+test_that("order_cols warns about deprecated dataset_name parameter", {
+  library(metacore)
+  library(haven)
+
+  load(metacore_example("pilot_ADaM.rda"))
+  data <- read_xpt(metatools_example("adsl.xpt"))
+
+  # Test that using dataset_name triggers deprecation warning
+  lifecycle::expect_deprecated(
+    order_cols(data, metacore, dataset_name = "ADSL")
+  )
+})
+
+test_that("order_cols still works with deprecated dataset_name parameter", {
+  library(metacore)
+  library(haven)
+
+  load(metacore_example("pilot_ADaM.rda"))
+  data <- read_xpt(metatools_example("adsl.xpt"))
+
+  # Suppress warning to test functionality
+  suppressWarnings({
+    result <- order_cols(data, metacore, dataset_name = "ADSL")
+  })
+
+  expect_s3_class(result, "data.frame")
+})
+
+test_that("sort_by_key warns about deprecated dataset_name parameter", {
+  library(metacore)
+  library(haven)
+
+  load(metacore_example("pilot_ADaM.rda"))
+  data <- read_xpt(metatools_example("adsl.xpt"))
+
+  lifecycle::expect_deprecated(
+    sort_by_key(data, metacore, dataset_name = "ADSL")
+  )
+})
+
+test_that("sort_by_key still works with deprecated dataset_name parameter", {
+  library(metacore)
+  library(haven)
+
+  load(metacore_example("pilot_ADaM.rda"))
+  data <- read_xpt(metatools_example("adsl.xpt"))
+
+  suppressWarnings({
+    result <- sort_by_key(data, metacore, dataset_name = "ADSL")
+  })
+
+  expect_s3_class(result, "data.frame")
+})