unit test updates for Gmail labels

--HG-- branch : stable
davebrown · Jun 7, 2013 · 1bbf1e3 · 1bbf1e3
1 parent ec5678d
commit 1bbf1e3
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/imapclient/test/test_imapclient.py b/imapclient/test/test_imapclient.py
@@ -418,15 +418,15 @@ def test_get(self):
 
     def test_add(self):
         self.client.add_gmail_labels(sentinel.messages, sentinel.labels)
-        self.client._store.assert_called_with('+X-GM-LABELS', sentinel.messages, sentinel.labels)
+        self.client._store.assert_called_with('+X-GM-LABELS', sentinel.messages, sentinel.labels, 'X-GM-LABELS')
 
     def test_remove(self):
         self.client.remove_gmail_labels(sentinel.messages, sentinel.labels)
-        self.client._store.assert_called_with('-X-GM-LABELS', sentinel.messages, sentinel.labels)
+        self.client._store.assert_called_with('-X-GM-LABELS', sentinel.messages, sentinel.labels, 'X-GM-LABELS')
 
     def test_set(self):
         self.client.set_gmail_labels(sentinel.messages, sentinel.labels)
-        self.client._store.assert_called_with('X-GM-LABELS', sentinel.messages, sentinel.labels)
+        self.client._store.assert_called_with('X-GM-LABELS', sentinel.messages, sentinel.labels, 'X-GM-LABELS')
 
 
 class TestNamespace(IMAPClientTest):