fix rubocop and spec tests

This commit is contained in:
Samuel Beaulieu 2021-12-29 12:02:55 -06:00
parent daa55fe5b8
commit a4c730df7b
No known key found for this signature in database
GPG key ID: 12030F74136D0F34
5 changed files with 49 additions and 31 deletions

View file

@ -36,16 +36,18 @@ Style/SwapValues: # (new in 1.1)
#disabled
Metrics/AbcSize:
Max: 77
Enabled: false
Metrics/ClassLength:
Max: 430
Enabled: false
Metrics/CyclomaticComplexity:
Max: 14
Enabled: false
Metrics/MethodLength:
Max: 48
Enabled: false
Metrics/PerceivedComplexity:
Max: 14
Enabled: false
Metrics/ParameterLists:
Max: 6
Enabled: false
Layout/LineLength:
Max: 220
Enabled: false
Metrics/BlockLength:
Enabled: false

View file

@ -2,7 +2,7 @@
require 'googleauth'
require 'google/apis/compute_v1'
require "google/cloud/dns"
require 'google/cloud/dns'
require 'bigdecimal'
require 'bigdecimal/util'
require 'vmpooler/providers/base'
@ -45,7 +45,6 @@ module Vmpooler
{ connection: new_conn }
end
@redis = redis_connection_pool
@dns = Google::Cloud::Dns.new(project_id: project)
end
# name of the provider class
@ -59,6 +58,10 @@ module Vmpooler
end
end
def dns
@dns ||= Google::Cloud::Dns.new(project_id: project)
end
# main configuration options
def project
provider_config['project']
@ -176,7 +179,7 @@ module Vmpooler
network_interfaces = Google::Apis::ComputeV1::NetworkInterface.new(
network: network_name
)
network_interfaces.subnetwork=subnetwork_name(pool_name) if subnetwork_name(pool_name)
network_interfaces.subnetwork = subnetwork_name(pool_name) if subnetwork_name(pool_name)
init_params = {
source_image: pool['template'], # The source image to create this disk.
labels: { 'vm' => new_vmname, 'pool' => pool_name },
@ -195,15 +198,16 @@ module Vmpooler
network_interfaces: [network_interfaces],
labels: { 'vm' => new_vmname, 'pool' => pool_name, project => nil }
)
=begin TODO: Maybe this will be needed to set the hostname (usually internal DNS name but in opur case for some reason its nil)
given_hostname = "#{new_vmname}.#{dns_zone}"
client.hostname = given_hostname if given_hostname
=end
# TODO: Maybe this will be needed to set the hostname (usually internal DNS name but in opur case for some reason its nil)
# given_hostname = "#{new_vmname}.#{dns_zone}"
# client.hostname = given_hostname if given_hostname
debug_logger('trigger insert_instance')
result = connection.insert_instance(project, zone(pool_name), client)
wait_for_operation(project, pool_name, result)
created_instance = get_vm(pool_name, new_vmname)
dns_setup(created_instance)
created_instance
end
# create_disk creates an additional disk for an existing VM. It will name the new
@ -554,25 +558,25 @@ module Vmpooler
# END BASE METHODS
def dns_setup(created_instance)
zone = @dns.zone dns_zone_resource_name if dns_zone_resource_name
if zone && created_instance
zone = dns.zone dns_zone_resource_name if dns_zone_resource_name
return unless zone && created_instance
name = created_instance['name']
change = zone.add name, "A", 60, [created_instance['ip']]
change = zone.add name, 'A', 60, [created_instance['ip']]
debug_logger("#{change.id} - #{change.started_at} - #{change.status}") if change
end
# TODO: should we catch Google::Cloud::AlreadyExistsError that is thrown when it already exist?
# and then delete and recreate?
# eg the error is Google::Cloud::AlreadyExistsError: alreadyExists: The resource 'entity.change.additions[0]' named 'instance-8.test.vmpooler.puppet.net. (A)' already exists
end
def dns_teardown(created_instance)
zone = @dns.zone dns_zone_resource_name if dns_zone_resource_name
if zone && created_instance
zone = dns.zone dns_zone_resource_name if dns_zone_resource_name
return unless zone && created_instance
name = created_instance['name']
change = zone.remove name, "A"
change = zone.remove name, 'A'
debug_logger("#{change.id} - #{change.started_at} - #{change.status}") if change
end
end
def should_be_ignored(item, allowlist)
return false if allowlist.nil?
@ -652,7 +656,7 @@ module Vmpooler
'machine_type' => vm_object.machine_type,
'labels' => vm_object.labels,
'label_fingerprint' => vm_object.label_fingerprint,
'ip' => vm_object.network_interfaces.first.network_ip
'ip' => vm_object.network_interfaces ? vm_object.network_interfaces.first.network_ip : nil
# 'powerstate' => powerstate
}
end

View file

@ -0,0 +1,9 @@
MockDNS = Struct.new(
# https://rubydoc.info/gems/google-cloud-dns/0.35.1/Google/Cloud/Dns
:change, :credentials, :project, :record, :zone,
keyword_init: true
) do
def zone(zone)
self.zone = zone
end
end

View file

@ -1,15 +1,18 @@
# frozen_string_literal: true
require 'simplecov'
=begin
SimpleCov.start do
add_filter '/spec/'
end
=end
require 'helpers'
require 'rspec'
require 'vmpooler'
require 'redis'
require 'vmpooler/metrics'
require 'computeservice_helper'
require 'dnsservice_helper'
def project_root_dir
File.dirname(File.dirname(__FILE__))

View file

@ -26,8 +26,6 @@ describe 'Vmpooler::PoolManager::Provider::Gce' do
zone: '#{zone}'
network_name: global/networks/default
# network_name: 'projects/itsysopsnetworking/global/networks/shared1'
dns_zone_resource_name: 'example-com'
dns_zone: 'example.com'
:pools:
- name: '#{poolname}'
alias: [ 'mockpool' ]
@ -56,6 +54,8 @@ EOT
subject { Vmpooler::PoolManager::Provider::Gce.new(config, logger, metrics, redis_connection_pool, 'gce', provider_options) }
before(:each) { allow(subject).to receive(:dns).and_return(MockDNS.new()) }
describe '#name' do
it 'should be gce' do
expect(subject.name).to eq('gce')
@ -89,7 +89,7 @@ EOT
end
context 'in itsysops' do
let(:vmname) { "instance-10" }
let(:vmname) { "instance-15" }
let(:project) { 'vmpooler-test' }
let(:config) { YAML.load(<<-EOT
---