diff --git a/refstack/tests/unit/test_api_utils.py b/refstack/tests/unit/test_api_utils.py index 5d249977..1e91b337 100644 --- a/refstack/tests/unit/test_api_utils.py +++ b/refstack/tests/unit/test_api_utils.py @@ -390,7 +390,7 @@ class APIUtilsTestCase(base.BaseTestCase): ('fake_test', const.USER): 'some_other_user', ('fake_test', const.SHARED_TEST_RUN): None, }.get(args) - self.assertEqual(None, api_utils.get_user_role('fake_test')) + self.assertIsNone(api_utils.get_user_role('fake_test')) self.assertRaises(exc.HTTPError, api_utils.enforce_permissions, 'fake_test', const.ROLE_USER) self.assertRaises(exc.HTTPError, api_utils.enforce_permissions, diff --git a/refstack/tests/unit/test_db.py b/refstack/tests/unit/test_db.py index 552de813..e8f70a1a 100644 --- a/refstack/tests/unit/test_db.py +++ b/refstack/tests/unit/test_db.py @@ -532,7 +532,7 @@ class DBBackendTestCase(base.BaseTestCase): # Test no key match. filtered.all.return_value = [] result = api.get_pubkey(key) - self.assertEqual(None, result) + self.assertIsNone(result) session.query.assert_called_once_with(mock_model) query.filter_by.assert_called_once_with(md5_hash=khash)