smart_proxy_omaha 0.0.1

data/test/omaha/omaha_protocol/request_test.rb

@@ -0,0 +1,77 @@
+require 'test_helper'
+require 'smart_proxy_omaha/omaha_protocol/request'
+
+class RequestTest < Test::Unit::TestCase
+
+  def setup
+    Resolv.any_instance.stubs(:getname).returns('omaha.example.com')
+    @request = Proxy::Omaha::OmahaProtocol::Request.new(
+      xml_fixture('request_update_complete_update'),
+      :ip => '1.1.1.1',
+      :base_url => 'http://www.example.org/',
+    )
+  end
+
+  def test_parses_request
+    assert_equal 'e96281a6-d1af-4bde-9a0a-97b76e56dc57', @request.appid
+    assert_equal '1068', @request.osmajor
+    assert_equal '9.0', @request.osminor
+    assert_equal 'stable', @request.track
+    assert_equal 'amd64-usr', @request.board
+    assert_equal '1010.5.0', @request.alephversion
+    assert_equal '', @request.oemversion
+    assert_equal '', @request.oem
+    assert_equal 'CoreOS', @request.platform
+    assert_equal 3, @request.eventtype
+    assert_equal 2, @request.eventresult
+    assert @request.updatecheck
+  end
+
+  def test_from_coreos
+    assert @request.from_coreos?
+  end
+
+  def test_facts_data
+    expected = {
+      :facts => {
+        :_timestamp => 'now',
+        :_type => :foreman_omaha,
+        :alephversion => '1010.5.0',
+        :appid => 'e96281a6-d1af-4bde-9a0a-97b76e56dc57',
+        :board => 'amd64-usr',
+        :hostname => 'omaha.example.com',
+        :ipaddress => '1.1.1.1',
+        :ipaddress6 => nil,
+        :oem => '',
+        :oemversion => '',
+        :osmajor => '1068',
+        :osminor => '9.0',
+        :platform => 'CoreOS',
+        :track => 'stable',
+        :version =>'1068.9.0'
+      },
+      :name => 'omaha.example.com'
+    }
+    received = @request.facts_data
+    received[:facts][:_timestamp] = 'now'
+    assert_equal expected, received
+  end
+
+  def test_to_status
+    assert_equal :complete, @request.to_status
+  end
+
+  def test_parse_ipaddress
+    assert_equal '1.1.1.1', @request.ipaddress
+    assert_nil @request.ipaddress6
+
+    request6 = Proxy::Omaha::OmahaProtocol::Request.new(
+      xml_fixture('request_update_complete_update'),
+      :ip => '2001:db8::1',
+      :base_url => 'http://www.example.org/',
+    )
+
+    assert_nil request6.ipaddress
+    assert_equal '2001:db8::1', request6.ipaddress6
+  end
+end

data/test/omaha/release_provider_test.rb

@@ -0,0 +1,59 @@
+require 'test_helper'
+require 'smart_proxy_omaha/release'
+require 'smart_proxy_omaha/release_provider'
+
+class ReleaseProviderTest < Test::Unit::TestCase
+  def setup
+    @provider = Proxy::Omaha::ReleaseProvider.new(:track => :stable)
+  end
+
+  def test_releases
+    stub_request(:get, 'https://stable.release.core-os.net/amd64-usr/').
+      to_return(:status => [200, 'OK'], :body => fixture('stable.html'))
+
+    expected = ['367.1.0',
+                '410.0.0',
+                '410.1.0',
+                '410.2.0',
+                '444.4.0',
+                '444.5.0',
+                '494.3.0',
+                '494.4.0',
+                '494.5.0',
+                '522.4.0',
+                '522.5.0',
+                '522.6.0',
+                '557.2.0',
+                '607.0.0',
+                '633.1.0',
+                '647.0.0',
+                '647.2.0',
+                '681.0.0',
+                '681.1.0',
+                '681.2.0',
+                '717.1.0',
+                '717.3.0',
+                '723.3.0',
+                '766.3.0',
+                '766.4.0',
+                '766.5.0',
+                '835.8.0',
+                '835.9.0',
+                '835.10.0',
+                '835.11.0',
+                '835.12.0',
+                '835.13.0',
+                '899.13.0',
+                '899.15.0',
+                '899.17.0',
+                '1010.5.0',
+                '1010.6.0',
+                '1068.6.0',
+                '1068.8.0',
+                '1068.9.0',
+                '1068.10.0',
+                '1122.2.0']
+
+    assert_equal expected.map {|v| Proxy::Omaha::Release.new(:track => @provider.track, :version => v) }, @provider.releases
+  end
+end

data/test/omaha/release_test.rb

@@ -0,0 +1,110 @@
+require 'test_helper'
+require 'fileutils'
+require 'tmpdir'
+require 'smart_proxy_omaha/release'
+
+class ReleaseTest < Test::Unit::TestCase
+
+  def setup
+    @contentpath = Dir.mktmpdir
+    Proxy::Omaha::Plugin.load_test_settings(
+      {
+        :contentpath => @contentpath
+      }
+    )
+    @release = Proxy::Omaha::Release.new(
+      :track => :stable,
+      :version => '1068.9.0',
+    )
+  end
+
+  def teardown
+    FileUtils.rm_rf(@contentpath)
+  end
+
+  def test_path
+    assert_equal "#{@contentpath}/stable/1068.9.0", @release.path
+  end
+
+  def test_exists?
+    assert_equal false, @release.exists?
+    FileUtils.mkdir_p(@release.path)
+    assert_equal true, @release.exists?
+  end
+
+  def test_create
+    @release.expects(:create_path).once.returns(true)
+    @release.expects(:download).once.returns(true)
+    @release.expects(:create_metadata).once.returns(true)
+    assert @release.create
+  end
+
+  def test_create_with_failure
+    @release.expects(:create_path).once.returns(false)
+    @release.expects(:download).never
+    @release.expects(:create_metadata).never
+    refute @release.create
+  end
+
+  def test_create_path
+    assert @release.create_path
+    assert File.directory?(@release.path)
+  end
+
+  def test_compare
+    older = Proxy::Omaha::Release.new(
+      :track => :stable,
+      :version => '100.0.0',
+    )
+    newer = Proxy::Omaha::Release.new(
+      :track => :stable,
+      :version => '2000.0.0',
+    )
+    assert_equal 0, @release.<=>(@release)
+    assert_equal -1, @release.<=>(newer)
+    assert_equal 1, @release.<=>(older)
+  end
+
+  def test_equality
+    other = Proxy::Omaha::Release.new(
+      :track => :stable,
+      :version => '100.0.0',
+    )
+    assert_equal @release, @release
+    refute_equal other, @release
+    refute_equal true, @release
+  end
+
+  def test_to_s
+    assert_equal '1068.9.0', @release.to_s
+  end
+
+  def test_download
+    stub_request(:get, /.*release\.core-os.*/).to_return(:status => [200, 'OK'], :body => "body")
+    expected = ['coreos_production_pxe.vmlinuz', 'coreos_production_pxe_image.cpio.gz', 'update.gz']
+
+    assert_equal true, @release.create_path
+    assert_equal true, @release.download
+
+    expected.each do |f|
+      assert File.exist?(File.join(@release.path, f))
+    end
+  end
+
+  def test_create_metadata
+    file = File.join(@release.path, 'metadata.json')
+
+    digest = mock()
+    digest.expects(:base64digest).twice.returns('foo')
+    File.expects(:size).once.returns(0)
+    Digest::SHA1.expects(:file).once.returns(digest)
+    Digest::SHA256.expects(:file).once.returns(digest)
+
+    expected = '{"release":"1068.9.0","sha1_b64":"foo","sha256_b64":"foo","size":0,"track":"stable"}'
+
+    assert_equal true, @release.create_path
+    assert_equal true, @release.create_metadata
+    assert File.exist?(file)
+    assert_equal expected, File.read(file)
+  end
+end

data/test/omaha/syncer_test.rb

@@ -0,0 +1,41 @@
+require 'test_helper'
+require 'smart_proxy_omaha/syncer'
+
+class SyncerTest < Test::Unit::TestCase
+
+  class FakeRelease
+    def exists?
+      false
+    end
+
+    def valid?
+      true
+    end
+
+    def create; end
+  end
+
+  class FakeReleaseProvider
+    def releases
+      3.times.map { FakeRelease.new }
+    end
+  end
+
+  def setup
+    Proxy::Omaha::Plugin.load_test_settings({:sync_releases => 1})
+    @provider = FakeReleaseProvider.new
+    @syncer = Proxy::Omaha::Syncer.new
+    @syncer.stubs(:release_provider).returns(@provider)
+  end
+
+  def test_sync
+    FakeRelease.any_instance.expects(:create).times(3)
+    @syncer.run
+  end
+
+  def test_sync_existing
+    FakeRelease.any_instance.stubs(:exists?).returns(true)
+    FakeRelease.any_instance.expects(:create).never
+    @syncer.run
+  end
+end

data/test/test_helper.rb

@@ -0,0 +1,18 @@
+require 'test/unit'
+require 'mocha/setup'
+require 'webmock/test_unit'
+require 'mocha/test_unit'
+require 'rack/test'
+
+require 'smart_proxy_for_testing'
+
+# create log directory in our (not smart-proxy) directory
+FileUtils.mkdir_p File.dirname(Proxy::SETTINGS.log_file)
+
+def fixture(name)
+  File.read(File.expand_path("../fixtures/#{name}", __FILE__))
+end
+
+def xml_fixture(name)
+  fixture("#{name}.xml")
+end

metadata

@@ -0,0 +1,167 @@
+--- !ruby/object:Gem::Specification
+name: smart_proxy_omaha
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+platform: ruby
+authors:
+- Timo Goebel
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2016-10-14 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: nokogiri
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: json
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: mocha
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: test-unit
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+description: This plug-in adds support for the Omaha Procotol to Foreman's Smart Proxy.
+email: mail@timogoebel.name
+executables:
+- smart-proxy-omaha-sync
+extensions: []
+extra_rdoc_files:
+- README.md
+- LICENSE
+files:
+- Gemfile
+- LICENSE
+- README.md
+- Rakefile
+- bin/smart-proxy-omaha-sync
+- bundler.d/omaha.rb
+- extra/foreman-proxy-omaha-sync.cron
+- lib/smart_proxy_omaha.rb
+- lib/smart_proxy_omaha/configuration_loader.rb
+- lib/smart_proxy_omaha/dependency_injection.rb
+- lib/smart_proxy_omaha/foreman_client.rb
+- lib/smart_proxy_omaha/http_download.rb
+- lib/smart_proxy_omaha/http_request.rb
+- lib/smart_proxy_omaha/http_shared.rb
+- lib/smart_proxy_omaha/metadata.rb
+- lib/smart_proxy_omaha/metadata_provider.rb
+- lib/smart_proxy_omaha/omaha_api.rb
+- lib/smart_proxy_omaha/omaha_http_config.ru
+- lib/smart_proxy_omaha/omaha_plugin.rb
+- lib/smart_proxy_omaha/omaha_protocol.rb
+- lib/smart_proxy_omaha/omaha_protocol/eventacknowledgeresponse.rb
+- lib/smart_proxy_omaha/omaha_protocol/handler.rb
+- lib/smart_proxy_omaha/omaha_protocol/noupdateresponse.rb
+- lib/smart_proxy_omaha/omaha_protocol/request.rb
+- lib/smart_proxy_omaha/omaha_protocol/response.rb
+- lib/smart_proxy_omaha/omaha_protocol/updateresponse.rb
+- lib/smart_proxy_omaha/release.rb
+- lib/smart_proxy_omaha/release_provider.rb
+- lib/smart_proxy_omaha/release_repository.rb
+- lib/smart_proxy_omaha/syncer.rb
+- lib/smart_proxy_omaha/version.rb
+- settings.d/omaha.yml.example
+- smart_proxy_omaha.gemspec
+- test/fixtures/request_update_complete_error.xml
+- test/fixtures/request_update_complete_noupdate.xml
+- test/fixtures/request_update_complete_update.xml
+- test/fixtures/request_update_download_started.xml
+- test/fixtures/response_update_complete_error.xml
+- test/fixtures/response_update_complete_noupdate.xml
+- test/fixtures/response_update_complete_update.xml
+- test/fixtures/stable.html
+- test/omaha/http_download_test.rb
+- test/omaha/http_request_test.rb
+- test/omaha/metadata_provider_test.rb
+- test/omaha/omaha_api_test.rb
+- test/omaha/omaha_protocol/request_test.rb
+- test/omaha/release_provider_test.rb
+- test/omaha/release_test.rb
+- test/omaha/syncer_test.rb
+- test/test_helper.rb
+homepage: http://github.com/theforeman/smart_proxy_omaha
+licenses:
+- GPLv3
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.4.5
+signing_key: 
+specification_version: 4
+summary: Omaha protocol support for smart-proxy
+test_files: []
+has_rdoc: