spec/lib/generators/relation_generator_spec.rb in rom-rails-0.2.1 vs spec/lib/generators/relation_generator_spec.rb in rom-rails-0.3.0.beta1

@@ -1,33 +1,43 @@
 require 'spec_helper'
 
 require 'generators/rom/relation_generator'
 
-describe ROM::Generators::RelationGenerator do
+describe ROM::Generators::RelationGenerator, type: :generator do
   destination File.expand_path('../../../../tmp', __FILE__)
 
-  before(:all) do
+  before(:each) do
     prepare_destination
-    run_generator ['users']
   end
 
   specify do
+    run_generator ['users']
+    default_adapter = ROM.adapters.keys.first
+
     expect(destination_root).to have_structure {
       directory 'app' do
         directory 'relations' do
-          file 'users.rb' do
+          file 'users_relation.rb' do
             contains <<-CONTENT.strip_heredoc
-              ROM.relation(:users) do
+              class UsersRelation < ROM::Relation[:#{default_adapter}]
+                dataset :users
+
                 # define your methods here ie:
                 #
                 # def all
                 #   select(:id, :name).order(:id)
                 # end
-                #
               end
             CONTENT
           end
         end
       end
     }
+  end
+
+  specify "with given adapter" do
+    run_generator ['users', '--adapter=memory']
+
+    relation = File.read(File.join(destination_root, 'app', 'relations', 'users_relation.rb'))
+    expect(relation).to include("class UsersRelation < ROM::Relation[:memory]")
   end
 end