fix: adapt asserts to ok/eq_

issue #674

autopush/tests/test_db.py

@@ -151,8 +151,8 @@ def test_provisioning(self):
         db_name = "storage_%s" % uuid.uuid4()
 
         s = create_storage_table(db_name, 8, 11)
-        assert s.throughput["read"] == 8
-        assert s.throughput["write"] == 11
+        eq_(s.throughput["read"], 8)
+        eq_(s.throughput["write"], 11)
 
     def test_dont_save_older(self):
         s = get_storage_table()
@@ -236,7 +236,7 @@ def test_register(self):
         # Verify its in the db
         rows = m.query_2(uaid__eq=self.uaid, chidmessageid__eq=" ")
         results = list(rows)
-        assert(len(results) == 1)
+        eq_(len(results), 1)
 
     def test_unregister(self):
         chid = str(uuid.uuid4())
@@ -247,15 +247,15 @@ def test_unregister(self):
         # Verify its in the db
         rows = m.query_2(uaid__eq=self.uaid, chidmessageid__eq=" ")
         results = list(rows)
-        assert(len(results) == 1)
+        eq_(len(results), 1)
         eq_(results[0]["chids"], set([chid]))
 
         message.unregister_channel(self.uaid, chid)
 
         # Verify its not in the db
         rows = m.query_2(uaid__eq=self.uaid, chidmessageid__eq=" ")
         results = list(rows)
-        assert(len(results) == 1)
+        eq_(len(results), 1)
         eq_(results[0]["chids"], None)
 
         # Test for the very unlikely case that there's no 'chid'
@@ -301,7 +301,7 @@ def test_all_channels_no_uaid(self):
         m = get_rotating_message_table()
         message = Message(m, SinkMetrics())
         exists, chans = message.all_channels(dummy_uaid)
-        assert(chans == set([]))
+        eq_(chans, set([]))
 
     def test_message_storage(self):
         chid = str(uuid.uuid4())
@@ -396,8 +396,8 @@ def test_provisioning(self):
         db_name = "router_%s" % uuid.uuid4()
 
         r = create_router_table(db_name, 3, 17)
-        assert r.throughput["read"] == 3
-        assert r.throughput["write"] == 17
+        eq_(r.throughput["read"], 3)
+        eq_(r.throughput["write"], 17)
 
     def test_no_uaid_found(self):
         uaid = str(uuid.uuid4())

autopush/tests/test_diagnostic_cli.py

@@ -37,7 +37,8 @@ def test_bad_endpoint(self):
             "--router_tablename=fred",
             "http://someendpoint",
         ])
-        assert cli.run()
+        returncode = cli.run()
+        ok_(returncode not in (None, 0))
 
     @patch("autopush.diagnostic_cli.AutopushSettings")
     def test_successfull_lookup(self, mock_settings_class):

autopush/tests/test_main.py

@@ -280,15 +280,17 @@ def test_ssl(self):
         ], False)
 
     def test_bad_senderidlist(self):
-        assert endpoint_main([
+        returncode = endpoint_main([
             "--gcm_enabled",
             "--senderid_list='[Invalid'"
         ], False)
+        ok_(returncode not in (None, 0))
 
     def test_bad_apnsconf(self):
-        assert endpoint_main([
+        returncode = endpoint_main([
             "--apns_creds='[Invalid'"
         ], False)
+        ok_(returncode not in (None, 0))
 
     def test_client_certs(self):
         cert = self.TestArg._client_certs['partner1'][0]