diff --git a/util/seed/user.go b/util/seed/user.go index a1be9c6..0651765 100644 --- a/util/seed/user.go +++ b/util/seed/user.go @@ -18,6 +18,8 @@ const ( var osoba *model.User var admin *model.User +var hak *model.User +var vehicle *model.Vehicle func Insert() { if err := createSuperAdmin(); err != nil { @@ -28,8 +30,14 @@ func Insert() { if err := createUser(); err != nil { zap.S().Panicf("Failed to create user, err = %+v\n", err) } + if err := createHakUser(); err != nil { + zap.S().Panicf("Failed to create user(HAK), err = %+v\n", err) + } if err := createDevice(); err != nil { - + zap.S().Panicf("Failed to create device, err = %+v\n", err) + } + if err := createVehicle(); err != nil { + zap.S().Panicf("Failed to create vehicle, err = %+v\n", err) } } } @@ -43,6 +51,7 @@ func createUser() error { Email: "osoba@test.hr", OIB: "72352576276", Role: model.RoleOsoba, + Residence: "Zagreb", BirthDate: time.Now().AddDate(-20, 0, 0), Uuid: uuid.New(), } @@ -57,6 +66,30 @@ func createUser() error { return nil } +func createHakUser() error { + userCrud := service.NewUserCrudService() + + newUser := model.User{ + FirstName: "hak", + LastName: "hakovac", + Email: "hak@test.hr", + OIB: "30998630164", + Role: model.RoleHAK, + Residence: "Zagreb", + BirthDate: time.Now().AddDate(-20, 0, 0), + Uuid: uuid.New(), + } + + user, err := userCrud.Create(&newUser, _TEST_PASSWORD) + if err != nil { + return err + } + + zap.S().Infof("User (HAK) created, %+v\n", user) + hak = user + return nil +} + func createDevice() error { var db *gorm.DB app.Invoke(func(database *gorm.DB) { diff --git a/util/seed/vehicle.go b/util/seed/vehicle.go new file mode 100644 index 0000000..98c2e4c --- /dev/null +++ b/util/seed/vehicle.go @@ -0,0 +1,47 @@ +package seed + +import ( + "ePrometna_Server/model" + "ePrometna_Server/service" + "time" + + "github.com/google/uuid" + "go.uber.org/zap" +) + +func createVehicle() error { + vservice := service.NewVehicleService() + + vehicleInfo := model.Vehicle{ + Uuid: uuid.New(), + VehicleType: "Car", + VehicleModel: "Golf", + Mark: "Volkswagen", + VehicleCategory: "M1", + ChassisNumber: "WVWZZZ1KZAW123456", + BodyShape: "Hatchback", + VehicleUse: "Personal", + DateFirstRegistration: "2020-01-15", + EngineCapacity: "1598", + EnginePower: "110", + FuelOrPowerSource: "Petrol", + NumberOfSeats: "5", + ColourOfVehicle: "Silver", + Registration: &model.RegistrationInfo{ + Uuid: uuid.New(), + PassTechnical: true, + TraveledDistance: 25000, + TechnicalDate: time.Now().AddDate(0, 6, 0), + Registration: "ZG1234AA", + }, + } + + newVehicle, err := vservice.Create(&vehicleInfo, osoba.Uuid) + if err != nil { + return err + } + + zap.S().Infof("Vehicle created, %+v\n", newVehicle) + vehicle = newVehicle + return nil +}