hammer_cli_foreman 3.0.0 → 3.3.0

data/test/functional/compute_attribute_test.rb

@@ -11,11 +11,11 @@ describe "parameters" do
           }
     end
 
-    it "should print error on missing --compute-profile-id or --compute-profile" do
+    it 'should print error on missing --compute-profile-id or --compute-profile' do
       expected_result = common_error_result(
         @cmd,
-        "Could not find compute_profile, please set one of options --compute-profile, --compute-profile-id.",
-        "Could not set the compute profile attributes"
+        'Could not find compute_profile, please set one of options --compute-profile-id, --compute-profile.',
+        'Could not set the compute profile attributes'
       )
 
       api_expects_no_call
@@ -23,12 +23,12 @@ describe "parameters" do
       assert_cmd(expected_result, result)
     end
 
-    it "should print error on missing --compute-resource-id or --compute-resource" do
+    it 'should print error on missing --compute-resource-id or --compute-resource' do
       params = ['--compute-profile-id=1']
       expected_result = common_error_result(
         @cmd,
-        "Could not find compute_resource, please set one of options --compute-resource, --compute-resource-id.",
-        "Could not set the compute profile attributes"
+        'Could not find compute_resource, please set one of options --compute-resource-id, --compute-resource.',
+        'Could not set the compute profile attributes'
       )
 
       api_expects_no_call
@@ -103,11 +103,11 @@ describe "parameters" do
       params = ['--compute-profile-id=1', '--compute-resource-id=1', '--compute-attributes', 'cores=1']
 
       api_expects(:compute_profiles, :show) do |par|
-        par['id'] == 1
+        par['id'] == '1'
       end.returns(@compute_profile)
 
       api_expects(:compute_resources, :show) do |par|
-        par['id'] == 1
+        par['id'] == '1'
       end.returns(@compute_resource)
 
       api_expects(:compute_attributes, :update) do |par|
@@ -177,7 +177,7 @@ describe "parameters" do
       expected_result.expected_exit_code = HammerCLI::EX_USAGE
 
       api_expects(:compute_profiles, :show) do |par|
-        par['id'] == 1
+        par['id'] == '1'
       end.returns(@compute_profile)
 
       result = run_cmd(@cmd + params)
@@ -189,7 +189,7 @@ describe "parameters" do
       params = ['--compute-profile-id=1', '--compute-resource-id=1', '--volume', 'size_gb=1']
 
       api_expects(:compute_profiles, :show) do |par|
-        par['id'] == 1
+        par['id'] == '1'
       end.returns(@compute_profile)
 
       api_expects(:compute_attributes, :update) do |par|
@@ -261,7 +261,7 @@ describe "parameters" do
         params = ['--compute-profile-id=1', '--compute-resource-id=1', '--volume-id=1', '--volume', 'size_gb=1']
 
         api_expects(:compute_profiles, :show) do |par|
-          par['id'] == 1
+          par['id'] == '1'
         end.returns(@compute_profile)
 
         api_expects(:compute_attributes, :update) do |par|
@@ -339,7 +339,7 @@ describe "parameters" do
       params = ['--compute-profile-id=1', '--compute-resource-id=1', '--volume-id=1']
 
       api_expects(:compute_profiles, :show) do |par|
-        par['id'] == 1
+        par['id'] == '1'
       end.returns(@compute_profile)
 
       api_expects(:compute_attributes, :update) do |par|
@@ -438,7 +438,7 @@ describe "parameters" do
       expected_result.expected_exit_code = HammerCLI::EX_USAGE
 
       api_expects(:compute_profiles, :show) do |par|
-        par['id'] == 1
+        par['id'] == '1'
       end.returns(@compute_profile)
       result = run_cmd(@cmd + params)
       assert_cmd(expected_result, result)
@@ -455,7 +455,7 @@ describe "parameters" do
       response = HammerCLIForeman.foreman_api.api.send(:create_fake_response, 404,
                                                        expected_message, "GET", "http://example.com/", {})
       api_expects(:compute_profiles, :show) do |par|
-        par['id'] == 200
+        par['id'] == '200'
       end.raises(RestClient::NotFound, response)
 
       result = run_cmd(@cmd + params)
@@ -465,11 +465,11 @@ describe "parameters" do
     it "should add interface" do
       params = [ '--interface', 'name=eth0', '--compute-profile-id=1', '--compute-resource-id=1']
       api_expects(:compute_profiles, :show) do |par|
-        par['id'] == 1
+        par['id'] == '1'
       end.returns(@compute_profile)
 
       api_expects(:compute_resources, :show) do |par|
-        par['id'] == 1
+        par['id'] == '1'
       end.returns(@compute_resource)
 
       api_expects(:compute_attributes, :update) do |par|
@@ -548,11 +548,11 @@ describe "parameters" do
       params = ['--compute-profile-id=1', '--compute-resource-id=1', '--interface-id=1', '--interface', 'compute_name=eth0']
 
       api_expects(:compute_profiles, :show) do |par|
-        par['id'] == 1
+        par['id'] == '1'
       end.returns(@compute_profile)
 
       api_expects(:compute_resources, :show) do |par|
-        par['id'] == 1
+        par['id'] == '1'
       end.returns(@compute_resource)
 
       api_expects(:compute_attributes, :update) do |par|
@@ -637,11 +637,11 @@ describe "parameters" do
       params = ['--compute-profile-id=1', '--compute-resource-id=1', '--interface-id=1']
 
       api_expects(:compute_profiles, :show) do |par|
-        par['id'] == 1
+        par['id'] == '1'
       end.returns(@compute_profile)
 
       api_expects(:compute_resources, :show) do |par|
-        par['id'] == 1
+        par['id'] == '1'
       end.returns(@compute_resource)
 
       api_expects(:compute_attributes, :update) do |par|

data/test/functional/domain/create_test.rb

@@ -0,0 +1,91 @@
+require_relative '../test_helper'
+
+describe 'Domain' do
+  describe 'CreateCommand' do
+    let(:cmd) { %w[domain create] }
+    let(:minimal_params) { %w[--name=dom1.com] }
+
+    def domain_params(additional_params = {})
+      params = {
+        domain: {
+          name: 'dom1.com'
+        }
+      }
+      params[:domain].merge!(additional_params)
+      params
+    end
+
+    it 'should print error on missing --name' do
+      expected_result = missing_args_error_result(cmd, '--name')
+
+      api_expects_no_call
+
+      result = run_cmd(cmd)
+      assert_cmd(expected_result, result)
+    end
+
+    it 'allows minimal options' do
+      api_expects(:domains, :create).with_params(domain_params)
+
+      run_cmd(cmd + minimal_params)
+    end
+
+    it 'allows description' do
+      params = %w[--description=shortdesc]
+      api_expects(:domains, :create).with_params(domain_params(fullname: 'shortdesc'))
+
+      run_cmd(cmd + minimal_params + params)
+    end
+
+    it 'allows dns id' do
+      params = %w[--dns-id=1]
+      api_expects(:domains, :create).with_params(domain_params(dns_id: 1))
+
+      run_cmd(cmd + minimal_params + params)
+    end
+
+    it 'allows dns name' do
+      params = %w[--dns=sp1]
+      api_expects_search(:smart_proxies, { name: 'sp1' }).returns(
+        index_response([{ 'id' => 1 }])
+      )
+      api_expects(:domains, :create).with_params(domain_params(dns_id: 1))
+
+      run_cmd(cmd + minimal_params + params)
+    end
+
+    it 'allows location ids' do
+      params = %w[--location-ids=1,4]
+      api_expects(:domains, :create).with_params(domain_params(location_ids: %w[1 4]))
+
+      run_cmd(cmd + minimal_params + params)
+    end
+
+    it 'allows location names' do
+      params = %w[--locations=loc1,loc2]
+      api_expects(:locations, :index) do |p|
+        p[:search] == 'name = "loc1" or name = "loc2"'
+      end.returns(index_response([{ 'id' => 1 }, { 'id' => 2 }]))
+      api_expects(:domains, :create).with_params(domain_params(location_ids: [1, 2]))
+
+      run_cmd(cmd + minimal_params + params)
+    end
+
+    it 'allows organization ids' do
+      params = %w[--organization-ids=1,4]
+      api_expects(:domains, :create).with_params(domain_params(organization_ids: %w[1 4]))
+
+      run_cmd(cmd + minimal_params + params)
+    end
+
+    it 'allows organization names' do
+      params = %w[--organizations=org1,org2]
+      api_expects(:organizations, :index) do |p|
+        p[:search] == 'name = "org1" or name = "org2"'
+      end.returns(index_response([{ 'id' => 1 }, { 'id' => 2 }]))
+      api_expects(:domains, :create).with_params(domain_params(organization_ids: [1, 2]))
+
+      run_cmd(cmd + minimal_params + params)
+    end
+  end
+end

data/test/functional/domain/update_test.rb

@@ -0,0 +1,90 @@
+require_relative '../test_helper'
+
+describe 'Domain' do
+  describe 'UpdateCommand' do
+    let(:cmd) { %w[domain update] }
+    let(:minimal_params) { %w[--id=1] }
+
+    def domain_params(additional_params = {})
+      params = {
+        id: '1',
+        domain: {}
+      }
+      params[:domain].merge!(additional_params)
+      params
+    end
+
+    it 'should print error on missing --id' do
+      expected_result = missing_args_error_result(cmd, '--id')
+
+      api_expects_no_call
+
+      result = run_cmd(cmd)
+      assert_cmd(expected_result, result)
+    end
+
+    it 'allows minimal options' do
+      api_expects(:domains, :update).with_params(domain_params)
+
+      run_cmd(cmd + minimal_params)
+    end
+
+    it 'allows description' do
+      params = %w[--description=shortdesc]
+      api_expects(:domains, :update).with_params(domain_params(fullname: 'shortdesc'))
+
+      run_cmd(cmd + minimal_params + params)
+    end
+
+    it 'allows dns id' do
+      params = %w[--dns-id=1]
+      api_expects(:domains, :update).with_params(domain_params(dns_id: 1))
+
+      run_cmd(cmd + minimal_params + params)
+    end
+
+    it 'allows dns name' do
+      params = %w[--dns=sp1]
+      api_expects_search(:smart_proxies, { name: 'sp1' }).returns(
+        index_response([{ 'id' => 1 }])
+      )
+      api_expects(:domains, :update).with_params(domain_params(dns_id: 1))
+
+      run_cmd(cmd + minimal_params + params)
+    end
+
+    it 'allows location ids' do
+      params = %w[--location-ids=1,4]
+      api_expects(:domains, :update).with_params(domain_params(location_ids: %w[1 4]))
+
+      run_cmd(cmd + minimal_params + params)
+    end
+
+    it 'allows location names' do
+      params = %w[--locations=loc1,loc2]
+      api_expects(:locations, :index) do |p|
+        p[:search] == 'name = "loc1" or name = "loc2"'
+      end.returns(index_response([{ 'id' => 1 }, { 'id' => 2 }]))
+      api_expects(:domains, :update).with_params(domain_params(location_ids: [1, 2]))
+
+      run_cmd(cmd + minimal_params + params)
+    end
+
+    it 'allows organization ids' do
+      params = %w[--organization-ids=1,4]
+      api_expects(:domains, :update).with_params(domain_params(organization_ids: %w[1 4]))
+
+      run_cmd(cmd + minimal_params + params)
+    end
+
+    it 'allows organization names' do
+      params = %w[--organizations=org1,org2]
+      api_expects(:organizations, :index) do |p|
+        p[:search] == 'name = "org1" or name = "org2"'
+      end.returns(index_response([{ 'id' => 1 }, { 'id' => 2 }]))
+      api_expects(:domains, :update).with_params(domain_params(organization_ids: [1, 2]))
+
+      run_cmd(cmd + minimal_params + params)
+    end
+  end
+end

data/test/functional/host_test.rb

@@ -568,7 +568,41 @@ describe 'disassociate host from vm' do
   end
 end
 
+describe 'defaults' do
+  let(:cmd) { %w[host list] }
+  let(:defaults) do
+    HammerCLI::Defaults.new(
+      {
+        organization_id: {
+          value: '1'
+        },
+        organization: {
+          value: 'Default Organization'
+        }
+      }
+    )
+  end
+
+  it 'works with default org name overridden via cli' do
+    params = %w[--organization=Org2]
 
+    api_expects_search(:organizations, name: 'Org2').returns(index_response([{ 'id' => '2' }]))
+    api_expects(:hosts, :index) do |p|
+      p['organization_id'] == '2'
+    end
+
+    run_cmd(cmd + params, { use_defaults: true, defaults: defaults })
+  end
+  it 'works with default org id overridden via cli' do
+    params = %w[--organization-id=2]
+
+    api_expects(:hosts, :index) do |p|
+      p['organization_id'] == '2'
+    end
+
+    run_cmd(cmd + params, { use_defaults: true, defaults: defaults })
+  end
+end
 
 describe 'list' do
   before do

data/test/functional/hostgroup/create_test.rb

@@ -165,6 +165,17 @@ module HammerCLIForeman
         run_cmd(%w(hostgroup create --name hg1 --parent parent_hg))
       end
 
+      it 'allows parent hostgroup title' do
+        api_expects(:hostgroups, :index) do |p|
+          p[:search] = 'title = "parent_hg"'
+        end.returns(index_response([{ 'id' => 1 }]))
+        api_expects(:hostgroups, :create) do |p|
+          p['hostgroup']['parent_id'] == 1 &&
+            p['hostgroup']['name'] == 'hg1'
+        end
+        run_cmd(%w[hostgroup create --name hg1 --parent parent_hg])
+      end
+
       it 'allows partition table id' do
         api_expects(:hostgroups, :create) do |p|
           p['hostgroup']['ptable_id'] == 1 &&

data/test/functional/hostgroup/update_test.rb

@@ -163,6 +163,17 @@ module HammerCLIForeman
         run_cmd(%w(hostgroup update --id 1 --parent parent_hg))
       end
 
+      it 'allows parent hostgroup title' do
+        api_expects(:hostgroups, :index) do |p|
+          p[:search] = 'title = "parent_hg"'
+        end.returns(index_response([{ 'id' => 1 }]))
+        api_expects(:hostgroups, :update) do |p|
+          p['hostgroup']['parent_id'] == 1 &&
+            p['id'] == '1'
+        end
+        run_cmd(%w[hostgroup update --id 1 --parent parent_hg])
+      end
+
       it 'allows partition table id' do
         api_expects(:hostgroups, :update) do |p|
           p['hostgroup']['ptable_id'] == 1 &&

data/test/functional/partition_table_test.rb

@@ -0,0 +1,63 @@
+require File.join(File.dirname(__FILE__), 'test_helper')
+
+describe 'partition-table' do
+  describe 'import' do
+    let(:template) do
+      {
+        'id' => 1,
+        'template' => 'Template content'
+      }
+    end
+    let(:cmd) { %w(partition-table import) }
+    let(:tempfile) { Tempfile.new('template') }
+
+    it 'requires --name and --file' do
+      params = ['--name=test']
+      api_expects_no_call
+      expected_result = usage_error_result(
+        cmd,
+        'Options --name, --file are required.',
+        'Could not import partition table template')
+      result = run_cmd(cmd + params)
+      assert_cmd(expected_result, result)
+    end
+
+    it 'import template' do
+      params = ['--name=test', "--file=#{tempfile.path}"]
+      tempfile.write('Template content')
+      tempfile.rewind
+      api_expects(:ptables, :import, 'Import partition table template').with_params(
+        'ptable' => {
+          'name' => 'test',
+          'template' => 'Template content'
+        }).returns(template)
+
+      result = run_cmd(cmd + params)
+      assert_cmd(success_result("Import partition table template succeeded.\n"), result)
+    end
+  end
+
+  describe 'export' do
+    let(:cmd) { %w(partition-table export) }
+    let(:tempfile) { Tempfile.new('template', '/tmp') }
+    let(:params) { ['--id=1', '--path=/tmp'] }
+    let(:template_response) do
+      response = mock('TemplateResponse')
+      response.stubs(:code).returns(200)
+      response.stubs(:body).returns('Template content')
+      response.stubs(:headers).returns({:content_disposition => "filename=\"#{File.basename(tempfile.path)}\""})
+      response
+    end
+
+    it 'download template' do
+      api_expects(:ptables, :export, 'Export partition table template').with_params(
+        'id' => '1').returns(template_response)
+
+      output = OutputMatcher.new("The partition table template has been saved to #{tempfile.path}")
+      expected_result = success_result(output)
+      result = run_cmd(cmd + params)
+      assert_cmd(expected_result, result)
+      assert_equal('Template content', tempfile.read)
+    end
+  end
+end

data/test/functional/personal_access_token_test.rb

@@ -37,7 +37,7 @@ describe 'personal_access_token' do
 
     it 'lists all access tokens for a given user' do
       api_expects(:personal_access_tokens, :index, 'List').with_params(
-        'user_id' => 1, 'page' => 1, 'per_page' => 1000
+        'user_id' => '1', 'page' => 1, 'per_page' => 1000
       ).returns(index_response([access_token, active_access_token]))
 
       output = IndexMatcher.new([
@@ -67,7 +67,7 @@ describe 'personal_access_token' do
 
     it 'creates an access token to a given user' do
       api_expects(:personal_access_tokens, :create).with_params(
-        'user_id' => 1, 'personal_access_token' => {
+        'user_id' => '1', 'personal_access_token' => {
           'expires_at' => '01/01/2048', 'name' => 'test'
         }
       ).returns(access_token)
@@ -85,7 +85,7 @@ describe 'personal_access_token' do
 
     it 'shows the personal access token' do
       api_expects(:personal_access_tokens, :show, 'Show PAT').with_params(
-        'id' => '1', 'user_id' => 1
+        'id' => '1', 'user_id' => '1'
       ).returns(access_token)
 
       output = OutputMatcher.new([
@@ -109,7 +109,7 @@ describe 'personal_access_token' do
     let(:params) { ['--id=1', '--user-id=1'] }
     it 'deletes an access token to a given user' do
       api_expects(:personal_access_tokens, :destroy, 'Revoke PAT').with_params(
-        'id' => '1', 'user_id' => 1
+        'id' => '1', 'user_id' => '1'
       ).returns(access_token)
 
       expected_result = success_result(

data/test/functional/report_template_test.rb

@@ -384,6 +384,30 @@ describe 'report-template' do
     end
   end
 
+  describe 'export' do
+    let(:cmd) { %w(report-template export) }
+    let(:tempfile) { Tempfile.new('template', '/tmp') }
+    let(:params) { ['--id=1', '--path=/tmp'] }
+    let(:template_response) do
+      response = mock('TemplateResponse')
+      response.stubs(:code).returns(200)
+      response.stubs(:body).returns('Template content')
+      response.stubs(:headers).returns({:content_disposition => "filename=\"#{File.basename(tempfile.path)}\""})
+      response
+    end
+
+    it 'download template' do
+      api_expects(:report_templates, :export, 'Export report template').with_params(
+        'id' => '1').returns(template_response)
+
+      output = OutputMatcher.new("The report template has been saved to #{tempfile.path}")
+      expected_result = success_result(output)
+      result = run_cmd(cmd + params)
+      assert_cmd(expected_result, result)
+      assert_equal('Template content', tempfile.read)
+    end
+  end
+
   describe 'report-data' do
     let(:cmd) { %w(report-template report-data) }
     let(:tempfile) { Tempfile.new('template', '/tmp') }