Fix mysql and sqlite3 database option bug

lib/suspenders/app_builder.rb

@@ -148,6 +148,16 @@ def use_postgres_config_template
         force: true
     end
 
+    def add_database_to_gemfile(database)
+      if database == "sqlite3"
+        @generator.gem "sqlite3", "~> 1.3"
+      elsif database == "mysql"
+        @generator.gem "mysql2", "~> 0.5"
+      else
+        @generator.gem "pg", "~> 0.18"
+      end
+    end
+
     def create_database
       bundle_command "exec rails db:create db:migrate"
     end

lib/suspenders/generators/app_generator.rb

@@ -71,10 +71,8 @@ def customize_gemfile
     def setup_database
       say 'Setting up database'
 
-      if 'postgresql' == options[:database]
-        build :use_postgres_config_template
-      end
-
+      build :add_database_to_gemfile, options[:database]
+      build :use_postgres_config_template if options[:database] == "postgresql"
       build :create_database
     end
 

spec/features/database_spec.rb

@@ -0,0 +1,34 @@
+# frozen_string_literal: true
+
+require "spec_helper"
+
+RSpec.shared_context "Project Setup Hook" do |option|
+  before :context do
+    drop_dummy_database
+    remove_project_directory
+    run_suspenders option
+    setup_app_dependencies
+  end
+end
+
+RSpec.describe "Suspend a new project with custom database option" do
+  context "--database=sqlite3" do
+    include_context "Project Setup Hook", "--database=sqlite3"
+
+    it "adds sqlite3 to Gemfile" do
+      gemfile = IO.read "#{project_path}/Gemfile"
+
+      expect(gemfile).to match(/sqlite3/)
+    end
+  end
+
+  context "--database=mysql" do
+    include_context "Project Setup Hook", "--database=mysql"
+
+    it "adds mysql2 to Gemfile" do
+      gemfile = IO.read "#{project_path}/Gemfile"
+
+      expect(gemfile).to match(/mysql2/)
+    end
+  end
+end

templates/Gemfile.erb

@@ -11,7 +11,6 @@ gem "autoprefixer-rails"
 gem "flutie"
 gem "honeybadger"
 gem "jquery-rails"
-gem "pg", "~> 0.18"
 gem "puma"
 gem "rack-canonical-host"
 gem "rails", "<%= Suspenders::RAILS_VERSION %>"