summaryrefslogtreecommitdiff
path: root/spec/family_tree_spec.rb
blob: 90d08b36b38c9d8bade631467bbb56631a307b17 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
# frozen_string_literal: true

require_relative '../lib/family_tree'
require_relative '../lib/person'
require_relative '../lib/family'

RSpec.describe FamilyTree do
  let(:mother) { Person.new('Jane', Gender::FEMALE) }
  let(:father) { Person.new('John', Gender::MALE) }
  let(:child1) { Person.new('Anna', Gender::FEMALE) }
  let(:child2) { Person.new('Bob', Gender::MALE) }
  let(:child3) { Person.new('Charlie', Gender::MALE) }
  let(:maternal_aunt) { Person.new('Alice', Gender::FEMALE) }
  let(:paternal_aunt) { Person.new('Catherine', Gender::FEMALE) }
  let(:maternal_uncle) { Person.new('Mark', Gender::MALE) }
  let(:paternal_uncle) { Person.new('David', Gender::MALE) }

  let(:family) { Family.new(mother, father, [child1, child2]) }

  before do
    FamilyTree.instance.families.clear
    FamilyTree.instance.add_family(Family.new(NilPerson.new, NilPerson.new, [mother, maternal_aunt, maternal_uncle]))
    FamilyTree.instance.add_family(Family.new(NilPerson.new, NilPerson.new, [father, paternal_aunt, paternal_uncle]))
    FamilyTree.instance.add_family(family)
  end

  describe '#add_family' do
    it 'adds a family to the family list' do
      expect(FamilyTree.instance.families).to include(family)
    end

    it 'does not add a duplicate family' do
      expect { FamilyTree.instance.add_family(family) }.not_to(change { FamilyTree.instance.families.count })
    end
  end

  describe '#add_child' do
    context 'when the mother is present' do
      it 'successfully adds a child to the family' do
        result = FamilyTree.instance.add_child(mother.name, 'Charlie', Gender::MALE)
        expect(result).to eq('CHILD_ADDED')
        expect(family.children.last.name).to eq('Charlie')
      end

      it 'does not add a duplicate child' do
        FamilyTree.instance.add_child(mother.name, 'Charlie', Gender::MALE)
        result = FamilyTree.instance.add_child(mother.name, 'Charlie', Gender::MALE)
        expect(result).to eq('CHILD_ADDITION_FAILED')
        expect(family.children.count { |child| child.name.casecmp('Charlie').zero? }).to eq(1)
      end
    end

    context 'when the mother is not present' do
      it 'fails to add a child and returns CHILD_ADDITION_FAILED' do
        FamilyTree.instance.families.clear # Clear existing families
        result = FamilyTree.instance.add_child('Unknown Mother', 'Charlie', Gender::MALE)
        expect(result).to eq('CHILD_ADDITION_FAILED')
      end
    end
  end

  describe '#get_relationship' do
    context 'finding parents' do
      it 'returns the mother\'s name if present' do
        expect(FamilyTree.instance.get_relationship('Anna', 'mother')).to eq(mother.name)
      end

      it 'returns the father\'s name if present' do
        expect(FamilyTree.instance.get_relationship('Anna', 'father')).to eq(father.name)
      end

      it 'returns PERSON_NOT_FOUND if the person has no mother' do
        orphan_family = Family.new(NilPerson.new, father, [child1])
        FamilyTree.instance.families.clear
        FamilyTree.instance.add_family(orphan_family)

        expect(FamilyTree.instance.get_relationship('Anna', 'mother')).to eq('PERSON_NOT_FOUND')
      end

      it 'returns PERSON_NOT_FOUND if the person has no father' do
        orphan_family = Family.new(mother, NilPerson.new, [child1])
        FamilyTree.instance.families.clear
        FamilyTree.instance.add_family(orphan_family)

        expect(FamilyTree.instance.get_relationship('Anna', 'father')).to eq('PERSON_NOT_FOUND')
      end
    end

    context 'finding siblings' do
      it 'returns sibling\'s name if present' do
        expect(FamilyTree.instance.get_relationship('Anna', 'siblings')).to eq(child2.name)
      end

      it 'returns NONE if there are no siblings' do
        single_child_family = Family.new(mother, father, [child1])
        FamilyTree.instance.families.clear
        FamilyTree.instance.add_family(single_child_family)

        expect(FamilyTree.instance.get_relationship('Anna', 'siblings')).to eq('NONE')
      end
    end

    context 'finding children' do
      it 'returns all children when queried' do
        expect(FamilyTree.instance.get_relationship(father.name, 'child')).to eq("#{child1.name} #{child2.name}")
      end

      it 'returns only sons when queried' do
        FamilyTree.instance.add_child(mother.name, 'Charlie', Gender::MALE)
        expect(FamilyTree.instance.get_relationship(father.name, 'son')).to eq("#{child2.name} Charlie")
      end

      it 'returns only daughters when queried' do
        expect(FamilyTree.instance.get_relationship(father.name, 'daughter')).to eq(child1.name)
      end

      it 'returns NONE if there are no children' do
        single_child_family = Family.new(mother, father, [])
        FamilyTree.instance.families.clear
        FamilyTree.instance.add_family(single_child_family)

        expect(FamilyTree.instance.get_relationship(father.name, 'child')).to eq('NONE')
      end
    end

    context 'finding aunts and uncles' do
      it 'returns maternal aunts' do
        expect(FamilyTree.instance.get_relationship(child1.name, 'maternal-aunt')).to eq(maternal_aunt.name)
      end

      it 'returns paternal aunts' do
        expect(FamilyTree.instance.get_relationship(child1.name, 'paternal-aunt')).to eq(paternal_aunt.name)
      end

      it 'returns maternal uncles' do
        expect(FamilyTree.instance.get_relationship(child1.name, 'maternal-uncle')).to eq(maternal_uncle.name)
      end

      it 'returns paternal uncles' do
        expect(FamilyTree.instance.get_relationship(child1.name, 'paternal-uncle')).to eq(paternal_uncle.name)
      end
    end

    # TODO: Add tests for finding in-laws
    context 'finding in-laws' do
      it 'returns sister-in-law' do; end

      it 'returns brother-in-law' do; end
    end

    context 'invalid relationships' do
      it 'returns UNSUPPORTED_RELATIONSHIP for unsupported types' do
        expect(FamilyTree.instance.get_relationship('Anna', 'uncle')).to eq('UNSUPPORTED_RELATIONSHIP')
      end
    end
  end
end