fix(ruby): update test examples

Bearer · Oct 7, 2024 · 1cc8376 · 1cc8376
1 parent d3325ab
commit 1cc8376
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/tests/ruby/rails/sql_injection/testdata/injected_params.rb b/tests/ruby/rails/sql_injection/testdata/injected_params.rb
@@ -4,8 +4,6 @@
 find_by!("oops #{params[:oops]}")
 # bearer:expected ruby_rails_sql_injection
 User.joins("INNER JOIN #{params[:oops]}")
-# bearer:expected ruby_rails_sql_injection
-select("#{params[:oops]} AS oops")
 
 # chained case
 # bearer:expected ruby_rails_sql_injection

diff --git a/tests/ruby/rails/sql_injection/testdata/shared_methods.rb b/tests/ruby/rails/sql_injection/testdata/shared_methods.rb
@@ -14,3 +14,7 @@
 ITEMS.count
 ITEMS.count(:apple)
 
+# bearer:expected ruby_rails_sql_injection
+User.select("#{params[:oops]} AS oops")
+# bearer:expected ruby_rails_sql_injection
+user.posts.select("#{params[:oops]} AS oops")