Contents

require 'spec_helper'

RSpec.describe RubySMB::SMB2::Tree do
  let(:dispatcher) { RubySMB::Dispatcher::Socket.new(nil) }

  let(:client) { RubySMB::Client.new(dispatcher, username: 'msfadmin', password: 'msfadmin') }
  let(:tree_id) { 2049 }
  let(:path) { '\\192.168.1.1\example' }
  let(:response) {
    packet = RubySMB::SMB2::Packet::TreeConnectResponse.new
    packet.smb2_header.tree_id = tree_id
    packet.maximal_access.read("\xff\x01\x1f\x00")
    packet.share_type = 0x01
    packet
  }
  subject(:tree) {
    described_class.new(client:client, share:path, response:response )
  }

  it { is_expected.to respond_to :client }
  it { is_expected.to respond_to :permissions }
  it { is_expected.to respond_to :share }
  it { is_expected.to respond_to :id }

  it 'inherits the client that spawned it' do
    expect(tree.client).to eq client
  end

  it 'inherits the permissions from the response packet' do
    expect(tree.permissions).to eq response.maximal_access
  end

  it 'inherits the Tree id from the response packet' do
    expect(tree.id).to eq response.smb2_header.tree_id
  end

  describe '#disconnect!' do
    let(:disco_req) { RubySMB::SMB2::Packet::TreeDisconnectRequest.new }
    let(:disco_resp) { RubySMB::SMB2::Packet::TreeDisconnectResponse.new }

    it 'sends a TreeDisconnectRequest with the Tree ID in the header' do
      allow(RubySMB::SMB2::Packet::TreeDisconnectRequest).to receive(:new).and_return(disco_req)
      modified_req = disco_req
      modified_req.smb2_header.tree_id = tree.id
      expect(client).to receive(:send_recv).with(modified_req).and_return(disco_resp.to_binary_s)
      tree.disconnect!
    end

    it 'returns the NTStatus code from the response' do
      allow(client).to receive(:send_recv).and_return(disco_resp.to_binary_s)
      expect(tree.disconnect!).to eq disco_resp.status_code
    end
  end
end

Version data entries

6 entries across 6 versions & 1 rubygems

Version	Path
ruby_smb-0.0.14	spec/lib/ruby_smb/smb2/tree_spec.rb
ruby_smb-0.0.13	spec/lib/ruby_smb/smb2/tree_spec.rb
ruby_smb-0.0.12	spec/lib/ruby_smb/smb2/tree_spec.rb
ruby_smb-0.0.11	spec/lib/ruby_smb/smb2/tree_spec.rb
ruby_smb-0.0.10	spec/lib/ruby_smb/smb2/tree_spec.rb
ruby_smb-0.0.9	spec/lib/ruby_smb/smb2/tree_spec.rb