From 1b2a32762604910290a2e39728c193c4a1ddeb8c Mon Sep 17 00:00:00 2001
From: Juanito Fatas <juanito-fatas@cookpad.jp>
Date: Fri, 18 Aug 2017 00:54:52 +0900
Subject: [PATCH] Add a test for GitHub::SQL#models

---
 test/github/sql_test.rb | 26 ++++++++++++++++++++++++++
 1 file changed, 26 insertions(+)

diff --git a/test/github/sql_test.rb b/test/github/sql_test.rb
index 8a770fa..9493f44 100644
--- a/test/github/sql_test.rb
+++ b/test/github/sql_test.rb
@@ -187,4 +187,30 @@ class GitHub::SQLTest < Minitest::Test
       GitHub::SQL.run("DROP TABLE affected_rows_test")
     end
   end
+
+  DEFINE_REPOSITORY = -> { class Repository < ActiveRecord::Base; end }
+  def test_models
+    begin
+      ActiveRecord::Base.connection.execute("DROP TABLE IF EXISTS `repositories`")
+      ActiveRecord::Base.connection.execute <<-SQL
+        CREATE TABLE `repositories` (
+          `id` int(11) NOT NULL AUTO_INCREMENT,
+          `name` varchar(255) DEFAULT NULL,
+          `updated_at` datetime DEFAULT NULL,
+          `created_at` datetime DEFAULT NULL,
+          PRIMARY KEY (`id`)
+        ) ENGINE=InnoDB DEFAULT CHARSET=utf8;
+      SQL
+      DEFINE_REPOSITORY.call
+      Repository.create
+
+      repository = GitHub::SQL.new(<<-SQL).models(Repository).first
+        SELECT `repositories`.* FROM `repositories`
+      SQL
+
+      assert_kind_of Repository, repository
+    ensure
+      ActiveRecord::Base.connection.execute("DROP TABLE IF EXISTS `repositories`")
+    end
+  end
 end