[#1307] added check for account plan

Author: lidiamokevnina

lib/uffizzi/cli/install.rb

@@ -20,9 +20,10 @@ class Cli::Install < Thor
     method_option :'node-selector-template', required: false, type: :string
     def controller(hostname)
       Uffizzi::AuthHelper.check_login
+      check_account_can_install
 
-      # InstallService.kubectl_exists?
-      # InstallService.helm_exists?
+      InstallService.kubectl_exists?
+      InstallService.helm_exists?
 
       if options[:context].present? && options[:context] != InstallService.kubeconfig_current_context
         InstallService.set_current_context(options[:context])
@@ -182,7 +183,18 @@ def update_controller_settings(controller_setting_id, params)
 
     def create_controller_settings(params)
       response = create_account_controller_settings(server, account_id, params)
-      Uffizzi::ResponseHelper.handle_failed_response(response) unless Uffizzi::ResponseHelper.created?(response)
+      unless Uffizzi::ResponseHelper.created?(response)
+        Uffizzi::ResponseHelper.handle_failed_response(response)
+        raise Uffizzi::Error.new
+      end
+    end
+
+    def check_account_can_install
+      response = check_can_install(server, account_id)
+      unless Uffizzi::ResponseHelper.ok?(response)
+        Uffizzi::ResponseHelper.handle_failed_response(response)
+        raise Uffizzi::Error.new
+      end
     end
 
     def build_controller_setting_params(uri, installation_options)

lib/uffizzi/clients/api/api_client.rb

@@ -44,6 +44,14 @@ def fetch_accounts(server)
     build_response(response)
   end
 
+  def check_can_install(server, account_id)
+    uri = account_can_install_uri(server, account_id)
+
+    response = http_client.make_get_request(uri)
+
+    build_response(response)
+  end
+
   def fetch_projects(server)
     uri = projects_uri(server)
     response = http_client.make_get_request(uri)

lib/uffizzi/clients/api/api_routes.rb

@@ -11,6 +11,10 @@ def account_uri(server, account_name)
     "#{server}/api/cli/v1/accounts/#{account_name}"
   end
 
+  def account_can_install_uri(server, account_id)
+    "#{server}/api/cli/v1/accounts/#{account_id}/check_can_install"
+  end
+
   def compose_file_uri(server, project_slug)
     "#{server}/api/cli/v1/projects/#{project_slug}/compose_file"
   end

test/support/uffizzi_stub_support.rb

@@ -44,6 +44,12 @@ def stub_uffizzi_accounts_success(body)
     stub_request(:get, url).to_return(status: 200, body: body.to_json)
   end
 
+  def stub_uffizzi_check_can_install_success(account_id)
+    url = account_can_install_uri(Uffizzi.configuration.server, account_id)
+
+    stub_request(:get, url).to_return(status: 200)
+  end
+
   def stub_uffizzi_account_success(body, account_name)
     url = account_uri(Uffizzi.configuration.server, account_name)
 

test/uffizzi/cli/install_test.rb

@@ -62,6 +62,7 @@ def test_install
     stub_get_account_controller_settings_request(empty_controller_settings_body, account_id)
     stub_create_account_controller_settings_request({}, account_id)
     stub_update_account_success(account_body, account_name)
+    stub_uffizzi_check_can_install_success(account_id)
 
     @install.options = command_options(email: '[email protected]')
     @install.controller(host)
@@ -116,6 +117,7 @@ def test_install_if_settgins_exists
     stub_get_account_controller_settings_request(account_controller_settings_body, account_id)
     stub_update_account_controller_settings_request(account_controller_settings_body, account_id,
                                                     account_controller_settings_body[:controller_settings][0][:id])
+    stub_uffizzi_check_can_install_success(account_id)
 
     @install.options = command_options(email: '[email protected]')
     @install.controller(host)