X-Git-Url: http://gitweb.michael.orlitzky.com/?p=dead%2Fhalcyon.git;a=blobdiff_plain;f=src%2FTwitter%2FStatus.hs;h=506a2c0bbe26ff67a561e7eee6b2e73d0629fb9f;hp=191b8e85b734820bf73541ec9e55d621a0e0ba73;hb=e3864f89a0cae34d4d280efeb52ea5f761ddb44a;hpb=4cc476a2714260980899ca5358196bbf5226b3c2 diff --git a/src/Twitter/Status.hs b/src/Twitter/Status.hs index 191b8e8..506a2c0 100644 --- a/src/Twitter/Status.hs +++ b/src/Twitter/Status.hs @@ -1,29 +1,34 @@ {-# LANGUAGE NoMonomorphismRestriction #-} -- | Functions and data for working with Twitter statuses. -module Twitter.Status +module Twitter.Status ( + Status(..), + Timeline, + get_max_status_id, + pretty_print, + status_tests, + utc_time_to_rfc822 ) where -import Control.Applicative ((<$>), (<*>)) -import Control.Monad (liftM) -import Data.Aeson ((.:), FromJSON(..), Value(Object)) -import Data.Maybe (fromMaybe, mapMaybe, isJust) -import Data.Monoid (mempty) -import Data.String.Utils (join, splitWs) -import Data.Text (pack) -import Data.Time (formatTime) -import Data.Time.Clock (UTCTime) -import Data.Time.Format (parseTime) -import Data.Time.LocalTime (TimeZone, utcToZonedTime) -import System.Locale (defaultTimeLocale, rfc822DateFormat) -import Test.Framework (Test, testGroup) -import Test.Framework.Providers.HUnit (testCase) -import Test.HUnit (Assertion, assertEqual) -import Text.Regex (matchRegex, mkRegex) - -import Html (replace_entities) -import StringUtils (listify) -import Twitter.User +import Control.Applicative ( (<$>), (<*>) ) +import Control.Monad ( liftM ) +import Data.Aeson ( (.:), FromJSON(..), Value(Object) ) +import Data.Maybe ( mapMaybe, isJust ) +import Data.Monoid ( mempty ) +import Data.String.Utils ( join, splitWs ) +import Data.Text ( pack ) +import Data.Time ( formatTime ) +import Data.Time.Clock ( UTCTime ) +import Data.Time.Format ( parseTime ) +import Data.Time.LocalTime ( TimeZone, utcToZonedTime ) +import System.Locale ( defaultTimeLocale, rfc822DateFormat ) +import Test.Tasty ( TestTree, testGroup ) +import Test.Tasty.HUnit ( (@?=), testCase ) +import Text.Regex ( matchRegex, mkRegex ) + +import Html ( replace_entities ) +import StringUtils ( listify ) +import Twitter.User ( User(..), screen_name_to_timeline_url ) data Status = Status { created_at :: Maybe UTCTime, @@ -79,10 +84,11 @@ utc_time_to_rfc822 mtz utc = show_created_at :: Maybe TimeZone -> Status -> String show_created_at mtz = - (fromMaybe "") . (fmap $ utc_time_to_rfc822 mtz) . created_at + (maybe "" (utc_time_to_rfc822 mtz)) . created_at -- | Returns a nicely-formatted String representing the given 'Status' -- object. +-- pretty_print :: Maybe TimeZone -> Status -> String pretty_print mtz status = concat [ name, @@ -104,6 +110,7 @@ pretty_print mtz status = -- | Given a list of statuses, returns the greatest status_id -- belonging to one of the statuses in the list. +-- get_max_status_id :: Timeline -> Integer get_max_status_id statuses = maximum status_ids @@ -112,6 +119,7 @@ get_max_status_id statuses = -- | Parse one username from a word. +-- parse_username :: String -> Maybe String parse_username word = case matches of @@ -124,13 +132,16 @@ parse_username word = -- | Parse all usernames of the form \@username from a status. +-- parse_usernames_from_status :: Status -> [String] parse_usernames_from_status status = mapMaybe parse_username status_words where status_words = splitWs (text status) + -- | Get all referenced users' timeline URLs. +-- make_user_timeline_urls :: Status -> [String] make_user_timeline_urls status = map screen_name_to_timeline_url usernames @@ -138,20 +149,17 @@ make_user_timeline_urls status = usernames = parse_usernames_from_status status -status_tests :: Test +status_tests :: TestTree status_tests = - testGroup "Status Tests" [ tc1 ] - where - tc1 = testCase "All usernames are parsed." test_parse_usernames + testGroup "Status Tests" [ test_parse_usernames ] -test_parse_usernames :: Assertion +test_parse_usernames :: TestTree test_parse_usernames = - assertEqual - "All usernames are parsed." - expected_usernames - actual_usernames + testCase description $ actual @?= expected where + description = "all usernames are parsed" + dummy_user = User { screen_name = "nobody" } dummy_text = "Hypothesis: @donsbot and @bonus500 are two " ++ "personalities belonging to the same person." @@ -163,5 +171,5 @@ test_parse_usernames = retweeted = False } - actual_usernames = parse_usernames_from_status dummy_status - expected_usernames = ["donsbot", "bonus500"] + actual = parse_usernames_from_status dummy_status + expected = ["donsbot", "bonus500"]