From b5ec96850f678e1ec1f828bf23a23fbb13f652ff Mon Sep 17 00:00:00 2001 From: Simon de Haan Date: Tue, 12 Nov 2013 09:03:35 +0200 Subject: [PATCH] test for OptOutStore.count() --- go/vumitools/opt_out/tests/test_models.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/go/vumitools/opt_out/tests/test_models.py b/go/vumitools/opt_out/tests/test_models.py index 2dd778756..793632e2c 100644 --- 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)