Added Link#id utility method.

cvut · Jul 17, 2014 · 5f00703 · 5f00703
1 parent cffbf5d
commit 5f00703
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 2 deletions.
diff --git a/lib/kosapi_client/entity/link.rb b/lib/kosapi_client/entity/link.rb
@@ -14,6 +14,11 @@ def self.parse(contents, context = {})
         href = contents[:xlink_href] || contents[:href]
         new(contents[:__content__], href, contents[:rel])
       end
+
+      def id
+        @href.split('/').last
+      end
+
     end
   end
 end
diff --git a/spec/kosapi_client/entity/link_spec.rb b/spec/kosapi_client/entity/link_spec.rb
@@ -4,16 +4,24 @@
 
   Link = KOSapiClient::Entity::Link
 
-  subject(:link) { Link.parse({href: 'http://example.com', __content__: 'Example Site', rel: 'next'}) }
+  subject(:link) { Link.parse({href: 'http://example.com/foo/bar/42', __content__: 'Example Site', rel: 'next'}) }
 
   describe '.parse' do
 
     it 'can be parsed from a hash' do
-      expect(link.href).to eq 'http://example.com'
+      expect(link.href).to eq 'http://example.com/foo/bar/42'
       expect(link.title).to eq 'Example Site'
       expect(link.rel).to eq 'next'
     end
 
   end
 
+  describe '#id' do
+
+    it 'returns last href segment' do
+      expect(link.id).to eq '42'
+    end
+
+  end
+
 end