diff options
| -rw-r--r-- | lib/colonial_twilight/board/spaces.rb | 2 | ||||
| -rw-r--r-- | spec/mock_board.rb | 4 |
2 files changed, 5 insertions, 1 deletions
diff --git a/lib/colonial_twilight/board/spaces.rb b/lib/colonial_twilight/board/spaces.rb index 183a6a2..ee20127 100644 --- a/lib/colonial_twilight/board/spaces.rb +++ b/lib/colonial_twilight/board/spaces.rb @@ -99,7 +99,7 @@ module ColonialTwilight %i[gov gov_bases gov_cubes french_cubes algerian_cubes troops police french_troops french_police algerian_troops algerian_police - fln fln_bases guerrillas fln_underground fln_active max_bases].each do |sym| + fln fln_bases guerrillas fln_underground fln_active bases max_bases].each do |sym| define_method(sym) { @forces.send(sym) } end diff --git a/spec/mock_board.rb b/spec/mock_board.rb index 128c074..133c09e 100644 --- a/spec/mock_board.rb +++ b/spec/mock_board.rb @@ -21,6 +21,10 @@ class Sector @name == 'country' end + def bases + (@data[:fln_bases] || 0) + (@data[:gov_bases] || 0) + end + def max_bases 3 end |