Detect custom unit class

[qqqbbb]

I have this code in my mod

if pUnit:GetUnitClassType() == GameInfoTypes.UNITCLASS_TEST then 
	print("custom unit class")
end

but it does not fire for some reason. The new unit class is in database, game spawns units of this class without problems. Any ideas?

 

[LeeS]

Without knowing the larger context within which the quoted code occurs it's impossible to say, really.

 

[qqqbbb]

This is my code:

Events.RunCombatSim.Add (function (attackingPlayer, attackingUnit, attackingUnitDamage, attackingUnitFinalDamage, attackingUnitMaxHitPoints, defendingPlayer, defendingUnit, defendingUnitDamage, defendingUnitFinalDamage, defendingUnitMaxHitPoints)
    local pAttackingPlayer = Players[attackingPlayer];
    local pDefendingPlayer = Players[defendingPlayer];
    local pAttackUnit = pAttackingPlayer:GetUnitByID(attackingUnit);
    local pDefendUnit = pDefendingPlayer:GetUnitByID(defendingUnit);

    if pDefendUnit:GetUnitClassType() == GameInfoTypes.UNITCLASS_TEST then 
        print("custom unit class");
    end
end )

If I replace UNITCLASS_TEST with any vanilla unit class it works but it does not with my custom unit class.