require 'test_helper'

module Shipit
  class DeploySpecTest < ActiveSupport::TestCase
    setup do
      @app_dir = '/tmp/'
      @spec = DeploySpec::FileSystem.new(@app_dir, 'env')
      @spec.stubs(:load_config).returns({})
    end

    test '#supports_fetch_deployed_revision? returns false by default' do
      refute @spec.supports_fetch_deployed_revision?
    end

    test '#supports_fetch_deployed_revision? returns true if steps are defined' do
      @spec.stubs(:load_config).returns('fetch' => ['curl --silent https://example.com/status/version'])
      assert @spec.supports_fetch_deployed_revision?
    end

    test '#supports_rollback? returns false by default' do
      refute @spec.supports_rollback?
    end

    test '#supports_rollback? returns true if steps are defined' do
      @spec.stubs(:load_config).returns('rollback' => {'override' => ['rm -rf /usr /lib/nvidia-current/xorg/xorg']})
      assert @spec.supports_rollback?
    end

    test '#supports_rollback? returns true if stack is detected as capistrano' do
      @spec.expects(:capistrano?).returns(true)
      assert @spec.supports_rollback?
    end

    test '#dependencies_steps returns `dependencies.override` if present' do
      @spec.stubs(:load_config).returns('dependencies' => {'override' => %w(foo bar baz)})
      assert_equal %w(foo bar baz), @spec.dependencies_steps
    end

    test '#dependencies_steps returns `bundle install` if a `Gemfile` is present' do
      @spec.expects(:bundler?).returns(true).at_least_once
      @spec.expects(:bundle_install).returns(['bundle install'])
      assert_equal ['bundle install'], @spec.dependencies_steps
    end

    test "#dependencies_steps prepend and append pre and post steps" do
      @spec.stubs(:load_config).returns('dependencies' => {'pre' => ['before'], 'post' => ['after']})
      @spec.expects(:bundler?).returns(true).at_least_once
      @spec.expects(:bundle_install).returns(['bundle install'])
      assert_equal ['before', 'bundle install', 'after'], @spec.dependencies_steps
    end

    test '#fetch_deployed_revision_steps! is unknown by default' do
      assert_raises DeploySpec::Error do
        @spec.fetch_deployed_revision_steps!
      end
    end

    test '#fetch_deployed_revision_steps use `fetch` is present' do
      @spec.stubs(:load_config).returns('fetch' => ['echo l33t'])
      assert_equal ['echo l33t'], @spec.fetch_deployed_revision_steps
    end

    test '#bundle_install return a sane default bundle install command' do
      @spec.stubs(:gemfile_lock_exists?).returns(true)
      command = %(
        bundle check --path=#{DeploySpec.bundle_path} ||
        bundle install
        --frozen
        --path=#{DeploySpec.bundle_path}
        --retry=2
        --without=default:production:development:test:staging:benchmark:debug
      ).gsub(/\s+/, ' ').strip
      assert_equal command, @spec.bundle_install.last
    end

    test '#bundle_install use `dependencies.bundler.without` if present to build the --without argument' do
      @spec.stubs(:gemfile_lock_exists?).returns(true)
      @spec.stubs(:load_config).returns('dependencies' => {'bundler' => {'without' => %w(some custom groups)}})
      command = %(
        bundle check --path=#{DeploySpec.bundle_path} ||
        bundle install
        --frozen
        --path=#{DeploySpec.bundle_path}
        --retry=2
        --without=some:custom:groups
      ).gsub(/\s+/, ' ').strip
      assert_equal command, @spec.bundle_install.last
    end

    test '#bundle_install has --frozen option if Gemfile.lock is present' do
      @spec.stubs(:load_config).returns('dependencies' => {'bundler' => {'without' => %w(some custom groups)}})
      @spec.stubs(:gemfile_lock_exists?).returns(true)
      assert @spec.bundle_install.last.include?('--frozen')
    end

    test '#bundle_install does not have --frozen option if Gemfile.lock is not present' do
      @spec.stubs(:load_config).returns('dependencies' => {'bundler' => {'without' => %w(some custom groups)}})
      @spec.stubs(:gemfile_lock_exists?).returns(false)
      refute @spec.bundle_install.last.include?('--frozen')
    end

    test '#bundle_install does not have --frozen if overridden in shipit.yml' do
      @spec.stubs(:load_config).returns('dependencies' => {'bundler' => {'frozen' => false}})
      @spec.stubs(:gemfile_lock_exists?).returns(true)
      refute @spec.bundle_install.last.include?('--frozen')
    end

    test '#deploy_steps returns `deploy.override` if present' do
      @spec.stubs(:load_config).returns('deploy' => {'override' => %w(foo bar baz)})
      assert_equal %w(foo bar baz), @spec.deploy_steps
    end

    test '#deploy_steps returns `cap $ENVIRONMENT deploy` if a `Capfile` is present' do
      @spec.expects(:bundler?).returns(true).at_least_once
      @spec.expects(:capistrano?).returns(true)
      assert_equal ['bundle exec cap $ENVIRONMENT deploy'], @spec.deploy_steps
    end

    test "#deploy_steps returns `kubernetes-deploy <namespace> <context>` if `kubernetes` is present" do
      @spec.stubs(:load_config).returns(
        'kubernetes' => {
          'namespace' => 'foo',
          'context' => 'bar',
        },
      )
      assert_equal ["kubernetes-deploy foo bar"], @spec.deploy_steps
    end

    test "#deploy_steps returns kubernetes-deploy command if both capfile and `kubernetes` are present" do
      @spec.stubs(:bundler?).returns(true)
      @spec.stubs(:capistrano?).returns(true)
      @spec.stubs(:load_config).returns(
        'kubernetes' => {
          'namespace' => 'foo',
          'context' => 'bar',
        },
      )
      assert_equal ["kubernetes-deploy foo bar"], @spec.deploy_steps
    end

    test "#deploy_steps returns kubernetes command if `kubernetes` is present and template_dir is set" do
      @spec.stubs(:load_config).returns(
        'kubernetes' => {
          'namespace' => 'foo',
          'context' => 'bar',
          'template_dir' => 'k8s_templates/',
        },
      )
      assert_equal ["kubernetes-deploy --template-dir k8s_templates/ foo bar"], @spec.deploy_steps
    end

    test "#deploy_steps prepend and append pre and post steps" do
      @spec.stubs(:load_config).returns('deploy' => {'pre' => ['before'], 'post' => ['after']})
      @spec.expects(:bundler?).returns(true).at_least_once
      @spec.expects(:capistrano?).returns(true)
      assert_equal ['before', 'bundle exec cap $ENVIRONMENT deploy', 'after'], @spec.deploy_steps
    end

    test '#deploy_steps raise a DeploySpec::Error! if it dont know how to deploy the app' do
      @spec.expects(:capistrano?).returns(false)
      assert_raise DeploySpec::Error do
        @spec.deploy_steps!
      end
    end

    test '#rollback_steps returns `rollback.override` if present' do
      @spec.stubs(:load_config).returns('rollback' => {'override' => %w(foo bar baz)})
      assert_equal %w(foo bar baz), @spec.rollback_steps
    end

    test '#rollback_steps returns `cap $ENVIRONMENT deploy:rollback` if a `Capfile` is present' do
      @spec.expects(:bundler?).returns(true).at_least_once
      @spec.expects(:capistrano?).returns(true)
      assert_equal ['bundle exec cap $ENVIRONMENT deploy:rollback'], @spec.rollback_steps
    end

    test "#rollback_steps prepend and append pre and post steps" do
      @spec.stubs(:load_config).returns('rollback' => {'pre' => ['before'], 'post' => ['after']})
      @spec.expects(:bundler?).returns(true).at_least_once
      @spec.expects(:capistrano?).returns(true)
      assert_equal ['before', 'bundle exec cap $ENVIRONMENT deploy:rollback', 'after'], @spec.rollback_steps
    end

    test "#rollback_steps returns `kubernetes-deploy <namespace> <context>` if `kubernetes` is present" do
      @spec.stubs(:load_config).returns(
        'kubernetes' => {
          'namespace' => 'foo',
          'context' => 'bar',
        },
      )
      assert_equal ["kubernetes-deploy foo bar"], @spec.rollback_steps
    end

    test "#rollback_steps returns kubernetes-deploy command when both capfile and `kubernetes` are present" do
      @spec.stubs(:bundler?).returns(true)
      @spec.stubs(:capistrano?).returns(true)
      @spec.stubs(:load_config).returns(
        'kubernetes' => {
          'namespace' => 'foo',
          'context' => 'bar',
        },
      )
      assert_equal ["kubernetes-deploy foo bar"], @spec.rollback_steps
    end

    test '#machine_env returns an environment hash' do
      @spec.stubs(:load_config).returns('machine' => {'environment' => {'GLOBAL' => '1'}})
      assert_equal({'GLOBAL' => '1'}, @spec.machine_env)
    end

    test '#load_config can grab the env-specific shipit.yml file' do
      config = {}
      config.expects(:exist?).returns(true)
      config.expects(:read).returns({'dependencies' => {'override' => %w(foo bar baz)}}.to_yaml)
      spec = DeploySpec::FileSystem.new('.', 'staging')
      spec.expects(:file).with('shipit.staging.yml', root: true).returns(config)
      assert_equal %w(foo bar baz), spec.dependencies_steps
    end

    test '#load_config grabs the global shipit.yml file if there is no env-specific file' do
      not_config = {}
      not_config.expects(:exist?).returns(false)

      config = {}
      config.expects(:exist?).returns(true)
      config.expects(:read).returns({'dependencies' => {'override' => %w(foo bar baz)}}.to_yaml)

      spec = DeploySpec::FileSystem.new('.', 'staging')
      spec.expects(:file).with('shipit.staging.yml', root: true).returns(not_config)
      spec.expects(:file).with('shipit.yml', root: true).returns(config)
      assert_equal %w(foo bar baz), spec.dependencies_steps
    end

    test '#gemspec gives the path of the repo gemspec if present' do
      spec = DeploySpec::FileSystem.new('foobar/', 'production')
      Dir.expects(:[]).with('foobar/*.gemspec').returns(['foobar/foobar.gemspec'])
      assert_equal 'foobar/foobar.gemspec', spec.gemspec
    end

    test '#gem? is true if a gemspec is present' do
      @spec.expects(:gemspec).returns('something')
      assert @spec.gem?
    end

    test '#gem? is false if there is no gemspec' do
      @spec.expects(:gemspec).returns(nil)
      refute @spec.gem?
    end

    test '#publish_gem first check if version tag have been created, and then invoke bundler release task' do
      @spec.stubs(:gemspec).returns('/tmp/shipit.gemspec')
      refute @spec.capistrano?
      assert_equal ['assert-gem-version-tag /tmp/shipit.gemspec', 'bundle exec rake release'], @spec.deploy_steps
    end

    test '#setup_dot_py gives the path of the repo setup.py if present' do
      spec = DeploySpec::FileSystem.new('foobar/', 'production')
      assert_equal Pathname.new('foobar/setup.py'), spec.setup_dot_py
    end

    test '#egg? is true if a setup.py is present' do
      @spec.expects(:setup_dot_py).returns(Shipit::Engine.root.join('Gemfile'))
      assert @spec.egg?
    end

    test '#egg? is false if there is no setup.py' do
      @spec.expects(:setup_dot_py).returns(Shipit::Engine.root.join("tmp-#{SecureRandom.hex}"))
      refute @spec.egg?
    end

    test '#publish_egg first check if version tag have been created and then invoke setup.py upload' do
      file = Pathname.new('/tmp/fake_setup.py')
      file.write('foo')
      @spec.stubs(:setup_dot_py).returns(file)
      steps = ['assert-egg-version-tag /tmp/fake_setup.py', 'python setup.py register sdist upload']
      assert_equal steps, @spec.deploy_steps
    end

    test '#cacheable returns a DeploySpec instance that can be serialized' do
      assert_instance_of DeploySpec::FileSystem, @spec
      assert_instance_of DeploySpec, @spec.cacheable
      config = {
        'merge' => {
          'require' => [],
          'ignore' => [],
          'revalidate_after' => nil,
        },
        'ci' => {
          'hide' => [],
          'allow_failures' => [],
          'require' => [],
        },
        'machine' => {'environment' => {}, 'directory' => nil, 'cleanup' => true},
        'review' => {'checklist' => [], 'monitoring' => [], 'checks' => []},
        'dependencies' => {'override' => []},
        'plugins' => {},
        'deploy' => {'override' => nil, 'variables' => [], 'max_commits' => 8, 'interval' => 0},
        'rollback' => {'override' => nil},
        'fetch' => nil,
        'tasks' => {},
      }
      assert_equal config, @spec.cacheable.config
    end

    test "#deploy_variables returns an empty array by default" do
      assert_equal [], @spec.deploy_variables
    end

    test "#deploy_variables returns an array of VariableDefinition instances" do
      @spec.stubs(:load_config).returns('deploy' => {'variables' => [{
        'name' => 'SAFETY_DISABLED',
        'title' => 'Set to 1 to do dangerous things',
        'default' => 0,
      }]})

      assert_equal 1, @spec.deploy_variables.size
      variable_definition = @spec.deploy_variables.first
      assert_equal 'SAFETY_DISABLED', variable_definition.name
    end

    test "task definitions prepend bundle exec if necessary" do
      @spec.expects(:load_config).returns('tasks' => {'restart' => {'steps' => %w(foo)}})
      @spec.expects(:bundler?).returns(true).at_least_once
      definition = @spec.find_task_definition('restart')

      assert_equal ['bundle exec foo'], definition.steps
    end

    test "task definitions do not prepend bundle exec if depedency step is overridden" do
      @spec.expects(:load_config).returns(
        'dependencies' => {'override' => []},
        'tasks' => {'restart' => {'steps' => %w(foo)}},
      )
      @spec.expects(:bundler?).returns(true).at_least_once
      definition = @spec.find_task_definition('restart')

      assert_equal ['foo'], definition.steps
    end

    test "task definitions prepend bundle exec before serialization" do
      @spec.expects(:load_config).returns('tasks' => {'restart' => {'steps' => %w(foo)}})
      @spec.expects(:bundler?).returns(true).at_least_once

      cached_spec = DeploySpec.load(DeploySpec.dump(@spec))
      definition = cached_spec.find_task_definition('restart')
      assert_equal ['bundle exec foo'], definition.steps
    end

    test "#task_definitions returns kubernetes commands as well as comands from the config" do
      @spec.stubs(:load_config).returns(
        'tasks' => {'another_task' => {'steps' => %w(foo)}},
        'kubernetes' => {
          'namespace' => 'foo',
          'context' => 'bar',
        },
      )
      tasks = @spec.task_definitions
      assert_equal 2, tasks.size

      restart_task = tasks.find { |t| t.id == "restart" }
      assert_equal ["kubernetes-restart foo bar"], restart_task.steps
    end

    test "task definitions returns an array of VariableDefinition instances" do
      @spec.expects(:load_config).returns('tasks' =>
        {'restart' =>
          {
            'variables' => [
              {
                'name' => 'SAFETY_DISABLED',
                'title' => 'Set to 1 to do dangerous things',
                'default' => 0,
              },
              {
                'name' => 'FOO',
                'title' => 'Set to 0 to foo',
                'default' => 1,
              },
            ],
            'steps' => %w(foo),
          },
        })

      assert_equal 2, @spec.task_definitions.first.variables.size
      variable_definition = @spec.task_definitions.first.variables.first
      assert_equal 'SAFETY_DISABLED', variable_definition.name
    end

    test "task definitions returns an empty array by default" do
      assert_equal [], @spec.task_definitions
    end

    test "#review_checklist returns an array" do
      @spec.expects(:load_config).returns('review' => {'checklist' => %w(foo bar)})
      assert_equal %w(foo bar), @spec.review_checklist
    end

    test "#review_checklist returns an empty array if the section is missing" do
      assert_equal [], @spec.review_checklist
    end

    test "#review_monitoring returns an array of hashes" do
      @spec.expects(:load_config).returns('review' => {'monitoring' => [
        {'image' => 'http://example.com/foo.png', 'width' => 200, 'height' => 400},
        {'iframe' => 'http://example.com/', 'width' => 200, 'height' => 400},
      ]})
      assert_equal [
        {'image' => 'http://example.com/foo.png', 'width' => 200, 'height' => 400},
        {'iframe' => 'http://example.com/', 'width' => 200, 'height' => 400},
      ], @spec.review_monitoring
    end

    test "#review_monitoring returns an empty array if the section is missing" do
      assert_equal [], @spec.review_monitoring
    end

    test "#hidden_statuses is empty by default" do
      assert_equal [], @spec.hidden_statuses
    end

    test "#hidden_statuses is an array even if the value is a string" do
      @spec.expects(:load_config).returns('ci' => {'hide' => 'ci/circleci'})
      assert_equal %w(ci/circleci), @spec.hidden_statuses
    end

    test "#hidden_statuses is an array even if the value is present" do
      @spec.expects(:load_config).returns('ci' => {'hide' => %w(ci/circleci ci/jenkins)})
      assert_equal %w(ci/circleci ci/jenkins), @spec.hidden_statuses
    end

    test "pull_request_ignored_statuses defaults to the union of ci.hide and ci.allow_failures" do
      @spec.expects(:load_config).returns(
        'ci' => {
          'hide' => %w(ci/circleci ci/jenkins),
          'allow_failures' => %w(ci/circleci ci/travis),
        },
      )
      assert_equal %w(ci/circleci ci/jenkins ci/travis).sort, @spec.pull_request_ignored_statuses.sort
    end

    test "pull_request_ignored_statuses defaults to empty if `merge.require` is present" do
      @spec.expects(:load_config).returns(
        'merge' => {
          'require' => 'bar',
        },
        'ci' => {
          'hide' => %w(ci/circleci ci/jenkins),
          'allow_failures' => %w(ci/circleci ci/travis),
        },
      )
      assert_equal [], @spec.pull_request_ignored_statuses
    end

    test "pull_request_ignored_statuses returns `merge.ignore` if present" do
      @spec.expects(:load_config).returns(
        'merge' => {
          'ignore' => 'bar',
        },
        'ci' => {
          'hide' => %w(ci/circleci ci/jenkins),
          'allow_failures' => %w(ci/circleci ci/travis),
        },
      )
      assert_equal ['bar'], @spec.pull_request_ignored_statuses
    end

    test "pull_request_required_statuses defaults to ci.require" do
      @spec.expects(:load_config).returns(
        'ci' => {
          'require' => %w(ci/circleci ci/jenkins),
        },
      )
      assert_equal %w(ci/circleci ci/jenkins), @spec.pull_request_required_statuses
    end

    test "pull_request_required_statuses defaults to empty if `merge.ignore` is present" do
      @spec.expects(:load_config).returns(
        'merge' => {
          'ignore' => 'bar',
        },
        'ci' => {
          'require' => %w(ci/circleci ci/jenkins),
        },
      )
      assert_equal [], @spec.pull_request_required_statuses
    end

    test "pull_request_required_statuses returns `merge.require` if present" do
      @spec.expects(:load_config).returns(
        'merge' => {
          'require' => 'bar',
        },
        'ci' => {
          'hide' => %w(ci/circleci ci/jenkins),
          'allow_failures' => %w(ci/circleci ci/travis),
        },
      )
      assert_equal ['bar'], @spec.pull_request_required_statuses
    end

    test "revalidate_pull_requests_after defaults to `nil" do
      @spec.expects(:load_config).returns({})
      assert_nil @spec.revalidate_pull_requests_after
    end

    test "revalidate_pull_requests_after defaults to `nil` if `merge.timeout` cannot be parsed" do
      @spec.expects(:load_config).returns(
        'merge' => {
          'revalidate_after' => 'ALSKhfjsdkf',
        },
      )
      assert_nil @spec.revalidate_pull_requests_after
    end

    test "revalidate_after returns `merge.revalidate_after` if present" do
      @spec.expects(:load_config).returns(
        'merge' => {
          'revalidate_after' => '5m30s',
        },
      )
      assert_equal 330, @spec.revalidate_pull_requests_after.to_i
    end

    test "#file is impacted by `machine.directory`" do
      subdir = '/foo/bar'
      @spec.stubs(:load_config).returns('machine' => {'directory' => subdir})
      assert_instance_of Pathname, @spec.file('baz')
      assert_equal File.join(@app_dir, subdir, 'baz'), @spec.file('baz').to_s
    end

    test "#clear_working_directory? returns true by default" do
      assert_predicate @spec, :clear_working_directory?
    end

    test 'bundler installs take priority over npm installs' do
      @spec.expects(:discover_package_json).never
      @spec.stubs(:discover_bundler).returns(['fake bundler task']).once

      assert_equal ['fake bundler task'], @spec.dependencies_steps
    end

    test 'Gems deploys take priority over npm deploys' do
      @spec.expects(:discover_npm_package).never
      @spec.stubs(:discover_gem).returns(['fake gem task']).once

      assert_equal ['fake gem task'], @spec.deploy_steps
    end

    test '#npm? is false if there is no package.json' do
      @spec.expects(:package_json).returns(Shipit::Engine.root.join("tmp-#{SecureRandom.hex}"))
      refute @spec.npm?
    end

    test '#npm? is false if npm package is private' do
      file = Pathname.new('/tmp/fake_package.json')
      file.write('{"private": true}')

      @spec.expects(:package_json).returns(file)
      refute @spec.npm?
    end

    test '#npm? is true if npm package is public' do
      file = Pathname.new('/tmp/fake_package.json')
      file.write('{"private": false}')

      @spec.expects(:package_json).returns(file)

      assert @spec.npm?
    end

    test 'npm packages have a checklist' do
      @spec.stubs(:npm?).returns(true).at_least_once
      assert_match(/npm version/, @spec.review_checklist[0])
    end

    test '#dependencies_steps returns `npm install` if a `package.json` is present' do
      @spec.expects(:npm?).returns(true).at_least_once
      assert_equal ['npm install --no-progress'], @spec.dependencies_steps
    end

    test '#publish_npm_package checks if version tag exists, and then invokes npm deploy script' do
      @spec.stubs(:npm?).returns(true)
      assert_equal ['assert-npm-version-tag', 'npm publish'], @spec.deploy_steps
    end

    test 'bundler installs take priority over yarn installs' do
      @spec.expects(:discover_yarn).never
      @spec.stubs(:discover_bundler).returns(['fake bundler task']).once

      assert_equal ['fake bundler task'], @spec.dependencies_steps
    end

    test 'Gems deploys take priority over yarn deploys' do
      @spec.expects(:discover_yarn_package).never
      @spec.stubs(:discover_gem).returns(['fake gem task']).once

      assert_equal ['fake gem task'], @spec.deploy_steps
    end

    test '#yarn? is false if there is no package.json' do
      @spec.expects(:package_json).returns(Shipit::Engine.root.join("tmp-#{SecureRandom.hex}"))
      @spec.expects(:yarn_lock).returns(Shipit::Engine.root.join('Gemfile'))

      refute @spec.yarn?
    end

    test '#yarn? is false if there is no yarn.lock' do
      @spec.expects(:yarn_lock).returns(Shipit::Engine.root.join("tmp-#{SecureRandom.hex}"))

      refute @spec.yarn?
    end

    test '#yarn? is false if a private package.json and yarn.lock are present' do
      package_json = Pathname.new('/tmp/fake_package.json')
      package_json.write('{"private": true}')

      @spec.expects(:package_json).returns(package_json)
      @spec.expects(:yarn_lock).returns(Shipit::Engine.root.join('Gemfile'))
      refute @spec.yarn?
    end

    test '#yarn? is true if a public package.json and yarn.lock are present' do
      package_json = Pathname.new('/tmp/fake_package.json')
      package_json.write('{"private": false}')

      yarn_lock = Pathname.new('/tmp/fake_yarn.lock')
      yarn_lock.write('')

      @spec.expects(:package_json).returns(package_json)
      @spec.expects(:yarn_lock).returns(yarn_lock)
      assert @spec.yarn?
    end

    test '#dependencies_steps returns `yarn install` if a `yarn.lock` is present' do
      @spec.expects(:yarn?).returns(true).at_least_once
      assert_equal ['yarn install --no-progress'], @spec.dependencies_steps
    end

    test '#publish_yarn_package checks if version tag exists, and then invokes npm publish script' do
      @spec.stubs(:yarn?).returns(true).at_least_once
      assert_equal ['assert-npm-version-tag', 'npm publish'], @spec.deploy_steps
    end

    test 'yarn checklist takes precedence over npm checklist' do
      @spec.stubs(:yarn?).returns(true).at_least_once
      assert_match(/yarn version/, @spec.review_checklist[0])
    end
  end
end