1 # Copyright (C) The Arvados Authors. All rights reserved.
3 # SPDX-License-Identifier: AGPL-3.0
7 class ArvadosModelTest < ActiveSupport::TestCase
10 def create_with_attrs attrs
11 a = Specimen.create({material: 'caloric'}.merge(attrs))
15 test 'non-admin cannot assign uuid' do
16 set_user_from_auth :active_trustedclient
17 want_uuid = Specimen.generate_uuid
18 a = create_with_attrs(uuid: want_uuid)
19 assert_nil a, "Non-admin should not assign uuid."
22 test 'admin can assign valid uuid' do
23 set_user_from_auth :admin_trustedclient
24 want_uuid = Specimen.generate_uuid
25 a = create_with_attrs(uuid: want_uuid)
26 assert_equal want_uuid, a.uuid, "Admin should assign valid uuid."
27 assert a.uuid.length==27, "Auto assigned uuid length is wrong."
30 test 'admin cannot assign uuid with wrong object type' do
31 set_user_from_auth :admin_trustedclient
32 want_uuid = Human.generate_uuid
33 a = create_with_attrs(uuid: want_uuid)
34 assert_nil a, "Admin should not be able to assign invalid uuid."
37 test 'admin cannot assign badly formed uuid' do
38 set_user_from_auth :admin_trustedclient
39 a = create_with_attrs(uuid: "ntoheunthaoesunhasoeuhtnsaoeunhtsth")
40 assert_nil a, "Admin should not be able to assign invalid uuid."
43 test 'admin cannot assign empty uuid' do
44 set_user_from_auth :admin_trustedclient
45 a = create_with_attrs(uuid: "")
46 assert_nil a, "Admin cannot assign empty uuid."
51 {:a => ['foo', 'bar']},
52 {'a' => [:foo, 'bar']},
53 {'a' => ['foo', :bar]},
55 {:a => {'foo' => {'bar' => 'baz'}}},
56 {'a' => {:foo => {'bar' => 'baz'}}},
57 {'a' => {'foo' => {:bar => 'baz'}}},
58 {'a' => {'foo' => {'bar' => :baz}}},
59 {'a' => {'foo' => ['bar', :baz]}},
61 test "prevent symbol keys in serialized db columns: #{x.inspect}" do
62 set_user_from_auth :active
63 link = Link.create!(link_class: 'test',
65 raw = ActiveRecord::Base.connection.
66 select_value("select properties from links where uuid='#{link.uuid}'")
67 refute_match(/:[fb]/, raw)
72 {'a' => {['foo', :foo] => 'bar'}},
73 {'a' => {{'foo' => 'bar'} => 'bar'}},
74 {'a' => {['foo', :foo] => ['bar', 'baz']}},
76 test "refuse non-string keys in serialized db columns: #{x.inspect}" do
77 set_user_from_auth :active
78 assert_raises(ArgumentError) do
79 Link.create!(link_class: 'test',
85 test "Stringify symbols coming from serialized attribute in database" do
86 set_user_from_auth :admin_trustedclient
87 fixed = Link.find_by_uuid(links(:has_symbol_keys_in_database_somehow).uuid)
88 assert_equal(["baz", "foo"], fixed.properties.keys.sort,
89 "Hash symbol keys from DB did not get stringified.")
90 assert_equal(['waz', 'waz', 'waz', 1, nil, false, true],
91 fixed.properties['baz'],
92 "Array symbol values from DB did not get stringified.")
93 assert_equal true, fixed.save, "Failed to save fixed model back to db."
96 test "No HashWithIndifferentAccess in database" do
97 set_user_from_auth :admin_trustedclient
98 link = Link.create!(link_class: 'test',
99 properties: {'foo' => 'bar'}.with_indifferent_access)
100 raw = ActiveRecord::Base.connection.
101 select_value("select properties from links where uuid='#{link.uuid}'")
102 assert_equal '{"foo":"bar"}', raw
105 test "store long string" do
106 set_user_from_auth :active
108 while longstring.length < 2**16
109 longstring = longstring + longstring
111 g = Group.create! name: 'Has a long description', description: longstring
112 g = Group.find_by_uuid g.uuid
113 assert_equal g.description, longstring
116 [['uuid', {unique: true}],
117 ['owner_uuid', {}]].each do |the_column, requires|
118 test "unique index on all models with #{the_column}" do
120 ActiveRecord::Base.connection.tables.each do |table|
121 columns = ActiveRecord::Base.connection.columns(table)
123 next unless columns.collect(&:name).include? the_column
125 indexes = ActiveRecord::Base.connection.indexes(table).reject do |index|
126 requires.map do |key, val|
127 index.send(key) == val
130 assert_includes indexes.collect(&:columns), [the_column], 'no index'
133 # Sanity check: make sure we didn't just systematically miss everything.
134 assert_operator(10, :<, checked,
135 "Only #{checked} tables have a #{the_column}?!")
139 test "search index exists on models that go into projects" do
140 all_tables = ActiveRecord::Base.connection.tables
141 all_tables.delete 'schema_migrations'
143 all_tables.each do |table|
144 table_class = table.classify.constantize
145 if table_class.respond_to?('searchable_columns')
146 search_index_columns = table_class.searchable_columns('ilike')
147 # Disappointing, but text columns aren't indexed yet.
148 search_index_columns -= table_class.columns.select { |c|
149 c.type == :text or c.name == 'description' or c.name == 'file_names'
152 indexes = ActiveRecord::Base.connection.indexes(table)
153 search_index_by_columns = indexes.select do |index|
154 index.columns.sort == search_index_columns.sort
156 search_index_by_name = indexes.select do |index|
157 index.name == "#{table}_search_index"
159 assert !search_index_by_columns.empty?, "#{table} has no search index with columns #{search_index_columns}. Instead found search index with columns #{search_index_by_name.first.andand.columns}"
164 test "full text search index exists on models" do
166 conn = ActiveRecord::Base.connection
167 conn.exec_query("SELECT i.relname as indname,
168 i.relowner as indowner,
169 idx.indrelid::regclass::text as table,
173 SELECT pg_get_indexdef(idx.indexrelid, k + 1, true)
174 FROM generate_subscripts(idx.indkey, 1) as k
177 idx.indexprs IS NOT NULL as indexprs,
178 idx.indpred IS NOT NULL as indpred
181 ON i.oid = idx.indexrelid
184 JOIN pg_namespace as ns
185 ON ns.oid = i.relnamespace
186 AND ns.nspname = ANY(current_schemas(false))").each do |idx|
187 if idx['keys'].match(/to_tsvector/)
188 indexes[idx['table']] ||= []
189 indexes[idx['table']] << idx
192 fts_tables = ["collections", "container_requests", "groups", "jobs",
193 "pipeline_instances", "pipeline_templates", "workflows"]
194 fts_tables.each do |table|
195 table_class = table.classify.constantize
196 if table_class.respond_to?('full_text_searchable_columns')
197 expect = table_class.full_text_searchable_columns
199 indexes[table].andand.each do |idx|
200 if expect == idx['keys'].scan(/COALESCE\(([A-Za-z_]+)/).flatten
204 assert ok, "#{table} has no full-text index\nexpect: #{expect.inspect}\nfound: #{indexes[table].inspect}"
209 test "selectable_attributes includes database attributes" do
210 assert_includes(Job.selectable_attributes, "success")
213 test "selectable_attributes includes non-database attributes" do
214 assert_includes(Job.selectable_attributes, "node_uuids")
217 test "selectable_attributes includes common attributes in extensions" do
218 assert_includes(Job.selectable_attributes, "uuid")
221 test "selectable_attributes does not include unexposed attributes" do
222 refute_includes(Job.selectable_attributes, "nodes")
225 test "selectable_attributes on a non-default template" do
226 attr_a = Job.selectable_attributes(:common)
227 assert_includes(attr_a, "uuid")
228 refute_includes(attr_a, "success")
231 test 'create and retrieve using created_at time' do
232 set_user_from_auth :active
233 group = Group.create! name: 'test create and retrieve group'
234 assert group.valid?, "group is not valid"
236 results = Group.where(created_at: group.created_at)
237 assert_includes results.map(&:uuid), group.uuid,
238 "Expected new group uuid in results when searched with its created_at timestamp"
241 test 'create and update twice and expect different update times' do
242 set_user_from_auth :active
243 group = Group.create! name: 'test create and retrieve group'
244 assert group.valid?, "group is not valid"
247 group.update_attributes!(name: "test create and update name 1")
248 results = Group.where(uuid: group.uuid)
249 assert_equal "test create and update name 1", results.first.name, "Expected name to be updated to 1"
250 updated_at_1 = results.first.updated_at.to_f
253 group.update_attributes!(name: "test create and update name 2")
254 results = Group.where(uuid: group.uuid)
255 assert_equal "test create and update name 2", results.first.name, "Expected name to be updated to 2"
256 updated_at_2 = results.first.updated_at.to_f
258 assert_equal true, (updated_at_2 > updated_at_1), "Expected updated time 2 to be newer than 1"