241 lines
8.6 KiB
Ruby
241 lines
8.6 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
require 'spec_helper'
|
|
|
|
RSpec.describe Types::BaseField do
|
|
context 'when considering complexity' do
|
|
let(:resolver) do
|
|
Class.new(described_class) do
|
|
def self.resolver_complexity(args, child_complexity:)
|
|
2 if args[:foo]
|
|
end
|
|
|
|
def self.complexity_multiplier(args)
|
|
0.01
|
|
end
|
|
end
|
|
end
|
|
|
|
it 'defaults to 1' do
|
|
field = described_class.new(name: 'test', type: GraphQL::Types::String, null: true)
|
|
|
|
expect(field.complexity).to eq 1
|
|
end
|
|
|
|
describe '#base_complexity' do
|
|
context 'with no gitaly calls' do
|
|
it 'defaults to 1' do
|
|
field = described_class.new(name: 'test', type: GraphQL::Types::String, null: true)
|
|
|
|
expect(field.base_complexity).to eq 1
|
|
end
|
|
end
|
|
|
|
context 'with a gitaly call' do
|
|
it 'adds 1 to the default value' do
|
|
field = described_class.new(name: 'test', type: GraphQL::Types::String, null: true, calls_gitaly: true)
|
|
|
|
expect(field.base_complexity).to eq 2
|
|
end
|
|
end
|
|
end
|
|
|
|
it 'has specified value' do
|
|
field = described_class.new(name: 'test', type: GraphQL::Types::String, null: true, complexity: 12)
|
|
|
|
expect(field.complexity).to eq 12
|
|
end
|
|
|
|
context 'when field has a resolver' do
|
|
context 'when a valid complexity is already set' do
|
|
let(:field) { described_class.new(name: 'test', type: GraphQL::Types::String.connection_type, resolver_class: resolver, complexity: 2, max_page_size: 100, null: true) }
|
|
|
|
it 'uses this complexity' do
|
|
expect(field.complexity).to eq 2
|
|
end
|
|
end
|
|
|
|
context 'and is a connection' do
|
|
let(:field) { described_class.new(name: 'test', type: GraphQL::Types::String.connection_type, resolver_class: resolver, max_page_size: 100, null: true) }
|
|
|
|
it 'sets complexity depending on arguments for resolvers' do
|
|
expect(field.complexity.call({}, {}, 2)).to eq 4
|
|
expect(field.complexity.call({}, { first: 50 }, 2)).to eq 3
|
|
end
|
|
|
|
it 'sets complexity depending on number load limits for resolvers' do
|
|
expect(field.complexity.call({}, { first: 1 }, 2)).to eq 2
|
|
expect(field.complexity.call({}, { first: 1, foo: true }, 2)).to eq 4
|
|
end
|
|
end
|
|
|
|
context 'and is not a connection' do
|
|
it 'sets complexity as normal' do
|
|
field = described_class.new(name: 'test', type: GraphQL::Types::String, resolver_class: resolver, max_page_size: 100, null: true)
|
|
|
|
expect(field.complexity.call({}, {}, 2)).to eq 2
|
|
expect(field.complexity.call({}, { first: 50 }, 2)).to eq 2
|
|
end
|
|
end
|
|
end
|
|
|
|
context 'calls_gitaly' do
|
|
context 'for fields with a resolver' do
|
|
it 'adds 1 if true' do
|
|
with_gitaly_field = described_class.new(name: 'test', type: GraphQL::Types::String, resolver_class: resolver, null: true, calls_gitaly: true)
|
|
without_gitaly_field = described_class.new(name: 'test', type: GraphQL::Types::String, resolver_class: resolver, null: true)
|
|
base_result = without_gitaly_field.complexity.call({}, {}, 2)
|
|
|
|
expect(with_gitaly_field.complexity.call({}, {}, 2)).to eq base_result + 1
|
|
end
|
|
end
|
|
|
|
context 'for fields without a resolver' do
|
|
it 'adds 1 if true' do
|
|
field = described_class.new(name: 'test', type: GraphQL::Types::String, null: true, calls_gitaly: true)
|
|
|
|
expect(field.complexity).to eq 2
|
|
end
|
|
end
|
|
|
|
it 'defaults to false' do
|
|
field = described_class.new(name: 'test', type: GraphQL::Types::String, null: true)
|
|
|
|
expect(field.base_complexity).to eq Types::BaseField::DEFAULT_COMPLEXITY
|
|
end
|
|
|
|
context 'with declared constant complexity value' do
|
|
it 'has complexity set to that constant' do
|
|
field = described_class.new(name: 'test', type: GraphQL::Types::String, null: true, complexity: 12)
|
|
|
|
expect(field.complexity).to eq 12
|
|
end
|
|
|
|
it 'does not raise an error even with Gitaly calls' do
|
|
allow(Gitlab::GitalyClient).to receive(:get_request_count).and_return([0, 1])
|
|
field = described_class.new(name: 'test', type: GraphQL::Types::String, null: true, complexity: 12)
|
|
|
|
expect(field.complexity).to eq 12
|
|
end
|
|
end
|
|
end
|
|
|
|
describe '#visible?' do
|
|
context 'and has a feature_flag' do
|
|
let(:flag) { :test_feature }
|
|
let(:field) { described_class.new(name: 'test', type: GraphQL::Types::String, feature_flag: flag, null: false) }
|
|
let(:context) { {} }
|
|
|
|
before do
|
|
skip_feature_flags_yaml_validation
|
|
end
|
|
|
|
it 'checks YAML definition for default_enabled' do
|
|
# Exception is indicative of a check for YAML definition
|
|
expect { field.visible?(context) }.to raise_error(Feature::InvalidFeatureFlagError, /The feature flag YAML definition for '#{flag}' does not exist/)
|
|
end
|
|
|
|
context 'skipping YAML check' do
|
|
before do
|
|
skip_default_enabled_yaml_check
|
|
end
|
|
|
|
it 'returns false if the feature is not enabled' do
|
|
stub_feature_flags(flag => false)
|
|
|
|
expect(field.visible?(context)).to eq(false)
|
|
end
|
|
|
|
it 'returns true if the feature is enabled' do
|
|
expect(field.visible?(context)).to eq(true)
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|
|
|
|
describe '#resolve' do
|
|
context "late_extensions is given" do
|
|
it 'registers the late extensions after the regular extensions' do
|
|
extension_class = Class.new(GraphQL::Schema::Field::ConnectionExtension)
|
|
field = described_class.new(name: 'test', type: GraphQL::Types::String.connection_type, null: true, late_extensions: [extension_class])
|
|
|
|
expect(field.extensions.last.class).to be(extension_class)
|
|
end
|
|
end
|
|
end
|
|
|
|
describe '#description' do
|
|
context 'feature flag given' do
|
|
let(:field) { described_class.new(name: 'test', type: GraphQL::Types::String, feature_flag: flag, null: false, description: 'Test description.') }
|
|
let(:flag) { :test_flag }
|
|
|
|
it 'prepends the description' do
|
|
expect(field.description).to start_with 'Test description. Available only when feature flag `test_flag` is enabled.'
|
|
end
|
|
|
|
context 'falsey feature_flag values' do
|
|
using RSpec::Parameterized::TableSyntax
|
|
|
|
where(:flag, :feature_value, :default_enabled) do
|
|
'' | false | false
|
|
'' | true | false
|
|
nil | false | true
|
|
nil | true | false
|
|
end
|
|
|
|
with_them do
|
|
it 'returns the correct description' do
|
|
expect(field.description).to eq('Test description.')
|
|
end
|
|
end
|
|
end
|
|
|
|
context 'with different default_enabled values' do
|
|
using RSpec::Parameterized::TableSyntax
|
|
|
|
where(:feature_value, :default_enabled, :expected_description) do
|
|
disabled_ff_description = "Test description. Available only when feature flag `test_flag` is enabled. This flag is disabled by default, because the feature is experimental and is subject to change without notice."
|
|
enabled_ff_description = "Test description. Available only when feature flag `test_flag` is enabled. This flag is enabled by default."
|
|
|
|
false | false | disabled_ff_description
|
|
true | false | disabled_ff_description
|
|
false | true | enabled_ff_description
|
|
true | true | enabled_ff_description
|
|
end
|
|
|
|
with_them do
|
|
before do
|
|
stub_feature_flags("#{flag}": feature_value)
|
|
|
|
allow(Feature::Definition).to receive(:has_definition?).with(flag).and_return(true)
|
|
allow(Feature::Definition).to receive(:default_enabled?).and_return(default_enabled)
|
|
end
|
|
|
|
it 'returns the correct availability in the description' do
|
|
expect(field.description). to eq expected_description
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|
|
|
|
include_examples 'Gitlab-style deprecations' do
|
|
def subject(args = {})
|
|
base_args = { name: 'test', type: GraphQL::Types::String, null: true }
|
|
|
|
described_class.new(**base_args.merge(args))
|
|
end
|
|
|
|
it 'interacts well with the `feature_flag` property' do
|
|
field = subject(
|
|
deprecated: { milestone: '1.10', reason: 'Deprecation reason' },
|
|
description: 'Field description.',
|
|
feature_flag: 'foo_flag'
|
|
)
|
|
|
|
expect(field.description).to start_with('Field description. Available only when feature flag `foo_flag` is enabled.')
|
|
expect(field.description).to end_with('Deprecated in 1.10: Deprecation reason.')
|
|
end
|
|
end
|
|
end
|