Change error handling to more humane

app/controllers/api/v1/registrant/companies_controller.rb

@@ -16,9 +16,9 @@ def index
         end
 
         def current_user_companies
-          current_registrant_user.companies
+          [:ok, current_registrant_user.companies]
         rescue CompanyRegister::NotAvailableError
-          []
+          [:service_unavailable, []]
         end
 
         def limit
@@ -35,8 +35,8 @@ def error_result(attr_name)
         end
 
         def companies_result(limit, offset)
-          @companies = current_user_companies.drop(offset).first(limit)
-          status = @companies.present? ? :ok : :not_found
+          status, all_companies = current_user_companies
+          @companies = all_companies.drop(offset).first(limit)
 
           serialized_companies = @companies.map do |item|
             country_code = current_registrant_user.country.alpha3

test/integration/api/registrant/registrant_api_companies_test.rb

@@ -1,3 +1,4 @@
+$VERBOSE=nil
 require 'test_helper'
 require 'auth_token/auth_token_creator'
 
@@ -10,6 +11,11 @@ def setup
     @auth_headers = { 'HTTP_AUTHORIZATION' => auth_token }
   end
 
+  def teardown
+
+    CompanyRegister.const_set(:Client, CompanyRegisterClientStub)
+  end
+
   def test_accepts_limit_and_offset_parameters
     contacts(:william).update!(ident: '1234', ident_type: 'priv', ident_country_code: 'US')
 
@@ -33,6 +39,18 @@ def test_format
     assert_equal(:companies, response_json.keys.first)
   end
 
+  def test_status_if_nil_result
+    contacts(:john).update!(ident: '12344321', ident_type: 'priv', ident_country_code: 'US')
+
+    CompanyRegister.const_set(:Client, CompanyRegisterClientZeroStub)
+
+    get '/api/v1/registrant/companies', headers: @auth_headers
+    response_json = JSON.parse(response.body, symbolize_names: true)
+    assert_equal(1, response_json.count)
+    assert_equal(200, response.status)
+    assert_equal(:companies, response_json.keys.first)
+  end
+
   private
 
   def auth_token
@@ -41,3 +59,12 @@ def auth_token
     "Bearer #{hash[:access_token]}"
   end
 end
+
+class CompanyRegisterClientZeroStub
+  Company = Struct.new(:registration_number, :company_name)
+
+  def representation_rights(citizen_personal_code:, citizen_country_code:)
+    []
+  end
+end
+