diff options
| -rw-r--r-- | spec/lesson_three/movie_library_spec.rb | 16 |
1 files changed, 7 insertions, 9 deletions
diff --git a/spec/lesson_three/movie_library_spec.rb b/spec/lesson_three/movie_library_spec.rb index 47254e4..d04fe50 100644 --- a/spec/lesson_three/movie_library_spec.rb +++ b/spec/lesson_three/movie_library_spec.rb @@ -6,19 +6,17 @@ class Movie end end - class MovieLibrary - attr_accessor :total_count - def initialize(total_count = 0) - @total_count = total_count - @array_of_movies = [] + def initialize(movies = []) + @movies = movies end def add(movie) - if !@array_of_movies.include? movie - @array_of_movies.push movie - @total_count+=1 - end + @movies.push(movie) unless @movies.include? movie + end + + def total_count + @movies.size end end |