Test ambiguities in package extension, Part 1

Author: lgoettgens

src/ambiguities.jl

@@ -101,10 +101,42 @@ function _find_ambiguities(
     packages::Vector{PkgId};
     color::Union{Bool,Nothing} = nothing,
     exclude::AbstractVector = [],
+    extensions::AbstractVector = [],
     # Options to be passed to `Test.detect_ambiguities`:
     detect_ambiguities_options...,
 )
+    extdeppackages = PkgId[]
+    @static if JULIA_HAS_EXTENSIONS
+        extensions = String.(extensions)
+        for ext in extensions
+            found = false
+            for pkg in setdiff(packages, [PkgId(Base), PkgId(Core)])
+                exts, weakdeps, deps = get_extension_data_from_toml(pkg)
+                if haskey(exts, ext)
+                    found = true
+                    extdeps = exts[ext] isa String ? [exts[ext]] : exts[ext]
+                    for extdepname in extdeps
+                        if haskey(deps, extdepname)
+                            push!(extdeppackages, deps[extdepname])
+                        elseif haskey(weakdeps, extdepname)
+                            push!(extdeppackages, weakdeps[extdepname])
+                        else
+                            error(
+                                "Extension $ext depends on $extdepname, but it is not found.",
+                            )
+                        end
+                    end
+                    push!(packages, PkgId(Base.uuid5(pkg.uuid, ext), ext))
+                    break
+                end
+            end
+            found && continue
+            error("Extension $ext is not found.")
+        end
+    end
+
     packages_repr = reprpkgids(collect(packages))
+    extdeppackages_repr = reprpkgids(collect(extdeppackages))
     options_repr = checked_repr((; recursive = true, detect_ambiguities_options...))
     exclude_repr = reprexclude(normalize_and_check_exclude(exclude))
 
@@ -115,6 +147,7 @@ function _find_ambiguities(
     using Aqua
     Aqua.test_ambiguities_impl(
         $packages_repr,
+        $extdeppackages_repr,
         $options_repr,
         $exclude_repr,
     ) || exit(1)
@@ -200,9 +233,11 @@ end
 
 function test_ambiguities_impl(
     packages::Vector{PkgId},
+    extdeppackages::Vector{PkgId},
     options::NamedTuple,
     exspecs::Vector{ExcludeSpec},
 )
+    deps = map(Base.require, extdeppackages)
     modules = map(Base.require, packages)
     @debug "Testing method ambiguities" modules
     ambiguities = detect_ambiguities(modules...; options...)

src/utils.jl

@@ -96,6 +96,22 @@ catch
     end
 end
 
+function get_extension_data_from_toml(pkg::PkgId)
+    root_project_path = root_project_or_failed_lazytest(pkg)
+    root_project_path isa LazyTestResult && return nothing, nothing
+
+    @debug "Parsing `$root_project_path`"
+    prj = TOML.parsefile(root_project_path)
+    raw_exts = get(prj, "extensions", nothing)
+
+    raw_weakdeps = get(prj, "weakdeps", nothing)
+    weakdeps = Dict(name => PkgId(UUID(uuid), name) for (name, uuid) in raw_weakdeps)
+
+    raw_deps = get(prj, "deps", nothing)
+    deps = Dict(name => PkgId(UUID(uuid), name) for (name, uuid) in raw_deps)
+    return raw_exts, weakdeps, deps
+end
+
 const _project_key_order = [
     "name",
     "uuid",