def test_get_repoid(self, mock_sql): """Ensure __get_repoid() retrieves a repoid.""" # pylint: disable=W0212,E1101 testdb = SktDb(self.database_file) mock_sql.connect().cursor().fetchone.return_value = [1] result = testdb._SktDb__get_repoid('git://example.com/repo') self.assertEqual(result, 1)
def test_get_repoid_missing(self, mock_sql): """Ensure __get_repoid() creates a repoid 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_repoid('git://example.com/repo') self.assertEqual(result, 1)