silverstripe / silverstripe-comments

                                $item->Comments()->getForeignClass());

    }

    public function testAddNonComment()

    {

        $item = $this->objFromFixture('CommentableItem', 'first');

        $comments = $item->Comments();

        $this->assertEquals(4, $comments->count());

        $member = Member::get()->first();

        try {

            $comments->add($member);

            $this->fail('Should not have been able to add member to comments');

        } catch (InvalidArgumentException $e) {

            $this->assertEquals(

                'CommentList::add() expecting a Comment object, or ID value',

                $e->getMessage()

            );

        }

    }

    public function testAddComment()

        $this->assertEquals(2, $item->Comments()->count());

    }

    public function testRemoveNonComment()

    {

        $item = $this->objFromFixture('CommentableItem', 'first');

        $this->assertEquals(4, $item->Comments()->count());

        $comments = $item->Comments();

        // try and remove a non comment

        $member = Member::get()->first();

        try {

            $comments->remove($member);

            $this->fail('Should not have been able to remove member from comments');

        } catch (InvalidArgumentException $e) {

            $this->assertEquals(

                'CommentList::remove() expecting a Comment object, or ID',

                $e->getMessage()

            );

        }

    }

}