def test_get_latest_empty(self, mock_sql, mock_get_repoid): """Ensure __get_latest() returns None when results are empty.""" # pylint: disable=W0212,E1101 testdb = SktDb(self.database_file) mock_get_repoid.return_value = 1 mock_sql.connect().cursor().fetchone.return_value = None result = testdb._SktDb__get_latest('baserepo_id') self.assertIsNone(result)
def test_get_latest(self, mock_sql, mock_get_repoid): """Ensure __get_latest() returns a result.""" # pylint: disable=W0212,E1101 testdb = SktDb(self.database_file) mock_get_repoid.return_value = 1 mock_sql.connect().cursor().fetchone.return_value = [1] result = testdb._SktDb__get_latest('baserepo_id') self.assertEqual(result, 1)