lifestream 0.1.1 → 0.1.2

data/VERSION

@@ -1 +1 @@
-0.1.1
+0.1.2

data/lib/lifestream/branch.rb

@@ -1,11 +1,16 @@
 module Lifestream
   class Branch
     
-    attr_accessor :published_at, :title, :body
+    attr_accessor :channel, :published_at, :title, :body
     
-    def initialize published_at, title, body = nil
-      @published_at, @title, @body = published_at, title, body
+    def initialize channel, published_at, title, body = nil
+      @channel, @published_at, @title, @body = channel, published_at, title, body
+      unless @channel.kind_of?(Lifestream::Channel)
+        raise Lifestream::Branch::InvalidBranch.new('@channel is not a kind if Lifestream::Channel') 
+      end
     end
     
+    class InvalidBranch < StandardError; end
+    
   end
 end

data/lib/lifestream/channel/rss2.rb

@@ -17,7 +17,7 @@ module Lifestream
     end
   
     def to_branch(branch)
-      Lifestream::Branch.new(branch.pubDate, branch.title, branch.description)
+      Lifestream::Branch.new(self, branch.pubDate, branch.title, branch.description)
     end
   
     class MalformedFeed < StandardError; end

data/test/lifestream/branch_test.rb

@@ -0,0 +1,22 @@
+require 'test_helper'
+
+class BranchTest < Test::Unit::TestCase
+  
+  context "A Lifestream::Branch instance" do
+    
+    setup { @lifestream = Lifestream.run }
+    
+    should "be aware of its parent channel" do
+      @lifestream.branches.first.channel.should_not be_nil
+      @lifestream.branches.first.channel.should be_a(Lifestream::Channel::Rss2)
+    end
+    
+    should "raise an exception if the channel is set to a non channel object" do
+      assert_raise Lifestream::Branch::InvalidBranch do
+        Lifestream::Branch.new nil, 'hi', 'hi', 'hi'
+      end
+    end
+    
+  end
+  
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification 
 name: lifestream
 version: !ruby/object:Gem::Version 
-  version: 0.1.1
+  version: 0.1.2
 platform: ruby
 authors: 
 - Adam Hunter