diff --git a/spec/ruby-enum/enum_spec.rb b/spec/ruby-enum/enum_spec.rb index bd0f1c9..aa5a07a 100644 --- a/spec/ruby-enum/enum_spec.rb +++ b/spec/ruby-enum/enum_spec.rb @@ -158,6 +158,18 @@ class SecondSubclass < FirstSubclass it 'returns values' do expect(Colors.values).to eq(%w[red green]) end + + context 'when a subclass is defined' do + it 'returns all values' do + expect(FirstSubclass.values).to eq(%w[red green orange]) + end + end + + context 'when a subclass of a subclass is defined' do + it 'returns all values' do + expect(SecondSubclass.values).to eq(%w[red green orange pink]) + end + end end describe '#to_h' do