def test_service_posts(self):
# create services
s1 = Service(parent_id=1, title='plumbing')
s2 = Service(parent_id=2, title='electrical')
db.session.add(s1)
db.session.add(s2)
db.session.commit()
# create four posts
now = datetime.utcnow()
p1 = Post(body="post from john",
timestamp=now + timedelta(seconds=1),
service_id=1)
p2 = Post(body="post from john",
timestamp=now + timedelta(seconds=1),
service_id=1)
p3 = Post(body="post from john",
timestamp=now + timedelta(seconds=1),
service_id=2)
db.session.add_all([p1, p2, p3])
db.session.commit()
# check service_posts function
check1 = s1.service_posts()
check2 = s2.service_posts()
# check that the correct posts are returned
self.assertEqual(check1.count(), 2)
self.assertEqual(check2.count(), 1)
