Make replace TwilioHttpClient in Python snippets more resilient

carlrempel · Sep 21, 2017 · 1c6a949 · 1c6a949
1 parent 0d256d2
commit 1c6a949
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/tools/snippet-testing/language_handler/python_6.rb b/tools/snippet-testing/language_handler/python_6.rb
@@ -23,9 +23,12 @@ def text_with_custom_header(file_content)
 
     def replace_twilio_client_initialization(file_content)
       cert_path = ENV['FAKE_CERT_PATH']
-      updated_file_content = file_content.gsub! 'Client(account_sid, auth_token)',
-        'Client(account_sid, auth_token, http_client=FakerHttpClient())'
-      updated_file_content == nil ? file_content : updated_file_content
+      result = ''
+      file_content.each_line do |line|
+        result += line.include?('client = Client(') ?
+           line.gsub!(')', ', http_client=FakerHttpClient())') : line
+      end
+      result
     end
   end
 end