fix: naics code mutation handles ciip_sector column

bcgov · Mar 22, 2021 · 8e145ad · 8e145ad
1 parent bb1acff
commit 8e145ad
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 7 deletions.
diff --git a/schema/deploy/mutations/create_naics_code_mutation.sql b/schema/deploy/mutations/create_naics_code_mutation.sql
@@ -5,13 +5,14 @@ begin;
 
 create or replace function ggircs_portal.create_naics_code_mutation(
   naics_code_input text,
+  ciip_sector_input text,
   naics_description_input text
 ) returns void as $function$
 
   -- Reset deleted at/by and update description on conflict
-  insert into ggircs_portal.naics_code(naics_code, naics_description)
-  values (naics_code_input, naics_description_input)
-  on conflict(naics_code) do update set naics_description=naics_description_input, deleted_at=null, deleted_by=null;
+  insert into ggircs_portal.naics_code(naics_code, ciip_sector, naics_description)
+  values (naics_code_input, ciip_sector_input, naics_description_input)
+  on conflict(naics_code) do update set naics_description=naics_description_input, ciip_sector=ciip_sector_input, deleted_at=null, deleted_by=null;
 
 $function$ language sql volatile;
 

diff --git a/schema/test/unit/mutations/create_naics_code_mutation_test.sql b/schema/test/unit/mutations/create_naics_code_mutation_test.sql
@@ -7,14 +7,14 @@ select plan(4);
 
 -- Function exists
 select has_function(
-  'ggircs_portal', 'create_naics_code_mutation', array['text', 'text'],
+  'ggircs_portal', 'create_naics_code_mutation', array['text', 'text', 'text'],
   'Function ggircs_portal.create_naics_code_mutation should exist'
 );
 
 -- Test Setup
 select test_helper.clean_ggircs_portal_schema();
 
-select ggircs_portal.create_naics_code_mutation('1234', 'init');
+select ggircs_portal.create_naics_code_mutation('1234', 'sector', 'init');
 
 select results_eq(
   $$
@@ -26,7 +26,7 @@ select results_eq(
 
 -- "Delete" naics code & re-run custom mutation
 update ggircs_portal.naics_code set deleted_at=now() where naics_code='1234';
-select ggircs_portal.create_naics_code_mutation('1234', 'updated');
+select ggircs_portal.create_naics_code_mutation('1234', 'sector', 'updated');
 
 select results_eq(
   $$

diff --git a/schema/verify/mutations/create_naics_code_mutation.sql b/schema/verify/mutations/create_naics_code_mutation.sql
@@ -2,6 +2,6 @@
 
 begin;
 
-select pg_get_functiondef('ggircs_portal.create_naics_code_mutation(text,text)'::regprocedure);
+select pg_get_functiondef('ggircs_portal.create_naics_code_mutation(text,text,text)'::regprocedure);
 
 rollback;