Add tests

tests/ruby/helpers/pgcat_process.rb

@@ -112,10 +112,16 @@ def admin_connection_string
  "postgresql://#{username}:#{password}@0.0.0.0:#{@port}/pgcat"
  end
 
- def connection_string(pool_name, username, password = nil)
+ def connection_string(pool_name, username, password = nil, parameters: {})
  cfg = current_config
  user_idx, user_obj = cfg["pools"][pool_name]["users"].detect { |k, user| user["username"] == username }
- "postgresql://#{username}:#{password || user_obj["password"]}@0.0.0.0:#{@port}/#{pool_name}"
+ connection_string = "postgresql://#{username}:#{password || user_obj["password"]}@0.0.0.0:#{@port}/#{pool_name}"
+
+ # Add the additional parameters to the connection string
+ parameter_string = parameters.map { |key, value| "#{key}=#{value}" }.join("&")
+ connection_string += "?#{parameter_string}" unless parameter_string.empty?
+
+ connection_string
  end
 
  def example_connection_string

tests/ruby/misc_spec.rb

@@ -294,6 +294,30 @@
 
  expect(processes.primary.count_query("DISCARD ALL")).to eq(10)
  end
+
+ it "Respects tracked parameters on startup" do
+ conn = PG::connect(processes.pgcat.connection_string("sharded_db", "sharding_user", parameters: { "application_name" => "my_pgcat_test" }))
+
+ expect(conn.async_exec("SHOW application_name")[0]["application_name"]).to eq("my_pgcat_test")
+ conn.close
+ end
+
+ it "Respect tracked parameter on set statemet" do
+ conn = PG::connect(processes.pgcat.connection_string("sharded_db", "sharding_user"))
+
+ conn.async_exec("SET application_name to 'my_pgcat_test'")
+ expect(conn.async_exec("SHOW application_name")[0]["application_name"]).to eq("my_pgcat_test")
+ end
+
+
+ it "Ignore untracked parameter on set statemet" do
+ conn = PG::connect(processes.pgcat.connection_string("sharded_db", "sharding_user"))
+ orignal_statement_timeout = conn.async_exec("SHOW statement_timeout")[0]["statement_timeout"]
+
+ conn.async_exec("SET statement_timeout to 1500")
+ expect(conn.async_exec("SHOW statement_timeout")[0]["statement_timeout"]).to eq(orignal_statement_timeout)
+ end
+
  end
 
  context "transaction mode with transactions" do