set_notice_receiver tests fixed

test/epgsql_cast.erl

@@ -6,7 +6,7 @@
 -module(epgsql_cast).
 
 -export([connect/2, connect/3, connect/4, close/1]).
--export([get_parameter/2, squery/2, equery/2, equery/3]).
+-export([get_parameter/2, set_notice_receiver/2, squery/2, equery/2, equery/3]).
 -export([prepared_query/3]).
 -export([parse/2, parse/3, parse/4, describe/2, describe/3]).
 -export([bind/3, bind/4, execute/2, execute/3, execute/4, execute_batch/2]).
@@ -43,6 +43,9 @@ close(C) ->
 get_parameter(C, Name) ->
     epgsqla:get_parameter(C, Name).
 
+set_notice_receiver(C, PidOrName) ->
+    epgsqla:set_notice_receiver(C, PidOrName).
+
 squery(C, Sql) ->
     Ref = epgsqla:squery(C, Sql),
     receive_result(C, Ref).

test/epgsql_incremental.erl

@@ -6,7 +6,7 @@
 -module(epgsql_incremental).
 
 -export([connect/2, connect/3, connect/4, close/1]).
--export([get_parameter/2, squery/2, equery/2, equery/3]).
+-export([get_parameter/2, set_notice_receiver/2, squery/2, equery/2, equery/3]).
 -export([prepared_query/3]).
 -export([parse/2, parse/3, parse/4, describe/2, describe/3]).
 -export([bind/3, bind/4, execute/2, execute/3, execute/4, execute_batch/2]).
@@ -41,6 +41,9 @@ close(C) ->
 get_parameter(C, Name) ->
     epgsqli:get_parameter(C, Name).
 
+set_notice_receiver(C, PidOrName) ->
+    epgsqli:set_notice_receiver(C, PidOrName).
+
 squery(C, Sql) ->
     Ref = epgsqli:squery(C, Sql),
     case receive_results(C, Ref, []) of