import factory

from oyaji.address.models import Address


class AddressFactory(factory.mongoengine.MongoEngineFactory):
    class Meta:
        model = Address

    zip_code = factory.Sequence(lambda n: "3199999%d" % n)
    address = factory.Sequence(lambda n: "some street %d" % n)
    number = factory.Sequence(lambda n: "%d" % n)
    complement = factory.Sequence(lambda n: "some complement %d" % n)
    neighborhood = factory.Sequence(lambda n: "some neighborhood %d" % n)
    city = factory.Sequence(lambda n: "City %d" % n)
    city_id = factory.Sequence(lambda n: "%d" % n)
    state = factory.Sequence(lambda n: "State %d" % n)
    country = factory.Sequence(lambda n: "Country %d" % n)