File: direct_file_server_spec.rb

package info (click to toggle)
puppet 5.5.22-2
  • links: PTS, VCS
  • area: main
  • in suites: bullseye
  • size: 21,316 kB
  • sloc: ruby: 254,925; sh: 1,608; xml: 219; makefile: 153; sql: 103
file content (96 lines) | stat: -rw-r--r-- 4,016 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
require 'spec_helper'

require 'puppet/indirector/direct_file_server'

describe Puppet::Indirector::DirectFileServer do
  before(:each) do
    allow(Puppet::Indirector::Terminus).to receive(:register_terminus_class)
    @model = double('model')
    @indirection = double('indirection', :name => :mystuff, :register_terminus_type => nil, :model => @model)
    allow(Puppet::Indirector::Indirection).to receive(:instance).and_return(@indirection)

    module Testing; end

    # The Indirector does a lot of class-level caching of things, and indirections register themselves
    # whenever they're created (at include time), which makes working with them in tests _very_ annoying.
    # We're effectively undefining the test class if it exists from a previous test so that we can
    # re-register the new one as any mocks/stubs that existed on the old one from a previous test will no
    # longer be valid, and will cause rspec-mocks to (rightfully) blow up.
    Testing.send(:remove_const, :Mytype) if Testing.constants.include?(:Mytype)

    @direct_file_class = class Testing::Mytype < Puppet::Indirector::DirectFileServer
      self
    end

    @server = @direct_file_class.new

    @path = File.expand_path('/my/local')
    @uri = Puppet::Util.path_to_uri(@path).to_s

    @request = Puppet::Indirector::Request.new(:mytype, :find, @uri, nil)
  end

  describe Puppet::Indirector::DirectFileServer, "when finding a single file" do

    it "should return nil if the file does not exist" do
      expect(Puppet::FileSystem).to receive(:exist?).with(@path).and_return(false)
      expect(@server.find(@request)).to be_nil
    end

    it "should return a Content instance created with the full path to the file if the file exists" do
      expect(Puppet::FileSystem).to receive(:exist?).with(@path).and_return(true)
      mycontent = double('content', :collect => nil)
      expect(mycontent).to receive(:collect)
      expect(@model).to receive(:new).and_return(mycontent)
      expect(@server.find(@request)).to eq(mycontent)
    end
  end

  describe Puppet::Indirector::DirectFileServer, "when creating the instance for a single found file" do
    before do
      @data = double('content')
      allow(@data).to receive(:collect)
      expect(Puppet::FileSystem).to receive(:exist?).with(@path).and_return(true)
    end

    it "should pass the full path to the instance" do
      expect(@model).to receive(:new).with(@path, anything).and_return(@data)
      @server.find(@request)
    end

    it "should pass the :links setting on to the created Content instance if the file exists and there is a value for :links" do
      expect(@model).to receive(:new).and_return(@data)
      expect(@data).to receive(:links=).with(:manage)

      allow(@request).to receive(:options).and_return(:links => :manage)
      @server.find(@request)
    end

    it "should set 'checksum_type' on the instances if it is set in the request options" do
      expect(@model).to receive(:new).and_return(@data)
      expect(@data).to receive(:checksum_type=).with(:checksum)

      allow(@request).to receive(:options).and_return(:checksum_type => :checksum)
      @server.find(@request)
    end
  end

  describe Puppet::Indirector::DirectFileServer, "when searching for multiple files" do
    it "should return nil if the file does not exist" do
      expect(Puppet::FileSystem).to receive(:exist?).with(@path).and_return(false)
      expect(@server.find(@request)).to be_nil
    end

    it "should use :path2instances from the terminus_helper to return instances if the file exists" do
      expect(Puppet::FileSystem).to receive(:exist?).with(@path).and_return(true)
      expect(@server).to receive(:path2instances)
      @server.search(@request)
    end

    it "should pass the original request to :path2instances" do
      expect(Puppet::FileSystem).to receive(:exist?).with(@path).and_return(true)
      expect(@server).to receive(:path2instances).with(@request, @path)
      @server.search(@request)
    end
  end
end