Làm cách nào tôi có thể giả mạo inet_client_addr () để kiểm tra đơn vị trong PostgreSQL?

Tôi có một thủ tục được lưu trữ đơn giản mà giá trị trả về phụ thuộc vào giá trị của inet_client_addr(). Làm cách nào tôi có thể ghi đè inet_client_addr()cho mục đích kiểm tra đơn vị khi kiểm tra quy trình được lưu trữ của mình?

Giải pháp duy nhất tôi nghĩ ra cho đến nay là tạo một hàm bao bọc xung quanh inet_client_addr():

CREATE FUNCTION my_inet_client_addr() RETURNS INET AS $$
    SELECT inet_client_addr();
$$ LANGUAGE sql;

Sau đó sử dụng nó trong chức năng của tôi:

CREATE local_connection() RETURNS BOOLEAN AS $$
    SELECT my_inet_client_addr() = '127.0.0.1';
$$ LANGUAGE sql;

Sau đó, trong bài kiểm tra đơn vị của tôi, tôi có thể xác định lại my_inet_client_addr():

BEGIN;
SELECT PLAN(2);
REPLACE FUNCTION my_inet_client_addr() RETURNS INET AS $$
    SELECT '127.0.0.1'::INET;
$$ LANGUAGE sql;

is(local_connection(),TRUE,'Connection from 127.0.0.1 is local');

REPLACE FUNCTION my_inet_client_addr() RETURNS INET AS $$
    SELECT '192.168.1.1'::INET;
$$ LANGUAGE sql;

is(local_connection(),FALSE,'Connection from 192.168.1.1. is not local');

ROLLBACK;

Có cách nào để thực hiện tương tự mà không có chức năng bao bọc my_inet_client_addr()?

inet_client_addr()là một chức năng thông tin hệ thống.
Nó nằm trong lược đồ pg_cataloggiống như các hàm dựng sẵn khác (ngoại trừ các mô đun bổ sung).

pg_catalogtự động là một phần của search_path. Mỗi tài liệu:

Ngoài publiccác lược đồ do người dùng tạo và, các cơ sở dữ liệu chứa một pg_cataloglược đồ chứa các bảng hệ thống và tất cả các kiểu dữ liệu, hàm và toán tử tích hợp. pg_catalogluôn luôn là một phần hiệu quả của con đường tìm kiếm. Nếu nó không được đặt tên rõ ràng trong đường dẫn thì nó sẽ được tìm kiếm ngầm trước khi tìm kiếm các lược đồ của đường dẫn. Điều này đảm bảo rằng các tên tích hợp sẽ luôn có thể tìm thấy. Tuy nhiên, bạn có thể đặt rõ ràng pg_catalogở cuối đường dẫn tìm kiếm của mình nếu bạn muốn có tên do người dùng xác định ghi đè tên tích hợp.

Nhấn mạnh đậm của tôi.
Vì vậy, chúng tôi tạo một lược đồ chuyên dụng và đặt nó trước pg_catalog trong search_path:

CREATE SCHEMA override;

CREATE OR REPLACE FUNCTION override.inet_client_addr()
  RETURNS inet AS
$func$
SELECT '127.0.0.1'::inet
$func$ language sql STABLE;

SET search_path = override, pg_catalog, public;

Sau đó, cuộc gọi của bạn sẽ tìm thấy chức năng ghi đè tùy chỉnh của bạn trước tiên:

SELECT inet_client_addr();

Câu đố SQL

Hãy chắc chắn rằng, người dùng không có đặc quyền không thể tạo các đối tượng trong overridelược đồ hoặc họ có thể chơi tất cả các loại thủ thuật trên bạn. Đó không phải là trường hợp mặc định. Mỗi tài liệu:

Không có đặc quyền nào được cấp cho PUBLIC theo mặc định trên các bảng, cột, lược đồ hoặc không gian bảng.

Nhấn mạnh đậm của tôi.

Cần thận trọng nếu cùng một người dùng được phép tạo các đối tượng trong cơ sở dữ liệu.
Mỗi tài liệu:

Lược đồ đầu tiên có tên trong đường dẫn tìm kiếm được gọi là lược đồ hiện tại. Ngoài việc là lược đồ đầu tiên được tìm kiếm, nó cũng là lược đồ trong đó các bảng mới sẽ được tạo nếu CREATE TABLElệnh không chỉ định tên lược đồ.

Luôn luôn chỉ định một tên lược đồ cho CREATEcác lệnh và không cho phép đối tượng tất cả theo mặc định để loại trừ các lỗi.

Phải là postgresngười dùng hoặc khác Superusercho tôi. Chỉ cần create schemađặc quyền cho câu trả lời của Erwin.

bắt đầu kiểm tra:

alter function inet_client_addr() rename to _inet_client_addr;

create function inet_client_addr() returns inet language sql stable cost 1 as '
  select ''192.168.0.1''::inet;
';

select inet_client_addr();

kiểm tra kết thúc

alter function inet_client_addr() rename to inet_client_addr_mock;

alter function _inet_client_adds() rename to inet_client_addr;

--the regular function:
select inet_client_addr();