Panda + Tiger + Eagle

lib/api.rb

@@ -3,22 +3,47 @@
 require "ostruct"
 require_relative "./movie"
 class Api
+ @@search_history = []
 
  APIKEY="4t6456xa33z8qhcqyuqgnkjh"
 
  def self.search_by_title(title)
- url = "http://api.rottentomatoes.com/api/public/v1.0/movies.json?apikey=#{APIKEY}&q=#{URI.encode(title)}&page_limit=1"
- struct = OpenStruct.new(get_url_as_json(url).fetch("movies").first)
- Movie.new(id: struct.id.to_i,
- title: struct.title,
- year: struct.year,
- score: struct.ratings["critics_score"]
- )
+ if title != "" then
+ url = "http://api.rottentomatoes.com/api/public/v1.0/movies.json?apikey=#{APIKEY}&q=#{URI.encode(title)}&page_limit=1"
+ struct = OpenStruct.new(get_url_as_json(url).fetch("movies").first)
+ if struct.ratings.nil? == false then
+ movie = Movie.new(id: struct.id.to_i,
+ title: struct.title,
+ year: struct.year,
+ score: struct.ratings["critics_score"]
+ )
+ @@search_history << movie
+ return movie
+ else
+ Movie.new(id: 0,
+ title: "NOTHINGFOUNDHERE",
+ year: 0,
+ score: 0
+ )
+ end
+ else
+ Movie.new(id: 0,
+ title: "NOTHINGFOUNDHERE",
+ year: 0,
+ score: 0
+ )
+ end
  end
 
-
  def self.get_url_as_json(url)
  JSON.parse(open(url).read)
  end
 
+ def self.get_search_history()
+ @@search_history
+ end
+
+ def self.reset_search_history()
+ @@search_history = []
+ end
 end

lib/search_history.rb

@@ -0,0 +1,68 @@
+require_relative "./movie"
+
+class Search_History
+ attr_reader :movies
+
+ def initialize()
+ @movies = []
+ end
+
+ def add_search(movie)
+ @movies << movie
+ end
+
+ def average_rating()
+ average = 0
+ @movies.each do |movie|
+ average += movie.score
+ end
+
+ return (average/@movies.count)
+ end
+
+ def average_year_for_ratings()
+ year = 0
+ @movies.each do |movie|
+ year += movie.year
+ end
+
+ return (year/@movies.count).ceil
+ end
+
+ def average_rating_for_year(year)
+ @rating = 0
+ movie_count = 0
+ @movies.each do |movie|
+ if movie.year == year then
+ @rating += movie.score
+ movie_count += 1
+ end
+ end
+
+ return (@rating/movie_count)
+ end
+
+ def average_rating_for_max_year
+ average_rating_for_year(max_year)
+ end
+
+ def average_rating_for_min_year
+ average_rating_for_year(min_year)
+ end
+
+ def max_year
+ @movies.max_by(&:year).year
+ end
+
+ def min_year
+ @movies.min_by(&:year).year
+ end
+
+ def ratings_trend
+ if @movies.count > 1 then
+ (average_rating_for_max_year - average_rating_for_min_year) / (max_year - min_year)
+ else
+ 0
+ end
+ end
+end

movie_json.rb

@@ -1,13 +1,23 @@
 require_relative "lib/movie"
 require_relative "lib/api"
+require_relative "lib/search_history"
 
 def find_movie
- puts "OH HAI. Search?"
+ puts "Add a movie you really like"
  movie_title = gets
  movie = Api.search_by_title(movie_title)
+ @search_history.add_search(movie)
  puts "Found: #{movie.title}. Score: #{movie.score}"
+ puts "Average movie score: #{@search_history.average_rating}."
+ puts "Average movie year: #{@search_history.average_year_for_ratings}."
+ if @search_history.ratings_trend > 0 then
+ puts "You are getting happier"
+ elsif @search_history.ratings_trend < 0 then
+ puts "You are getting madder"
+ end
 end
 
+@search_history = Search_History.new
 find_movie
 
 while true do

spec/api_spec.rb

@@ -2,26 +2,53 @@
 require "ostruct"
 
 describe Api do
+ context "Missing Movie" do
+ describe "Empty Search" do
+ it "should not throw an error" do
+ expect {
+ Api.search_by_title("NOTHINGFOUNDHERE")
+ }.to_not raise_error
+ end
+ it "should have the title 'NOTHINGFOUNDHERE'" do
+ movie = Api.search_by_title("")
+ movie.title.should eq("NOTHINGFOUNDHERE")
+ end
+ end
+ end
 
- let(:movie) { Api.search_by_title("Forrest Gump") }
+ context "Valid Movie" do
+ let(:movie) { Api.search_by_title("Forrest Gump") }
 
- before do
- Api.stub(:get_url_as_json) { JSON.parse(File.read("spec/fixtures/forrest.json")) }
- end
+  before do
+  Api.stub(:get_url_as_json) { JSON.parse(File.read("spec/fixtures/forrest.json")) }
+  end
 
- it "should search for movies" do
- movie.title.should eq("Forrest Gump")
- end
+  it "should search for movies" do
+  movie.title.should eq("Forrest Gump")
+  end
 
- it "should return the score" do
- movie.score.should eq(71)
- end
+ it "should return the score" do
+ movie.score.should eq(71)
+ end
+
+ it "should return the id" do
+ movie.id.should eq(10036)
+ end
 
- it "should return the id" do
- movie.id.should eq(10036)
+ it "should return the year" do
+ movie.year.should eq(1994)
+ end
+
+ it "should be able to reset the search history" do
+ Api.reset_search_history
+ Api.get_search_history.count.should eq(0)
+ end
  end
 
- it "should return the year" do
- movie.year.should eq(1994)
+ context "Search History" do
+ it "should store latest search history" do
+ movie = Api.search_by_title("Forrest Gump")
+ Api.get_search_history.count.should eq(1)
+ end 
  end
 end

spec/search_history_spec.rb

@@ -0,0 +1,68 @@
+require_relative "../lib/search_history"
+describe Search_History do
+ it "should store a search history" do
+ search_history = Search_History.new
+ search_history.add_search(Api.search_by_title("Forrest Gump"))
+ search_history.movies.count.should eq(1)
+ end
+
+ it "should store latest seach history" do
+ search_history = Search_History.new
+ search_history.add_search(Api.search_by_title("Forrest Gump"))
+ search_history.movies.last.title.should eq("Forrest Gump")
+ end
+
+ it "should get average rating for search history" do
+ search_history = Search_History.new
+ search_history.add_search(Api.search_by_title("Forrest Gump"))
+ search_history.add_search(Api.search_by_title("Joe Dirt"))
+ search_history.movies.count.should eq(2)
+ search_history.average_rating.should eq(41)
+ end
+
+ it "should have the average year for the ratings" do
+ search_history = Search_History.new
+ search_history.add_search(Api.search_by_title("Forrest Gump"))
+ search_history.add_search(Api.search_by_title("Joe Dirt"))
+ search_history.average_year_for_ratings.should eq(1997)
+ end
+
+ it "should have the average rating for the max year" do
+ search_history = Search_History.new
+ search_history.add_search(Api.search_by_title("Forrest Gump"))
+ search_history.add_search(Api.search_by_title("Joe Dirt"))
+ search_history.average_rating_for_max_year.should eq(11)
+ end
+
+ it "should have the average rating for the min year" do
+ search_history = Search_History.new
+ search_history.add_search(Api.search_by_title("Forrest Gump"))
+ search_history.add_search(Api.search_by_title("Joe Dirt"))
+ search_history.add_search(Api.search_by_title("Ace Ventura"))
+ search_history.average_rating_for_min_year.should eq(58)
+ end
+
+ it "should have the max year" do
+ search_history = Search_History.new
+ search_history.add_search(Api.search_by_title("Forrest Gump"))
+ search_history.add_search(Api.search_by_title("Joe Dirt"))
+ search_history.add_search(Api.search_by_title("Ace Ventura"))
+ search_history.max_year.should eq(2001)
+ end
+
+ it "should have the min year" do
+ search_history = Search_History.new
+ search_history.add_search(Api.search_by_title("Forrest Gump"))
+ search_history.add_search(Api.search_by_title("Joe Dirt"))
+ search_history.add_search(Api.search_by_title("Ace Ventura"))
+ search_history.min_year.should eq(1994)
+ end
+
+ it "should have the ratings trend" do
+ search_history = Search_History.new
+ search_history.add_search(Api.search_by_title("Forrest Gump"))
+ search_history.add_search(Api.search_by_title("Joe Dirt"))
+ search_history.add_search(Api.search_by_title("Ace Ventura"))
+ search_history.ratings_trend.should eq(-7)
+ end
+end