fix serialization of like matcher

botstory · Apr 12, 2017 · 6a2ca2f · 6a2ca2f
1 parent b6d175e
commit 6a2ca2f
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 3 deletions.
diff --git a/botstory/middlewares/sticker/sticker.py b/botstory/middlewares/sticker/sticker.py
@@ -33,7 +33,9 @@ class Like:
     type = 'Sticker.Like'
 
     def __init__(self, size=ANY_LIKE):
-        if size == ANY_LIKE:
+        if isinstance(size, list):
+            self.valid_likes = size
+        elif size == ANY_LIKE:
             self.valid_likes = LIKE_STICKERS
         else:
             self.valid_likes = [size]
@@ -42,8 +44,8 @@ def validate(self, ctx):
         return get_sticker(ctx) in self.valid_likes
 
     def serialize(self):
-        return self.option
+        return self.valid_likes
 
     @staticmethod
     def deserialize(option):
-        return Match(option)
+        return Like(option)
diff --git a/botstory/middlewares/sticker/sticker_test.py b/botstory/middlewares/sticker/sticker_test.py
@@ -102,3 +102,10 @@ def then(ctx):
         })
 
         assert trigger_likes[match_size].is_triggered
+
+
+def test_serialize_like():
+    m_old = sticker.Like(sticker.BIG_LIKE)
+    m_new = matchers.deserialize(matchers.serialize(m_old))
+    assert isinstance(m_new, sticker.Like)
+    assert sticker.BIG_LIKE in m_new.valid_likes