-
Notifications
You must be signed in to change notification settings - Fork 896
/
custom_buttons_spec.rb
171 lines (148 loc) · 6.59 KB
/
custom_buttons_spec.rb
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
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
RSpec.describe TaskHelpers::Imports::CustomButtons do
let(:data_dir) { File.join(File.expand_path(__dir__), 'data', 'custom_buttons') }
let(:custom_button_file) { 'CustomButtons.yaml' }
let(:bad_custom_button_file) { 'CustomButtonsBad.yaml' }
let(:custom_button_set_name) { 'group1|Vm|' }
let(:custom_button_set_description) { 'group1' }
let(:resource_action_ae_namespace) { 'SYSTEM' }
let(:connect_dialog_by_name) { true }
let(:options) { {:source => source, :connect_dialog_by_name => connect_dialog_by_name} }
let!(:test_dialog) { FactoryBot.create(:dialog, :label => 'dialog') }
let!(:test_dialog_2) { FactoryBot.create(:dialog, :label => 'dialog 2') }
describe "#import" do
describe "when the source is a directory" do
let(:source) { data_dir }
context "without existing buttons" do
it 'imports all formatted .yaml files in a specified directory' do
TaskHelpers::Imports::CustomButtons.new.import(options)
assert_test_custom_button_set_present
end
end
context "with existing identical buttons" do
it 'should raise' do
TaskHelpers::Imports::CustomButtons.new.import(options)
assert_raises_import_error
end
end
context "yaml import failure" do
it 'should raise' do
Tempfile.create(%w[foo .yaml], data_dir) do |file|
file.write("bad yaml here")
assert_raises_import_error
end
end
end
end
describe "when the source is a file" do
let(:source) { "#{data_dir}/#{custom_button_file}" }
context "without existing buttons" do
context "only imports good yaml" do
context "connect dialog flag is set" do
it 'imports a specified file' do
TaskHelpers::Imports::CustomButtons.new.import(options)
assert_test_custom_button_set_present
assert_imports_only_custom_button_set_one
assert_dialog_is_set(true)
end
end
context "connect dialog flag not set" do
let(:connect_dialog_by_name) { false }
it 'imports a specified file' do
TaskHelpers::Imports::CustomButtons.new.import(options)
assert_test_custom_button_set_present
assert_imports_only_custom_button_set_one
assert_dialog_is_set(false)
end
end
end
end
context "doesn't import bad yaml" do
let(:source) { "#{data_dir}/#{bad_custom_button_file}" }
it 'does not imports a specified file' do
TaskHelpers::Imports::CustomButtons.new.import(options)
assert_imports_no_custom_buttons
end
end
context "with existing identical buttons" do
it 'should not import anything' do
TaskHelpers::Imports::CustomButtons.new.import(options)
assert_imports_only_custom_button_set_one
end
end
end
# This is a sort of "integration test" where we test that the export format
# works with the import format directly, instead of with a saved copy.
#
# That way, if something changes with the export, we will know if the
# import works correctly still.
#
describe "imports and export properly" do
let(:export_dir) { Dir.mktmpdir }
let(:source) { File.join(export_dir, "CustomButtons.yaml") }
let!(:custom_button1) { FactoryBot.create(:custom_button, :name => "button1", :description => "Button One", :applies_to_class => "Vm", :resource_action => resource_action1) }
let!(:custom_button2) { FactoryBot.create(:custom_button, :name => "button2", :description => "Button Two", :applies_to_class => "Vm", :resource_action => resource_action2) }
let!(:custom_button_set1) { FactoryBot.create(:custom_button_set, :name => "set1", :description => "Set One") }
let(:resource_action1) do
FactoryBot.build(:resource_action,
:ae_namespace => "NAMESPACE",
:ae_class => "CLASS",
:ae_instance => "INSTANCE")
end
let(:resource_action2) do
FactoryBot.build(:resource_action,
:ae_namespace => "SYSTEM",
:ae_class => "PROCESS",
:ae_instance => "Request",
:dialog => FactoryBot.create(:dialog, :name => "label1"))
end
after do
FileUtils.remove_entry export_dir
end
it 'exports custom buttons to a given directory' do
expect(CustomButton.count).to eq(2)
expect(CustomButtonSet.count).to eq(1)
expect(Dialog.where(:label => "label1").count).to eq(1)
expect(ResourceAction.count).to eq(2)
TaskHelpers::Exports::CustomButtons.new.export(:directory => export_dir)
custom_button1.destroy
custom_button2.destroy
custom_button_set1.destroy
resource_action1.destroy
resource_action2.destroy
expect(CustomButton.count).to eq(0)
expect(CustomButtonSet.count).to eq(0)
expect(ResourceAction.count).to eq(0)
TaskHelpers::Imports::CustomButtons.new.import(options)
expect(CustomButton.count).to eq(2)
expect(CustomButtonSet.count).to eq(1)
expect(Dialog.where(:label => "label1").count).to eq(1)
expect(ResourceAction.count).to eq(2)
end
end
end
def assert_dialog_is_set(connect)
btn1 = CustomButton.find_by(:name => 'button 1')
expect(btn1).to be_an(CustomButton)
if connect
expect(btn1.resource_action.dialog.id).to eq(test_dialog_2.id)
else
expect(btn1.resource_action.dialog).to be_nil
end
end
def assert_test_custom_button_set_present
cbs = CustomButtonSet.find_by(:name => custom_button_set_name)
expect(cbs.custom_buttons.count).to eq(3)
expect(cbs.description).to eq(custom_button_set_description)
expect(cbs.custom_buttons.first.resource_action.ae_namespace).to eq(resource_action_ae_namespace)
expect(cbs.custom_buttons.pluck(:userid)).to eq(%w[admin admin admin])
end
def assert_imports_only_custom_button_set_one
expect(CustomButton.count).to eq(3)
end
def assert_imports_no_custom_buttons
expect(CustomButton.count).to eq(0)
end
def assert_raises_import_error
expect { TaskHelpers::Imports::CustomButtons.new.import(options) }.to raise_error(StandardError, /Error importing/)
end
end