test for OptOutStore.count()

praekeltfoundation · Nov 12, 2013 · b5ec968 · b5ec968
1 parent ec92899
commit b5ec968
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/go/vumitools/opt_out/tests/test_models.py b/go/vumitools/opt_out/tests/test_models.py
@@ -104,3 +104,10 @@ def test_opt_outs_for_addresses_with_no_addresses(self):
         opt_outs = yield self.opt_out_store.opt_outs_for_addresses(
             "msisdn", [])
         self.assertEqual(opt_outs, [])
+
+    @inlineCallbacks
+    def test_count(self):
+        store = self.opt_out_store
+        yield store.new_opt_out(
+            "msisdn", "+1234", self.msg_helper.make_inbound("inbound"))
+        self.assertEqual((yield store.count()), 1)