🐦 Fixed #320

fonsp · Aug 27, 2020 · f7ec03d · f7ec03d · fonsp · Aug 27, 2020
1 parent d4be9bf
commit f7ec03d
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/src/analysis/ExpressionExplorer.jl b/src/analysis/ExpressionExplorer.jl
@@ -267,9 +267,8 @@ end
 function explore!(ex::Expr, scopestate::ScopeState)::SymbolsState
     if ex.head == :(=)
         # Does not create scope
-
 
-        if ex.args[1] isa Expr && (ex.args[1].head == :call || ex.args[1].head == :where)
+        if ex.args[1] isa Expr && (ex.args[1].head == :call || ex.args[1].head == :where || (ex.args[1].head == :(::) && ex.args[1].args[1] isa Expr && ex.args[1].args[1].head == :call))
             # f(x, y) = x + y
             # Rewrite to:
             # function f(x, y) x + y end

diff --git a/test/ExpressionExplorer.jl b/test/ExpressionExplorer.jl
@@ -189,13 +189,16 @@ using Test
         @test testee(:(function f(y::Int64=a)::String string(y) end), [], [], [], [
             :f => ([:String, :Int64, :a], [], [:string], [])
         ])
+        @test testee(:(f(a::A)::C = a.a;), [], [], [], [
+            :f => ([:A, :C], [], [], [])
+        ])
         @test testee(:(function f(x::T; k=1) where T return x + 1 end), [], [], [], [
             :f => ([], [], [:+], [])
         ])
         @test testee(:(function f(x::T; k=1) where {T,S <: R} return x + 1 end), [], [], [], [
             :f => ([:R], [], [:+], [])
         ])
-        @test_broken testee(:(f(x)::String = x), [], [], [], [
+        @test testee(:(f(x)::String = x), [], [], [], [
             :f => ([:String], [], [], [])
         ])
         @test testee(:(MIME"text/html"), [], [], [Symbol("@MIME_str")], [])