Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
9 changes: 7 additions & 2 deletions main.go
Original file line number Diff line number Diff line change
Expand Up @@ -98,8 +98,14 @@ func (s ParcelService) Delete(number int) error {

func main() {
// настройте подключение к БД
db, err := sql.Open("sqlite", "tracker.db")
if err != nil {
fmt.Println(err)
return
}
defer db.Close()

store := // создайте объект ParcelStore функцией NewParcelStore
store := NewParcelStore(db) // создайте объект ParcelStore функцией NewParcelStore
service := NewParcelService(store)

// регистрация посылки
Expand Down Expand Up @@ -137,7 +143,6 @@ func main() {
err = service.Delete(p.Number)
if err != nil {
fmt.Println(err)
return
}

// вывод посылок клиента
Expand Down
67 changes: 59 additions & 8 deletions parcel.go
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,7 @@ package main

import (
"database/sql"
"fmt"
)

type ParcelStore struct {
Expand All @@ -14,47 +15,97 @@ func NewParcelStore(db *sql.DB) ParcelStore {

func (s ParcelStore) Add(p Parcel) (int, error) {
// реализуйте добавление строки в таблицу parcel, используйте данные из переменной p

res, err := s.db.Exec(
"INSERT INTO parcel (client, status, address, created_at) VALUES (?, ?, ?, ?)",
p.Client, p.Status, p.Address, p.CreatedAt,
)
// верните идентификатор последней добавленной записи
return 0, nil
if err != nil {
return 0, err
}

id, err := res.LastInsertId()
if err != nil {
return 0, err
}

return int(id), nil
}

func (s ParcelStore) Get(number int) (Parcel, error) {
// реализуйте чтение строки по заданному number
// здесь из таблицы должна вернуться только одна строка

row := s.db.QueryRow("SELECT number, client, status, address, created_at FROM parcel WHERE number = ?", number)
// заполните объект Parcel данными из таблицы
p := Parcel{}
err := row.Scan(&p.Number, &p.Client, &p.Status, &p.Address, &p.CreatedAt)
if err != nil {
return Parcel{}, err
}

return p, nil
}

func (s ParcelStore) GetByClient(client int) ([]Parcel, error) {
// реализуйте чтение строк из таблицы parcel по заданному client
// здесь из таблицы может вернуться несколько строк

rows, err := s.db.Query("SELECT number, client, status, address, created_at FROM parcel WHERE client = ?", client)
if err != nil {
return nil, err
}
defer rows.Close()
// заполните срез Parcel данными из таблицы
var res []Parcel
for rows.Next() {
p := Parcel{}
err := rows.Scan(&p.Number, &p.Client, &p.Status, &p.Address, &p.CreatedAt)
if err != nil {
return nil, err
}
res = append(res, p)
}

if err = rows.Err(); err != nil {
return nil, err
}

return res, nil
}

func (s ParcelStore) SetStatus(number int, status string) error {
// реализуйте обновление статуса в таблице parcel

return nil
_, err := s.db.Exec("UPDATE parcel SET status = ? WHERE number = ?", status, number)
return err
}

func (s ParcelStore) SetAddress(number int, address string) error {
// реализуйте обновление адреса в таблице parcel
// менять адрес можно только если значение статуса registered
parcel, err := s.Get(number)
if err != nil {
return err
}

return nil
if parcel.Status != ParcelStatusRegistered {
return fmt.Errorf("cannot change address for parcel in status %s", parcel.Status)
}

_, err = s.db.Exec("UPDATE parcel SET address = ? WHERE number = ?", address, number)
return err
}

func (s ParcelStore) Delete(number int) error {
// реализуйте удаление строки из таблицы parcel
// удалять строку можно только если значение статуса registered
parcel, err := s.Get(number)
if err != nil {
return err
}

if parcel.Status != ParcelStatusRegistered {
return fmt.Errorf("cannot delete parcel in status %s", parcel.Status)
}

return nil
_, err = s.db.Exec("DELETE FROM parcel WHERE number = ?", number)
return err
}
77 changes: 64 additions & 13 deletions parcel_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -31,57 +31,99 @@ func getTestParcel() Parcel {
// TestAddGetDelete проверяет добавление, получение и удаление посылки
func TestAddGetDelete(t *testing.T) {
// prepare
db, err := // настройте подключение к БД
db, err := sql.Open("sqlite", "tracker.db")
require.NoError(t, err)
defer db.Close()
// настройте подключение к БД
store := NewParcelStore(db)
parcel := getTestParcel()

// add
// добавьте новую посылку в БД, убедитесь в отсутствии ошибки и наличии идентификатора

id, err := store.Add(parcel)
require.NoError(t, err)
require.NotZero(t, id)
// get
// получите только что добавленную посылку, убедитесь в отсутствии ошибки
// проверьте, что значения всех полей в полученном объекте совпадают со значениями полей в переменной parcel

stored, err := store.Get(id)
require.NoError(t, err)
require.Equal(t, id, stored.Number)
require.Equal(t, parcel.Client, stored.Client)
require.Equal(t, parcel.Status, stored.Status)
require.Equal(t, parcel.Address, stored.Address)
require.Equal(t, parcel.CreatedAt, stored.CreatedAt)
// delete
// удалите добавленную посылку, убедитесь в отсутствии ошибки
// проверьте, что посылку больше нельзя получить из БД
err = store.Delete(id)
require.NoError(t, err)

_, err = store.Get(id)
require.ErrorIs(t, err, sql.ErrNoRows)
}

// TestSetAddress проверяет обновление адреса
func TestSetAddress(t *testing.T) {
// prepare
db, err := // настройте подключение к БД
db, err := sql.Open("sqlite", "tracker.db")
require.NoError(t, err)
defer db.Close() // настройте подключение к БД

store := NewParcelStore(db)
parcel := getTestParcel()
// add
// добавьте новую посылку в БД, убедитесь в отсутствии ошибки и наличии идентификатора

id, err := store.Add(parcel)
require.NoError(t, err)
require.NotZero(t, id)
// set address
// обновите адрес, убедитесь в отсутствии ошибки
newAddress := "new test address"

err = store.SetAddress(id, newAddress)
require.NoError(t, err)
// check
// получите добавленную посылку и убедитесь, что адрес обновился
stored, err := store.Get(id)
require.NoError(t, err)
require.Equal(t, newAddress, stored.Address)
}

// TestSetStatus проверяет обновление статуса
func TestSetStatus(t *testing.T) {
// prepare
db, err := // настройте подключение к БД
db, err := sql.Open("sqlite", "tracker.db")
require.NoError(t, err)
defer db.Close() // настройте подключение к БД

store := NewParcelStore(db)
parcel := getTestParcel()

// add
// добавьте новую посылку в БД, убедитесь в отсутствии ошибки и наличии идентификатора

id, err := store.Add(parcel)
require.NoError(t, err)
require.NotZero(t, id)
// set status
// обновите статус, убедитесь в отсутствии ошибки

newStatus := ParcelStatusSent
err = store.SetStatus(id, newStatus)
require.NoError(t, err)
// check
// получите добавленную посылку и убедитесь, что статус обновился
stored, err := store.Get(id)
require.NoError(t, err)
require.Equal(t, newStatus, stored.Status)
}

// TestGetByClient проверяет получение посылок по идентификатору клиента
func TestGetByClient(t *testing.T) {
// prepare
db, err := // настройте подключение к БД
db, err := sql.Open("sqlite", "tracker.db")
require.NoError(t, err)
defer db.Close() // настройте подключение к БД

store := NewParcelStore(db)

parcels := []Parcel{
getTestParcel(),
Expand All @@ -98,7 +140,9 @@ func TestGetByClient(t *testing.T) {

// add
for i := 0; i < len(parcels); i++ {
id, err := // добавьте новую посылку в БД, убедитесь в отсутствии ошибки и наличии идентификатора
id, err := store.Add(parcels[i])
require.NoError(t, err)
require.NotZero(t, id) // добавьте новую посылку в БД, убедитесь в отсутствии ошибки и наличии идентификатора

// обновляем идентификатор добавленной у посылки
parcels[i].Number = id
Expand All @@ -108,13 +152,20 @@ func TestGetByClient(t *testing.T) {
}

// get by client
storedParcels, err := // получите список посылок по идентификатору клиента, сохранённого в переменной client
storedParcels, err := store.GetByClient(client)
require.NoError(t, err)
require.Equal(t, len(parcels), len(storedParcels)) // получите список посылок по идентификатору клиента, сохранённого в переменной client
// убедитесь в отсутствии ошибки
// убедитесь, что количество полученных посылок совпадает с количеством добавленных

// check
for _, parcel := range storedParcels {
// в parcelMap лежат добавленные посылки, ключ - идентификатор посылки, значение - сама посылка
original, exists := parcelMap[parcel.Number]
require.True(t, exists)
require.Equal(t, original.Client, parcel.Client)
require.Equal(t, original.Status, parcel.Status)
require.Equal(t, original.Address, parcel.Address)
require.Equal(t, original.CreatedAt, parcel.CreatedAt) // в parcelMap лежат добавленные посылки, ключ - идентификатор посылки, значение - сама посылка
// убедитесь, что все посылки из storedParcels есть в parcelMap
// убедитесь, что значения полей полученных посылок заполнены верно
}
Expand Down