* add account query

* auto add an account for users without an account

Author: simonfranzen

app/graphql/resolvers/account.rb

@@ -0,0 +1,13 @@
+# frozen_string_literal: true
+
+module Resolvers
+  # Resolver to return a user
+  class Account < BaseResolver
+    type Types::AccountType, null: true
+    description 'Returns the account for the user.'
+
+    def resolve
+      context[:current_user] ? context[:current_user].account : nil
+    end
+  end
+end

app/graphql/types/account_type.rb

@@ -0,0 +1,8 @@
+# frozen_string_literal: true
+
+module Types
+  # GraphQL type for a user
+  class AccountType < BaseModel
+    field :name, String, null: false
+  end
+end

app/graphql/types/query_type.rb

@@ -8,10 +8,12 @@ class QueryType < BaseObject
 
     field :users, Types::UserType.connection_type, null: false
     def users(**_args)
-      ::User.accessible_by(current_ability)
+      ::User.accessible_by(current_ability).includes(:account)
     end
     field :user, resolver: Resolvers::User
 
+    field :account, resolver: Resolvers::Account
+
     def current_ability
       Ability.new(context[:current_user])
     end

app/graphql/types/user_type.rb

@@ -7,5 +7,6 @@ class UserType < BaseModel
     field :first_name, String, null: false
     field :last_name, String, null: false
     field :email, String, null: true
+    field :account, Types::AccountType, null: false
   end
 end

app/models/user.rb

@@ -64,6 +64,7 @@ class User < ApplicationRecord
 
   # - CALLBACKS
   after_initialize :setup_new_user, if: :new_record?
+  before_validation :setup_account
 
   # return first and lastname
   def name
@@ -79,6 +80,10 @@ def status_color
 
   private
 
+  def setup_account
+    self.account = Account.create!(name: 'My company') if account.nil?
+  end
+
   def setup_new_user
     self.role ||= :user
   end

db/seeds.rb

@@ -1,4 +1,4 @@
-user = User.create(
+user = User.create!(
   email: ENV['ADMIN_EMAIL'],
   password: ENV['ADMIN_PASSWORD'],
   password_confirmation: ENV['ADMIN_PASSWORD'],

spec/graphql/resolvers/account_spec.rb

@@ -0,0 +1,63 @@
+# frozen_string_literal: true
+
+require 'rails_helper'
+
+RSpec.describe Resolvers::Account, type: :request do
+  subject(:graphql!) { result }
+
+  let(:result) do
+    GraphqlSchema.execute(
+      query_string,
+      variables: variables,
+      context: context
+    )
+  end
+
+  let(:variables) do
+    {}
+  end
+
+  let(:query_string) do
+    <<-GRAPHQL
+    query {
+      account {
+        id
+        name
+      }
+    }
+    GRAPHQL
+  end
+
+  describe 'account' do
+    context 'when there\'s no current user' do
+      let(:context) do
+        {
+          current_user: nil
+        }
+      end
+
+      it 'returns nil' do
+        graphql!
+        expect(result['data']['account']).to eq(nil)
+      end
+    end
+
+    context 'when there\'s a current user' do
+      let!(:user) do
+        create(:user)
+      end
+
+      let(:context) do
+        {
+          current_user: user
+        }
+      end
+
+      it 'returns account of user' do
+        graphql!
+        account_id = result['data']['account']['id']
+        expect(account_id).to eq(user.account.id)
+      end
+    end
+  end
+end