Fix mysql and sqlite3 database option bug

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,13 @@ gem "autoprefixer-rails"
 gem "flutie"
 gem "honeybadger"
 gem "jquery-rails"
+<% if "sqlite3" == options[:database] %>
+gem "sqlite3", "~> 1.3"
+<% elsif "mysql" == options[:database] %>
+gem "mysql2", "~> 0.5.2"
+<% else %>
 gem "pg", "~> 0.18"
+<% end %>
 gem "puma"
 gem "rack-canonical-host"
 gem "rails", "<%= Suspenders::RAILS_VERSION %>"