Fix tests for new connector Context methods

services/importer/spec/unit/connector_runner_spec.rb

@@ -47,6 +47,8 @@ describe CartoDB::Importer2::ConnectorRunner do
       # Simulate connector success by ignoring all db opeartions
       Carto::Connector::Context.any_instance.stubs(:execute_as_superuser).returns(nil)
       Carto::Connector::Context.any_instance.stubs(:execute).returns(nil)
+      Carto::Connector::Context.any_instance.stubs(:execute_as_superuser_with_timeout).returns(nil)
+      Carto::Connector::Context.any_instance.stubs(:execute_with_timeout).returns(nil)
     end
 
     it "Succeeds if parameters are correct" do
@@ -71,6 +73,9 @@ describe CartoDB::Importer2::ConnectorRunner do
           Cartodb.with_config connectors: config do
             connector = CartoDB::Importer2::ConnectorRunner.new(parameters.merge(provider: provider).to_json, options)
             connector.run
+            puts ">"*60
+            puts @fake_log.to_s
+            puts ">"*60
             connector.success?.should be true
             connector.provider_name.should eq provider
           end
@@ -172,7 +177,9 @@ describe CartoDB::Importer2::ConnectorRunner do
     before(:all) do
       # Simulate connector success when executing non-privileged SQL
       Carto::Connector::Context.any_instance.stubs(:execute_as_superuser).returns(nil)
-      Carto::Connector::Context.any_instance.stubs(:execute).raises("SQL EXECUTION ERROR")
+      Carto::Connector::Context.any_instance.stubs(:execute_as_superuser_with_timeout).returns(nil)
+      Carto::Connector::Context.any_instance.stubs(:execute).returns(nil)
+      Carto::Connector::Context.any_instance.stubs(:execute_with_timeout).raises("SQL EXECUTION ERROR")
     end
 
     it "Always fails" do
@@ -209,6 +216,9 @@ describe CartoDB::Importer2::ConnectorRunner do
   describe 'with invalid provider' do
     Carto::Connector::Context.any_instance.stubs(:execute_as_superuser).returns(nil)
     Carto::Connector::Context.any_instance.stubs(:execute).returns(nil)
+    Carto::Connector::Context.any_instance.stubs(:execute_as_superuser_with_timeout).returns(nil)
+    Carto::Connector::Context.any_instance.stubs(:execute_with_timeout).returns(nil)
+
 
     it "Fails at creation" do
       with_feature_flag @user, 'carto-connectors', true do