spec/nokogiri_spec.rb in genit-0.1 vs spec/nokogiri_spec.rb in genit-0.2

@@ -10,17 +10,20 @@
     @doc = Nokogiri::HTML(f)
     f.close
   end
 
   it "should retrieve all genit tag" do
-    @doc.css("body genit").size.should == 1
+    @doc.css("body genit").size.should == 2
   end
   
-  it "should retrieve a tag" do
-    @doc.at_css("body genit").to_s.should == '<genit class="pages"></genit>'
+  it "should retrieve class of tag" do
+    tags = @doc.css("body genit")
+    tags[0]['class'].should == "menu"
+    tags[1]['class'].should == "pages"
   end
   
-  it "should retrieve class of tag" do
-    @doc.at_css("body genit")['class'].should == "pages"
+  it "should add an attribute" do
+    @doc.at_css("body")['class'] = "main"
+    @doc.at_css("body")['class'].should == 'main'
   end
   
 end