Beispiel #1
0
    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)
Beispiel #2
0
    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)