Assert that we split up line items into the appropriate counts per pr…

…ovider
praekeltfoundation · Jan 30, 2015 · ea80dab · ea80dab
1 parent 89e1c12
commit ea80dab
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/go/billing/tests/test_tasks.py b/go/billing/tests/test_tasks.py
@@ -563,7 +563,10 @@ def test_generate_monthly_statement_providers_inbound(self):
             description__startswith='Messages received')
 
         self.assertEqual(item1.description, 'Messages received - Provider 1')
+        self.assertEqual(item1.units, 2)
+
         self.assertEqual(item2.description, 'Messages received - Provider 2')
+        self.assertEqual(item2.units, 1)
 
     @mock.patch('go.billing.settings.PROVIDERS', {'provider1': 'Provider 1'})
     def test_generate_monthly_statement_unknown_provider_inbound(self):
@@ -633,7 +636,10 @@ def test_generate_monthly_statement_providers_outbound(self):
             description__startswith='Messages sent')
 
         self.assertEqual(item1.description, 'Messages sent - Provider 1')
+        self.assertEqual(item1.units, 2)
+
         self.assertEqual(item2.description, 'Messages sent - Provider 2')
+        self.assertEqual(item2.units, 1)
 
     @mock.patch('go.billing.settings.PROVIDERS', {'provider1': 'Provider 1'})
     def test_generate_monthly_statement_unknown_provider_outbound(self):
@@ -703,10 +709,14 @@ def test_generate_monthly_statement_providers_sessions(self):
             item1.description,
             'Sessions (billed per session) - Provider 1')
 
+        self.assertEqual(item1.units, 2)
+
         self.assertEqual(
             item2.description,
             'Sessions (billed per session) - Provider 2')
 
+        self.assertEqual(item2.units, 1)
+
     @mock.patch('go.billing.settings.PROVIDERS', {'provider1': 'Provider 1'})
     def test_generate_monthly_statement_unknown_provider_sessions(self):
         mk_transaction(