31_1_variant_without_parameter.py

import smartpy as sp


@sp.module
def main():
    
    drink_type:type = sp.variant(Coca = sp.unit, Fanta = sp.unit, SevenUp = sp.unit, Water = sp.unit)
    
    class Restaurant(sp.Contract):
        def __init__(self):
            self.data.items = []            

        @sp.entrypoint
        def orderDrink(self, drink):
            sp.cast(drink, drink_type)
            price = sp.tez(0)
            with sp.match(drink):
                with sp.case.Coca:
                    price = sp.tez(1)
                with sp.case.Fanta:
                    price = sp.tez(2)
                with sp.case.Water:
                    price = sp.tez(0)
                # etc.
            assert sp.amount == price
            self.data.items.push(drink)
            

@sp.add_test()
def test():
    scenario = sp.test_scenario("Test", main)
    c1 = main.Restaurant()
    scenario += c1
    scenario.h3("J'ai faim")
    c1.orderDrink(sp.variant("Water", ()))
    c1.orderDrink(sp.variant("Coca", ()), _valid = False)
    c1.orderDrink(sp.variant("Coca", ()), _amount = sp.tez(1), _valid = True)    
    c1.orderDrink(sp.variant("Fanta", ()), _valid = False)
    c1.orderDrink(sp.variant("Fanta", ()), _amount = sp.tez(2), _valid = True)