JYiajsXL

1## The Code
2
3import facebook
4
5
6class SimpleFacebook:
7
8    def __init__(self, oauth_token):
9        self.graph = facebook.GraphAPI(oauth_token)
10
11    def post_message(self, message):
12        self.graph.put_object("me", "feed", message=message)
13
14
15## The Test
16
17import mock
18import unittest
19import facebook_post
20
21
22class FacebookPostTestCase(unittest.TestCase):
23
24    def test_post_message(self):
25        with mock.patch('facebook_post.facebook.GraphAPI', autospec=True) as mock_fb:
26            facebook_handle = facebook_post.SimpleFacebook('authtoken')
27            facebook_handle.post_message("Hi")
28            mock_fb.put_object.assert_called_with(message="Hi")
29
30
31if __name__ == "__main__":
32    unittest.main()