Skip to content

Commit

Permalink
for #1, makes test for data import
Browse files Browse the repository at this point in the history
  • Loading branch information
nanchenchen committed Feb 13, 2015
1 parent d7e7757 commit a988a7d
Show file tree
Hide file tree
Showing 2 changed files with 18 additions and 1 deletion.
18 changes: 18 additions & 0 deletions msgvis/apps/importer/tests.py
Original file line number Diff line number Diff line change
@@ -1,3 +1,21 @@
# -*- coding: utf-8 -*-
from django.test import TestCase
from models import Dataset, Message

from models import create_an_instance_from_json


# Create your tests here.
class ImportTest(TestCase):
def test_import_tweets(self):
"""Dataset.created_at should get set automatically."""
dset = Dataset.objects.create(name="Test Corpus", description="My Dataset")
result = create_an_instance_from_json(test_tweet, dset)
self.assertTrue(result)

msg = dset.message_set.filter(original_id="562057573306609665")
self.assertEquals(msg.count(), 1)



test_tweet = r"""{"contributors": null, "truncated": false, "text": "RT @vifubutojiwa: \uff13\u6708\u30fb\u30fb\u30fb\u5225\u308c\u306e\u5b63\u7bc0\u307e\u3067\u306b\u3001\n\u304d\u308c\u3044\u306b\u306a\u308a\u305f\u3044\u5973\u5b50\u96c6\u5408\u301c\uff01\uff01\n\u3084\u3063\u305f\u3089\u5f7c\u6c0f\u3067\u6765\u3061\u3083\u3063\u305f\u2661\n\n\u30b3\u30ec\u21d2http://t.co/ZOjvtpicy3\n\n\u544a\u767d\u3001\u5927\u6210\u529f\u3057\u305f\u3044\u4eba\uff01\n\u6025\u3052\u301c\u301c\u301c\u301c\uff01\uff01\uff01\nhttp://t.co/ZIXytozeHN", "in_reply_to_status_id": null, "id": 562057573306609665, "favorite_count": 0, "source": "<a href=\"http://yahoo.co.jp\" rel=\"nofollow\">\u3010\u753b\u50cf\u3011\u6e21\u8fba\u9ebb\u53cb\u306e\u904e\u6fc0\u3067\u30b8\u30e7\u30ea\u30b8\u30e7\u30ea\u306a\u30ef\u30ad\uff57\uff57\uff57\uff57\uff57\uff57</a>", "retweeted": false, "coordinates": null, "timestamp_ms": "1422839942658", "entities": {"symbols": [], "media": [{"source_status_id_str": "562042011570540544", "expanded_url": "http://twitter.com/vifubutojiwa/status/562042011570540544/photo/1", "display_url": "pic.twitter.com/ZIXytozeHN", "url": "http://t.co/ZIXytozeHN", "media_url_https": "https://pbs.twimg.com/media/B8zGOoFCUAAbIsM.jpg", "source_status_id": 562042011570540544, "id_str": "562042008093478912", "sizes": {"small": {"h": 232, "resize": "fit", "w": 340}, "large": {"h": 438, "resize": "fit", "w": 640}, "medium": {"h": 410, "resize": "fit", "w": 600}, "thumb": {"h": 150, "resize": "crop", "w": 150}}, "indices": [113, 135], "type": "photo", "id": 562042008093478912, "media_url": "http://pbs.twimg.com/media/B8zGOoFCUAAbIsM.jpg"}], "hashtags": [], "user_mentions": [{"id": 2537819460, "indices": [3, 16], "id_str": "2537819460", "screen_name": "vifubutojiwa", "name": "\u30e2\u30c6\u5973\u306b\u306a\u308b\u65b9\u6cd5"}], "trends": [], "urls": [{"url": "http://t.co/ZOjvtpicy3", "indices": [67, 89], "expanded_url": "http://bit.ly/15rFGhE", "display_url": "bit.ly/15rFGhE"}]}, "in_reply_to_screen_name": null, "id_str": "562057573306609665", "retweet_count": 0, "in_reply_to_user_id": null, "favorited": false, "retweeted_status": {"contributors": null, "truncated": false, "text": "\uff13\u6708\u30fb\u30fb\u30fb\u5225\u308c\u306e\u5b63\u7bc0\u307e\u3067\u306b\u3001\n\u304d\u308c\u3044\u306b\u306a\u308a\u305f\u3044\u5973\u5b50\u96c6\u5408\u301c\uff01\uff01\n\u3084\u3063\u305f\u3089\u5f7c\u6c0f\u3067\u6765\u3061\u3083\u3063\u305f\u2661\n\n\u30b3\u30ec\u21d2http://t.co/ZOjvtpicy3\n\n\u544a\u767d\u3001\u5927\u6210\u529f\u3057\u305f\u3044\u4eba\uff01\n\u6025\u3052\u301c\u301c\u301c\u301c\uff01\uff01\uff01\nhttp://t.co/ZIXytozeHN", "in_reply_to_status_id": null, "id": 562042067321241601, "favorite_count": 39, "source": "<a href=\"https://about.twitter.com/products/tweetdeck\" rel=\"nofollow\">TweetDeck</a>", "retweeted": false, "coordinates": null, "entities": {"symbols": [], "media": [{"source_status_id_str": "562042011570540544", "expanded_url": "http://twitter.com/vifubutojiwa/status/562042011570540544/photo/1", "display_url": "pic.twitter.com/ZIXytozeHN", "url": "http://t.co/ZIXytozeHN", "media_url_https": "https://pbs.twimg.com/media/B8zGOoFCUAAbIsM.jpg", "source_status_id": 562042011570540544, "id_str": "562042008093478912", "sizes": {"small": {"h": 232, "resize": "fit", "w": 340}, "large": {"h": 438, "resize": "fit", "w": 640}, "medium": {"h": 410, "resize": "fit", "w": 600}, "thumb": {"h": 150, "resize": "crop", "w": 150}}, "indices": [95, 117], "type": "photo", "id": 562042008093478912, "media_url": "http://pbs.twimg.com/media/B8zGOoFCUAAbIsM.jpg"}], "hashtags": [], "user_mentions": [], "trends": [], "urls": [{"url": "http://t.co/ZOjvtpicy3", "indices": [49, 71], "expanded_url": "http://bit.ly/15rFGhE", "display_url": "bit.ly/15rFGhE"}]}, "in_reply_to_screen_name": null, "id_str": "562042067321241601", "retweet_count": 17805, "in_reply_to_user_id": null, "favorited": false, "user": {"follow_request_sent": null, "profile_use_background_image": true, "default_profile_image": false, "id": 2537819460, "verified": false, "profile_image_url_https": "https://pbs.twimg.com/profile_images/549131945985056768/pJRlrXA4_normal.png", "profile_sidebar_fill_color": "DDEEF6", "profile_text_color": "333333", "followers_count": 417, "profile_sidebar_border_color": "C0DEED", "id_str": "2537819460", "profile_background_color": "C0DEED", "listed_count": 2, "profile_background_image_url_https": "https://abs.twimg.com/images/themes/theme1/bg.png", "utc_offset": 32400, "statuses_count": 3, "description": null, "friends_count": 8, "location": "", "profile_link_color": "0084B4", "profile_image_url": "http://pbs.twimg.com/profile_images/549131945985056768/pJRlrXA4_normal.png", "following": null, "geo_enabled": false, "profile_background_image_url": "http://abs.twimg.com/images/themes/theme1/bg.png", "name": "\u30e2\u30c6\u5973\u306b\u306a\u308b\u65b9\u6cd5", "lang": "ja", "profile_background_tile": false, "favourites_count": 0, "screen_name": "vifubutojiwa", "notifications": null, "url": null, "created_at": "Sat May 31 16:44:10 +0000 2014", "contributors_enabled": false, "time_zone": "Seoul", "protected": false, "default_profile": true, "is_translator": false}, "geo": null, "in_reply_to_user_id_str": null, "possibly_sensitive": false, "lang": "ja", "created_at": "Mon Feb 02 00:17:25 +0000 2015", "filter_level": "low", "in_reply_to_status_id_str": null, "place": null, "extended_entities": {"media": [{"source_status_id_str": "562042011570540544", "expanded_url": "http://twitter.com/vifubutojiwa/status/562042011570540544/photo/1", "display_url": "pic.twitter.com/ZIXytozeHN", "url": "http://t.co/ZIXytozeHN", "media_url_https": "https://pbs.twimg.com/media/B8zGOoFCUAAbIsM.jpg", "source_status_id": 562042011570540544, "id_str": "562042008093478912", "sizes": {"small": {"h": 232, "resize": "fit", "w": 340}, "large": {"h": 438, "resize": "fit", "w": 640}, "medium": {"h": 410, "resize": "fit", "w": 600}, "thumb": {"h": 150, "resize": "crop", "w": 150}}, "indices": [95, 117], "type": "photo", "id": 562042008093478912, "media_url": "http://pbs.twimg.com/media/B8zGOoFCUAAbIsM.jpg"}]}}, "user": {"follow_request_sent": null, "profile_use_background_image": true, "default_profile_image": false, "id": 2451370675, "verified": false, "profile_image_url_https": "https://pbs.twimg.com/profile_images/503810863896080385/JY4ZNhjq_normal.jpeg", "profile_sidebar_fill_color": "DDEEF6", "profile_text_color": "333333", "followers_count": 157, "profile_sidebar_border_color": "C0DEED", "id_str": "2451370675", "profile_background_color": "C0DEED", "listed_count": 5, "profile_background_image_url_https": "https://abs.twimg.com/images/themes/theme1/bg.png", "utc_offset": null, "statuses_count": 3521, "description": null, "friends_count": 76, "location": "", "profile_link_color": "0084B4", "profile_image_url": "http://pbs.twimg.com/profile_images/503810863896080385/JY4ZNhjq_normal.jpeg", "following": null, "geo_enabled": false, "profile_banner_url": "https://pbs.twimg.com/profile_banners/2451370675/1408952925", "profile_background_image_url": "http://abs.twimg.com/images/themes/theme1/bg.png", "name": "\uff78\uff7f\uff88\uff90\u307e\u3055\u3068\u3045\u30fc", "lang": "ja", "profile_background_tile": false, "favourites_count": 1568, "screen_name": "masanagi31", "notifications": null, "url": null, "created_at": "Fri Apr 18 13:25:16 +0000 2014", "contributors_enabled": false, "time_zone": null, "protected": false, "default_profile": true, "is_translator": false}, "geo": null, "in_reply_to_user_id_str": null, "possibly_sensitive": false, "lang": "ja", "created_at": "Mon Feb 02 01:19:02 +0000 2015", "filter_level": "low", "in_reply_to_status_id_str": null, "place": null, "extended_entities": {"media": [{"source_status_id_str": "562042011570540544", "expanded_url": "http://twitter.com/vifubutojiwa/status/562042011570540544/photo/1", "display_url": "pic.twitter.com/ZIXytozeHN", "url": "http://t.co/ZIXytozeHN", "media_url_https": "https://pbs.twimg.com/media/B8zGOoFCUAAbIsM.jpg", "source_status_id": 562042011570540544, "id_str": "562042008093478912", "sizes": {"small": {"h": 232, "resize": "fit", "w": 340}, "large": {"h": 438, "resize": "fit", "w": 640}, "medium": {"h": 410, "resize": "fit", "w": 600}, "thumb": {"h": 150, "resize": "crop", "w": 150}}, "indices": [113, 135], "type": "photo", "id": 562042008093478912, "media_url": "http://pbs.twimg.com/media/B8zGOoFCUAAbIsM.jpg"}]}}"""
1 change: 0 additions & 1 deletion msgvis/settings/test.py
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,6 @@
"PASSWORD": "",
"HOST": "",
"PORT": "",
'OPTIONS': {'charset': 'utf8mb4'},
},
}

Expand Down

0 comments on commit a988a7d

Please sign in to comment.