Test for custom code

Author: aminya

src/@aml/@aml_create/get_struct_xml_.jl

@@ -6,7 +6,7 @@ check function for the aml struct
 function get_struct_xmlchecker(struct_function, args_var)
     # aml Function
     if !ismissing(struct_function[1])
-        F=struct_function[1]
+        F = struct_function[1]
         struct_xmlchecker = quote
             if !( ($(esc(F)))($(esc.(args_var)...)) )
                 error("struct criteria function ($($(esc(F)))) isn't meet")

test/customcode.jl

@@ -0,0 +1,90 @@
+# Definition
+@aml mutable struct Student "student"
+
+    # add custom code to the begining of creator function
+    # the following automatically fills IQ based on the name
+    @creator begin
+        if occursin("smart", name)
+            name = replace(name, "-smart" => "") # remove smart from name
+            IQ = "smart"
+        elseif occursin("genius", name)
+            name = replace(name, "-genius" => "") # remove smart from name
+            IQ = "genius"
+        else
+            error("Give a smart student.")
+        end
+    end
+
+    name::String, "~"
+    GPA::Float64, "~"
+    IQ::UN{String} = nothing, att"~" # default to nothing, but filled automatically by first @cretor macro
+    # add custom code to the end of extractor function
+
+    log::UN{String} = nothing, "~"
+
+    @extractor begin
+        if GPA > 4.0
+            log = "A genius with a GPA of $GPA is found" # setting fields using @extractor only changes the field and not the xml code
+        end
+    end
+end
+
+@aml mutable struct MathClass "math-class"
+    @creator begin
+        GPAsum = 0.0
+        for student in students
+            GPAsum = GPAsum + student.GPA
+        end
+        average = GPAsum / length(students) # fill average field
+    end
+
+    students::Vector{Student}, "student"
+    average::UN{Float64} = nothing, "average" # calculated automatically
+end
+
+################################################################
+# Creation
+smarts = [Student(name = "Jack-smart", GPA = 2.0), Student(name = "Sara-genius", GPA = 5.0)]
+mathclass = MathClass(students = smarts)
+
+@testset "Custom Code - Creation" begin
+    @test mathclass.students[1].name == "Jack"
+    @test mathclass.students[2].name == "Sara"
+    @test mathclass.average == 3.5
+    @test stripall(@capture_out(pprint(mathclass))) == stripall("""
+    <math-class>
+        <student IQ="smart">
+          <name>Jack</name>
+          <GPA>2.0</GPA>
+        </student>
+        <student IQ="genius">
+          <name>Sara</name>
+          <GPA>5.0</GPA>
+        </student>
+        <average>3.5</average>
+      </math-class>
+    """)
+end
+
+################################################################
+# Extraction
+
+@testset "Custom Code - Extraction" begin
+    xml = parsexml("""
+    <math-class>
+        <student>
+          <name>Jack</name>
+          <GPA>2.0</GPA>
+        </student>
+        <student>
+          <name>Sara</name>
+          <GPA>5.0</GPA>
+        </student>
+        <average>3.5</average>
+      </math-class>
+    """)
+
+    mathclass = MathClass(xml)
+
+    @test mathclass.students[2].log == "A genius with a GPA of 5.0 is found"
+end

test/runtests.jl

@@ -3,9 +3,11 @@ using Test, Suppressor, DataFrames
 
 stripall(x::String) = replace(x, r"\s|\n"=>"")
 
+##
 include("struct_definition.jl")
 include("creator.jl")
 include("extractor.jl")
 include("tables.jl")
 include("xmlutils.jl")
 # include("musicxml/musicxml.jl")
+include("customcode.jl")