Refactoring.

trema · Aug 26, 2015 · 5e08fcd · 5e08fcd
1 parent d539a6a
commit 5e08fcd
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 6 deletions.
diff --git a/lib/pio/open_flow10/actions.rb b/lib/pio/open_flow10/actions.rb
@@ -34,8 +34,8 @@ class Actions < BinData::Primitive
 
       string :binary, read_length: :length
 
-      def set(value)
-        self.binary = [value].flatten.map(&:to_binary).join
+      def set(actions)
+        self.binary = Array(actions).map(&:to_binary).join
       end
 
       # rubocop:disable MethodLength

diff --git a/lib/pio/open_flow13/actions.rb b/lib/pio/open_flow13/actions.rb
@@ -17,8 +17,8 @@ class Actions < BinData::Primitive
 
     string :binary, read_length: :length
 
-    def set(value)
-      self.binary = [value].flatten.map(&:to_binary).join
+    def set(actions)
+      self.binary = Array(actions).map(&:to_binary).join
     end
 
     # rubocop:disable MethodLength

diff --git a/lib/pio/open_flow13/flow_mod.rb b/lib/pio/open_flow13/flow_mod.rb
@@ -69,8 +69,8 @@ class Instructions < BinData::Primitive
         count_bytes_remaining :instructions_length
         string :instructions, read_length: :instructions_length
 
-        def set(object)
-          self.instructions = [object].flatten.map(&:to_binary_s).join
+        def set(instructions)
+          self.instructions = Array(instructions).map(&:to_binary_s).join
         end
 
         # rubocop:disable MethodLength