def test_get_sourceid(self, mock_sql): """Ensure __get_sourceid() retrieves a patchsource id.""" # pylint: disable=W0212,E1101 testdb = SktDb(self.database_file) mock_sql.connect().cursor().fetchone.return_value = [1] result = testdb._SktDb__get_sourceid('git://example.com/repo', 10) self.assertEqual(result, 1)
def test_get_sourceid_missing(self, mock_sql): """Ensure __get_sourceid() creates patchsource when it doesn't exist.""" # pylint: disable=W0212,E1101 testdb = SktDb(self.database_file) mock_sql.connect().cursor().fetchone.return_value = None mock_sql.connect().cursor().lastrowid = 1 result = testdb._SktDb__get_sourceid('git://example.com/repo', 10) self.assertEqual(result, 1)