def test_insert_or_update_batch_none_returned(self): """Test that None is returned when row count cannot be returned""" mock_connector = MagicMock() database = Database() database.connect(connector_impl=mock_connector) connection = mock_connector.connect() cursor = connection.cursor() cursor.rowcount = -1 cc_rows = [MagicMock(geo_id='CA', val=1, se=0, sample_size=0)] result = database.insert_or_update_batch(cc_rows) self.assertIsNone(result)
def test_insert_or_update_batch_row_count_returned(self): """Test that the row count is returned""" mock_connector = MagicMock() database = Database() database.connect(connector_impl=mock_connector) connection = mock_connector.connect() cursor = connection.cursor() cursor.rowcount = 3 cc_rows = [MagicMock(geo_id='CA', val=1, se=0, sample_size=0)] result = database.insert_or_update_batch(cc_rows) self.assertEqual(result, 3)