Fix tests

lib/datadog/tracing/contrib/propagation/sql_comment/mode.rb

@@ -14,19 +14,19 @@ def enabled?
             end
 
             def service?
-              @mode == Ext::SERVICE
+              mode == Ext::SERVICE
             end
 
             def full?
-              @mode == Ext::FULL
+              mode == Ext::FULL
             end
 
             def prepend?
-              !@append
+              !append
             end
 
             def append?
-              @append
+              append
             end
           end
         end

spec/datadog/tracing/contrib/propagation/sql_comment/mode_spec.rb

@@ -1,6 +1,8 @@
 require 'datadog/tracing/contrib/propagation/sql_comment/mode'
 
 RSpec.describe Datadog::Tracing::Contrib::Propagation::SqlComment::Mode do
+  let(:append) { false }
+
   describe '#enabled?' do
     [
       ['disabled', false],
@@ -9,7 +11,7 @@
       ['undefined', false]
     ].each do |string, result|
       context "when given `#{string}`" do
-        subject { described_class.new(string).enabled? }
+        subject { described_class.new(string, append).enabled? }
         it { is_expected.to be result }
       end
     end
@@ -23,7 +25,7 @@
       ['undefined', false]
     ].each do |string, result|
       context "when given `#{string}`" do
-        subject { described_class.new(string).service? }
+        subject { described_class.new(string, append).service? }
         it { is_expected.to be result }
       end
     end
@@ -37,7 +39,31 @@
       ['undefined', false]
     ].each do |string, result|
       context "when given `#{string}`" do
-        subject { described_class.new(string).full? }
+        subject { described_class.new(string, append).full? }
+        it { is_expected.to be result }
+      end
+    end
+  end
+
+  describe '#prepend?' do
+    [
+      [false, true],
+      [true, false]
+    ].each do |value, result|
+      context "when given `#{value}`" do
+        subject { described_class.new('full', value).prepend? }
+        it { is_expected.to be result }
+      end
+    end
+  end
+
+  describe '#append?' do
+    [
+      [false, false],
+      [true, true]
+    ].each do |value, result|
+      context "when given `#{value}`" do
+        subject { described_class.new('full', value).append? }
         it { is_expected.to be result }
       end
     end