require_relative '../../test_helper'

describe WpaCliRuby do
  before do
    @wrapper = mock()
    @wpa_cli = WpaCliRuby::WpaCli.new(@wrapper)
  end

  it "can be instantiated" do
    assert @wpa_cli
  end

  describe "scan" do
    before do
      @wrapper.expects(:scan).returns("Selected interface 'wlan0'\nOK\n")
    end

    it "returns the interface and the status code" do
      response = @wpa_cli.scan
      assert_equal 'wlan0', response.interface
      assert response.ok?
    end
  end

  describe "add_network" do
    before do
      @wrapper.expects(:add_network).returns("Selected interface 'wlan0'\n10\n")
    end

    it "returns the interface and the status code" do
      network_id = @wpa_cli.add_network
      assert_equal 10, network_id
    end
  end

  describe "save_config" do
    before do
      @wrapper.expects(:save_config).returns("Selected interface 'wlan0'\nOK\n")
    end

    it "returns the interface and the status code" do
      response = @wpa_cli.save_config
      assert_equal 'wlan0', response.interface
      assert response.ok?
    end
  end

  describe "set_network" do
    it "returns the interface and the status code" do
      @wrapper.expects(:set_network).with(0, 'test', 'test').returns("Selected interface 'wlan0'\nOK\n")

      response = @wpa_cli.set_network(0, 'test', 'test')
      assert_equal 'wlan0', response.interface
      assert response.ok?
    end

    it "raises for unsuccessful setting" do
      @wrapper.expects(:set_network).with(0, 'test', 'test').returns("Selected interface 'wlan0'\nFAIL\n")
      assert_raises(WpaCliRuby::SetNetworkFailure) { @wpa_cli.set_network(0, 'test', 'test') }
    end
  end

  describe "get_network" do
    it "returns the ssid" do
      @wrapper.expects(:get_network).with(0, 'ssid').returns("Selected interface 'wlan0'\nssid1")

      ssid = @wpa_cli.get_network(0, 'ssid')
      assert_equal 'ssid1', ssid
    end

    it "raises for unknown network" do
      @wrapper.expects(:get_network).with(1, 'ssid').returns("Selected interface 'wlan0'\nFAIL\n")
      assert_raises(WpaCliRuby::NetworkNotFound) { @wpa_cli.get_network(1, 'ssid') }
    end
  end

  describe "enable_network" do
    it "returns the interface and the status code for successful enabling" do
      @wrapper.expects(:enable_network).with(0).returns("Selected interface 'wlan0'\nOK\n")

      response = @wpa_cli.enable_network(0)

      assert_equal 'wlan0', response.interface
      assert response.ok?
    end

    it "raises for unsuccessful enabling" do
      @wrapper.expects(:enable_network).with(1).returns("Selected interface 'wlan0'\nFAIL\n")
      assert_raises(WpaCliRuby::EnableNetworkFailure) { @wpa_cli.enable_network(1) }
    end
  end

  describe "scan_results" do
    before do
      response = <<-eos
Selected interface 'wlan0'
bssid / frequency / signal level / flags / ssid
12:34:56:78:aa:bb	2437	-47	[WPA-EAP-TKIP][WPA2-EAP-CCMP][ESS]	ssid1
12:34:56:78:bb:cc	2412	-57	[WPA2-PSK-CCMP][ESS]	ssid2
eos
      @wrapper.expects(:scan_results).returns(response)
    end

    it "returns an array of ScanResult objects" do
      scan_results = @wpa_cli.scan_results
      assert_equal 2, scan_results.length

      assert_equal 'ssid1', scan_results[0].ssid
      assert_equal -47, scan_results[0].signal_level
      assert_equal 2437, scan_results[0].frequency

      assert_equal 'ssid2', scan_results[1].ssid
    end
  end
end