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)
