add classroom association

stephenkati · Jun 21, 2023 · b6bba6a · b6bba6a
1 parent 6cc1ed3
commit b6bba6a
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/student.rb b/student.rb
@@ -1,6 +1,8 @@
 require_relative 'person'
 
 class Student < Person
+  attr_accessor :name, :classroom
+
   def initialize(age, classroom, name: 'Unknown', parent_permission: true)
     super(age, parent_permission: parent_permission, name: name)
     @classroom = classroom
@@ -9,4 +11,10 @@ def initialize(age, classroom, name: 'Unknown', parent_permission: true)
   def play_hooky
     '¯\\(ツ)/¯'
   end
+
+  def new_classroom=(classroom)
+    @classroom&.remove_student(self)
+    @classroom = classroom
+    classroom&.add_student(self)
+  end
 end