Added JSON fields.

NOX73 · Jul 14, 2013 · b320054 · b320054
1 parent 517fb3e
commit b320054
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 5 deletions.
diff --git a/twitter_api.go b/twitter_api.go
@@ -32,7 +32,13 @@ type Message struct {
 
 type TweetJSON struct {
   Text string
-  User struct {Id int; SreenName string}
+  User struct {
+    Id int
+    Screen_name string
+    Name string
+    Description string
+    Profile_image_url_https string
+  }
 }
 
 type Tweet struct {
@@ -52,11 +58,9 @@ func (t *Tweet) UserID() int {
 
 func (t *Tweet) UserName() string {
   if t.JSON == nil{ t.ParseJSON() }
-  return t.JSON.User.SreenName
+  return t.JSON.User.Screen_name
 }
 
-
-
 func (t *Tweet) ParseJSON() {
   t.JSON = &TweetJSON{}
   _ = json.Unmarshal([]byte(t.Body), t.JSON) 

diff --git a/twitter_api_test.go b/twitter_api_test.go
@@ -22,10 +22,10 @@ func (s *OAuthSuite) TestCreateClient(c *C) {
   go TwitterStream(ch, credentials, params)
 
   message := <- ch
+
   c.Assert(message.Response.StatusCode, Equals, 200)
   c.Assert(message.Tweet, NotNil)
   c.Assert(message.Tweet.Body, NotNil)
-  c.Assert(message.Tweet.Body, NotNil)
   c.Assert(message.Tweet.Text(), NotNil)
   c.Assert(message.Tweet.UserID(), NotNil)
   c.Assert(message.Tweet.UserName(), NotNil)