! Copyright (C) 2009 Joe Groff.
! See https://factorcode.org/license.txt for BSD license.
USING: accessors classes.tuple compiler.units kernel qw roles sequences
tools.test ;
IN: roles.tests

ROLE: fork tines ;
ROLE: spoon bowl ;
ROLE: instrument tone ;
ROLE: tuning-fork <{ fork instrument } volume ;

ROLE-TUPLE: utensil handle ;

! role consumption and tuple inheritance can be mixed
ROLE-TUPLE: foon <{ utensil fork spoon } ;
ROLE-TUPLE: tuning-spork <{ utensil spoon tuning-fork } ;

! role class testing
{ t } [ fork role? ] unit-test
{ f } [ foon role? ] unit-test

! roles aren't tuple classes by themselves and can't be instantiated
{ f } [ fork tuple-class? ] unit-test
[ fork new ] must-fail

! tuples which consume roles fall under their class
{ t } [ foon new fork? ] unit-test
{ t } [ foon new spoon? ] unit-test
{ f } [ foon new tuning-fork? ] unit-test
{ f } [ foon new instrument? ] unit-test

{ t } [ tuning-spork new fork? ] unit-test
{ t } [ tuning-spork new spoon? ] unit-test
{ t } [ tuning-spork new tuning-fork? ] unit-test
{ t } [ tuning-spork new instrument? ] unit-test

! consumed role slots are placed in tuples in order
{ qw{ handle tines bowl } } [ foon all-slots [ name>> ] map ] unit-test
{ qw{ handle bowl tines tone volume } } [ tuning-spork all-slots [ name>> ] map ] unit-test

! can't combine roles whose slots overlap
ROLE: bong bowl ;
SYMBOL: spong

[ [ spong { spoon bong } { } define-tuple-class-with-roles ] with-compilation-unit ]
[ role-slot-overlap? ] must-fail-with

[ [ spong { spoon bong } { } define-role ] with-compilation-unit ]
[ role-slot-overlap? ] must-fail-with

! can't try to inherit multiple tuple classes
ROLE-TUPLE: tool blade ;
SYMBOL: knife

[ knife { utensil tool } { } define-tuple-class-with-roles ]
[ multiple-inheritance-attempted? ] must-fail-with

! make sure method dispatch works
GENERIC: poke ( pokee poker -- result )
GENERIC: scoop ( scoopee scooper -- result )
GENERIC: tune ( tunee tuner -- result )

M: fork poke drop " got poked" append ;
M: spoon scoop drop " got scooped" append ;
M: instrument tune drop " got tuned" append ;

{ "potato got poked" "potato got scooped" "potato got tuned" }
[ "potato" tuning-spork new [ poke ] [ scoop ] [ tune ] 2tri ] unit-test