reuse the Split Adapter provate method to return treatment type

Co-authored-by: Desmond Sadler <[email protected]> Co-authored-by: Si Cheng Zhou <[email protected]>
hoverinc · Oct 5, 2023 · 0977e38 · 0977e38
1 parent c8582a6
commit 0977e38
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/lib/this_feature/adapters/split_io.rb b/lib/this_feature/adapters/split_io.rb
@@ -30,8 +30,8 @@ def off?(flag_name, context: EMPTY_CONTEXT, data: {}, record: nil)
         treatment(flag_name, context: context, data: data, record: record).eql?('off')
       end
 
-      def get_treatment(flag_name, record: nil)
-        client.get_treatment(record, flag_name)
+      def get_treatment(flag_name, context: EMPTY_CONTEXT, data: {}, record: nil)
+        treatment(flag_name, context: context, data: data, record: record)
       end
 
       private

diff --git a/lib/this_feature/flag.rb b/lib/this_feature/flag.rb
@@ -23,7 +23,7 @@ def control?
     end
 
     def get_treatment
-      adapter.get_treatment(flag_name, record: record)
+      adapter.get_treatment(flag_name, context: context, data: data, record: record)
     end
   end
 end