tests

Author: iskyd

test/constraints.jl

@@ -21,4 +21,15 @@
     @test constraints[2].fields == Wasabi.foreign_keys(Role)[1].fields
     @test constraints[2].foreign_table == Wasabi.foreign_keys(Role)[1].foreign_table
     @test constraints[2].foreign_fields == Wasabi.foreign_keys(Role)[1].foreign_fields
+
+    constraints = Wasabi.constraints(UserProfile)
+    @test length(constraints) == 3
+    @test constraints[1] isa Wasabi.PrimaryKeyConstraint
+    @test constraints[1].fields == Wasabi.primary_key(UserProfile).fields
+    @test constraints[2] isa Wasabi.ForeignKeyConstraint
+    @test constraints[2].fields == Wasabi.foreign_keys(UserProfile)[1].fields
+    @test constraints[2].foreign_table == Wasabi.foreign_keys(UserProfile)[1].foreign_table
+    @test constraints[2].foreign_fields == Wasabi.foreign_keys(UserProfile)[1].foreign_fields
+    @test constraints[3] isa Wasabi.UniqueConstraint
+    @test constraints[3].fields == Wasabi.unique_constraints(UserProfile)[1].fields
 end

test/runtests.jl

@@ -36,29 +36,13 @@ mutable struct User <: Wasabi.Model
     User(id::Integer, name::String, created_at::DateTime, roles::Vector{Role} = Role[]) = new(id, name, created_at, roles, nothing)
 end
 
-# struct UserProfile <: Wasabi.Model
-#     id::Int
-#     user_id::Int
-#     bio::Union{String,Nothing}
-# end
-
-# struct UserPhone <: Wasabi.Model
-#     id::Int
-#     user_profile_id::Int
-#     phone::String
-# end
-
 Wasabi.primary_key(m::Type{User}) = Wasabi.PrimaryKeyConstraint(Symbol[:id])
 Wasabi.primary_key(m::Type{Role}) = Wasabi.PrimaryKeyConstraint(Symbol[:id])
 Wasabi.foreign_keys(m::Type{Role}) = [Wasabi.ForeignKeyConstraint(Symbol[:user_id], :user, Symbol[:id])]
+Wasabi.foreign_keys(m::Type{UserProfile}) = [Wasabi.ForeignKeyConstraint(Symbol[:user_id], :user, Symbol[:id])]
+Wasabi.unique_constraints(m::Type{UserProfile}) = [Wasabi.UniqueConstraint(Symbol[:user_id])]
 Wasabi.exclude_fields(m::Type{User}) = [:roles, :profile]
 
-# Wasabi.primary_key(m::Type{UserProfile}) = Wasabi.PrimaryKeyConstraint(Symbol[:id])
-
-# Wasabi.foreign_keys(m::Type{UserPhone}) = [Wasabi.ForeignKeyConstraint(Symbol[:user_profile_id], :user_profile, Symbol[:id])]
-
-# Wasabi.unique_constraints(m::Type{UserProfile}) = [Wasabi.UniqueConstraint(Symbol[:user_id])]
-
 enabled_tests = lowercase.(ARGS)
 function addtests(fname)
     key = lowercase(splitext(fname)[1])