test: added tests for same key being present in multiple sessions

Author: adrianprelipcean

db/tests/unit/agg_increment.test.sql

@@ -1,12 +1,17 @@
 BEGIN;
 -- Plan the tests.
-SELECT plan(2);
+SELECT plan(3);
 
 INSERT INTO rate_limit.sessions (id, name_, type_)
 SELECT
     '00000000-0000-0000-0000-000000000000'::uuid AS id,
     'dedicated-test'::text AS name_,
-    'aggregated'::text AS type_;
+    'aggregated'::text AS type_
+UNION
+SELECT
+    '00000000-1111-1111-1111-000000000000'::uuid AS id,
+    'dedicated-test-2'::text AS name_,
+    'aggregated'::text AS type_;;
 
 INSERT INTO rate_limit.records_aggregated (key, session_id)
 SELECT
@@ -23,6 +28,16 @@ SELECT results_eq(
     'rate_limit.agg_increment returns correct count for new key'
 );
 
+SELECT results_eq(
+    $have$
+    SELECT agg_increment AS count FROM rate_limit.agg_increment('new-key', '00000000-1111-1111-1111-000000000000')
+    $have$,
+    $want$
+    SELECT 1::int AS count;
+    $want$,
+    'rate_limit.agg_increment returns correct count for new key on different session'
+);
+
 SELECT results_eq(
     $have$
     SELECT agg_increment AS count FROM rate_limit.agg_increment('existing-key', '00000000-0000-0000-0000-000000000000')

db/tests/unit/ind_increment.test.sql

@@ -1,11 +1,16 @@
 BEGIN;
 -- Plan the tests.
-SELECT plan(3);
+SELECT plan(4);
 
 INSERT INTO rate_limit.sessions (id, name_, type_)
 SELECT
     '00000000-0000-0000-0000-000000000000'::uuid AS id,
     'dedicated-test'::text AS name_,
+    'individual'::text AS type_
+UNION
+SELECT
+    '00000000-1111-1111-1111-000000000000'::uuid AS id,
+    'dedicated-test-2'::text AS name_,
     'individual'::text AS type_;
 
 INSERT INTO rate_limit.individual_records (key, session_id)
@@ -23,6 +28,16 @@ SELECT results_eq(
     'rate_limit.ind_increment returns correct count for new key'
 );
 
+SELECT results_eq(
+    $have$
+    SELECT ind_increment AS count FROM rate_limit.ind_increment('new-key', '00000000-1111-1111-1111-000000000000')
+    $have$,
+    $want$
+    SELECT 1::int AS count;
+    $want$,
+    'rate_limit.ind_increment returns correct count for new key on another session'
+);
+
 SELECT results_eq(
     $have$
     SELECT ind_increment AS count FROM rate_limit.ind_increment('existing-key', '00000000-0000-0000-0000-000000000000')