# ********** Copyright Viacom, Inc. Apache 2.0 **********

require_relative "test_helper.rb"

class KeyerTest < Minitest::Test
  def test_keyer_dev_id
    connection = Minitest::Mock.new
    faraday = Minitest::Mock.new
    response = Minitest::Mock.new

    device_config = {
      ip: "111.222.333",
      user: "user",
      password: "password",
      logger: Logger.new("/dev/null")
    }
    path = "/plugin_package"
    body = "v class=\"roku-font-5\"><label>Your Dev ID: &nbsp;</label> dev_id<hr></div>"

    connection.expect(:get, response, [path])
    faraday.expect(:request, nil, [:digest, device_config[:user], device_config[:password]])
    faraday.expect(:adapter, nil, [Faraday.default_adapter])
    response.expect(:body, body)
    response.expect(:body, body)


    dev_id = nil
    keyer = RokuBuilder::Keyer.new(**device_config)
    Faraday.stub(:new, connection, faraday) do
      dev_id = keyer.dev_id
    end

    assert_equal "dev_id", dev_id

    connection.verify
    faraday.verify
    response.verify
  end
  def test_keyer_dev_id_old_interface
    connection = Minitest::Mock.new
    faraday = Minitest::Mock.new
    response = Minitest::Mock.new

    device_config = {
      ip: "111.222.333",
      user: "user",
      password: "password",
      logger: Logger.new("/dev/null")
    }
    path = "/plugin_package"
    body = "<p> Your Dev ID: <font face=\"Courier\">dev_id</font> </p>"

    connection.expect(:get, response, [path])
    faraday.expect(:request, nil, [:digest, device_config[:user], device_config[:password]])
    faraday.expect(:adapter, nil, [Faraday.default_adapter])
    response.expect(:body, body)


    dev_id = nil
    keyer = RokuBuilder::Keyer.new(**device_config)
    Faraday.stub(:new, connection, faraday) do
      dev_id = keyer.dev_id
    end

    assert_equal "dev_id", dev_id

    connection.verify
    faraday.verify
    response.verify
  end

  def test_keyer_rekey_changed
    connection = Minitest::Mock.new
    faraday = Minitest::Mock.new
    io = Minitest::Mock.new
    response = Minitest::Mock.new

    device_config = {
      ip: "111.222.333",
      user: "user",
      password: "password",
      logger: Logger.new("/dev/null")
    }
    path = "/plugin_inspect"
    password = "password"
    payload ={
      mysubmit: "Rekey",
      password: password,
      archive: io
    }

    connection.expect(:post, response) do |arg1, arg2|
      assert_equal path, arg1
      assert_equal payload[:mysubmit], arg2[:mysubmit]
      assert_equal payload[:password], arg2[:passwd]
      assert payload[:archive] === arg2[:archive]
    end
    faraday.expect(:headers, {})
    faraday.expect(:request, nil, [:digest, device_config[:user], device_config[:password]])
    faraday.expect(:request, nil, [:multipart])
    faraday.expect(:request, nil, [:url_encoded])
    faraday.expect(:adapter, nil, [Faraday.default_adapter])

    # This test fails with the following seeds due to the random number
    # generator spitting out the same number twice
    # SEED=21894
    # SEED=31813
    dev_id = Proc.new {"#{Random.rand(100)}"}
    keyer = RokuBuilder::Keyer.new(**device_config)
    key_changed = nil
    Faraday.stub(:new, connection, faraday) do
      Faraday::UploadIO.stub(:new, io) do
        keyer.stub(:dev_id, dev_id) do
          key_changed = keyer.rekey(keyed_pkg: "pkg/path", password: password)
        end
      end
    end

    assert key_changed

    connection.verify
    faraday.verify
    io.verify
    response.verify
  end
end