diff options
Diffstat (limited to 'spec/support/graphql/subscriptions/action_cable')
-rw-r--r-- | spec/support/graphql/subscriptions/action_cable/mock_action_cable.rb | 100 | ||||
-rw-r--r-- | spec/support/graphql/subscriptions/action_cable/mock_gitlab_schema.rb | 41 |
2 files changed, 0 insertions, 141 deletions
diff --git a/spec/support/graphql/subscriptions/action_cable/mock_action_cable.rb b/spec/support/graphql/subscriptions/action_cable/mock_action_cable.rb deleted file mode 100644 index 5467564a79e..00000000000 --- a/spec/support/graphql/subscriptions/action_cable/mock_action_cable.rb +++ /dev/null @@ -1,100 +0,0 @@ -# frozen_string_literal: true - -# A stub implementation of ActionCable. -# Any methods to support the mock backend have `mock` in the name. -module Graphql - module Subscriptions - module ActionCable - class MockActionCable - class MockChannel - def initialize - @mock_broadcasted_messages = [] - end - - attr_reader :mock_broadcasted_messages - - def stream_from(stream_name, coder: nil, &block) - # Rails uses `coder`, we don't - block ||= ->(msg) { @mock_broadcasted_messages << msg } - MockActionCable.mock_stream_for(stream_name).add_mock_channel(self, block) - end - end - - class MockStream - def initialize - @mock_channels = {} - end - - def add_mock_channel(channel, handler) - @mock_channels[channel] = handler - end - - def mock_broadcast(message) - @mock_channels.each do |channel, handler| - handler && handler.call(message) - end - end - end - - class << self - def clear_mocks - @mock_streams = {} - end - - def server - self - end - - def broadcast(stream_name, message) - stream = @mock_streams[stream_name] - stream && stream.mock_broadcast(message) - end - - def mock_stream_for(stream_name) - @mock_streams[stream_name] ||= MockStream.new - end - - def get_mock_channel - MockChannel.new - end - - def mock_stream_names - @mock_streams.keys - end - end - end - - class MockSchema < GraphQL::Schema - class << self - def find_by_gid(gid) - return unless gid - - if gid.model_class < ApplicationRecord - Gitlab::Graphql::Loaders::BatchModelLoader.new(gid.model_class, gid.model_id).find - elsif gid.model_class.respond_to?(:lazy_find) - gid.model_class.lazy_find(gid.model_id) - else - gid.find - end - end - - def id_from_object(object, _type = nil, _ctx = nil) - unless object.respond_to?(:to_global_id) - # This is an error in our schema and needs to be solved. So raise a - # more meaningful error message - raise "#{object} does not implement `to_global_id`. " \ - "Include `GlobalID::Identification` into `#{object.class}" - end - - object.to_global_id - end - end - - query(::Types::QueryType) - subscription(::Types::SubscriptionType) - - use GraphQL::Subscriptions::ActionCableSubscriptions, action_cable: MockActionCable, action_cable_coder: JSON - end - end - end -end diff --git a/spec/support/graphql/subscriptions/action_cable/mock_gitlab_schema.rb b/spec/support/graphql/subscriptions/action_cable/mock_gitlab_schema.rb deleted file mode 100644 index cd5d78cc78b..00000000000 --- a/spec/support/graphql/subscriptions/action_cable/mock_gitlab_schema.rb +++ /dev/null @@ -1,41 +0,0 @@ -# frozen_string_literal: true - -# A stub implementation of ActionCable. -# Any methods to support the mock backend have `mock` in the name. -module Graphql - module Subscriptions - module ActionCable - class MockGitlabSchema < GraphQL::Schema - class << self - def find_by_gid(gid) - return unless gid - - if gid.model_class < ApplicationRecord - Gitlab::Graphql::Loaders::BatchModelLoader.new(gid.model_class, gid.model_id).find - elsif gid.model_class.respond_to?(:lazy_find) - gid.model_class.lazy_find(gid.model_id) - else - gid.find - end - end - - def id_from_object(object, _type = nil, _ctx = nil) - unless object.respond_to?(:to_global_id) - # This is an error in our schema and needs to be solved. So raise a - # more meaningful error message - raise "#{object} does not implement `to_global_id`. " \ - "Include `GlobalID::Identification` into `#{object.class}" - end - - object.to_global_id - end - end - - query(::Types::QueryType) - subscription(::Types::SubscriptionType) - - use GraphQL::Subscriptions::ActionCableSubscriptions, action_cable: MockActionCable, action_cable_coder: JSON - end - end - end -end |