Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

First try #2

Open
wants to merge 5 commits into
base: master
Choose a base branch
from
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 @@ -97,9 +97,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)
service := NewParcelService(store)

// регистрация посылки
Expand Down
84 changes: 69 additions & 15 deletions parcel.go
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,9 @@ package main

import (
"database/sql"
"fmt"

_ "modernc.org/sqlite"
)

type ParcelStore struct {
Expand All @@ -13,48 +16,99 @@ func NewParcelStore(db *sql.DB) ParcelStore {
}

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

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

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

return int(id), nil
}

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

// заполните объект Parcel данными из таблицы
row := s.db.QueryRow("SELECT number, client, status, address, created_at FROM parcel WHERE number = :number",
sql.Named("number", number))

p := Parcel{}
err := row.Scan(&p.Number, &p.Client, &p.Status, &p.Address, &p.CreatedAt)
if err != nil {
fmt.Println(err)
return Parcel{}, err
}

return p, nil
}

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

// заполните срез Parcel данными из таблицы
rows, err := s.db.Query("SELECT number, client, status, address, created_at FROM parcel WHERE client = :client",
sql.Named("client", client))
if err != nil {
return nil, err
}
defer rows.Close()

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 rows.Err() != nil {
return nil, err
}

return res, nil
}

func (s ParcelStore) SetStatus(number int, status string) error {
// реализуйте обновление статуса в таблице parcel
_, err := s.db.Exec("UPDATE parcel SET status = :status WHERE number = :number",
sql.Named("status", status),
sql.Named("number", number))

if err != nil {
return err
}

return nil
}

func (s ParcelStore) SetAddress(number int, address string) error {
// реализуйте обновление адреса в таблице parcel
// менять адрес можно только если значение статуса registered

_, err := s.db.Exec("UPDATE parcel SET address = :address WHERE number = :number AND status = :status",
sql.Named("address", address),
sql.Named("number", number),
sql.Named("status", ParcelStatusRegistered))

if err != nil {
return err
}
return nil
}

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

_, err := s.db.Exec("DELETE FROM parcel WHERE number = :number AND status = :status",
sql.Named("number", number),
sql.Named("status", ParcelStatusRegistered))

if err != nil {
return err
}
return nil
}
106 changes: 72 additions & 34 deletions parcel_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -6,7 +6,9 @@ import (
"testing"
"time"

"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/require"
"github.com/stretchr/testify/suite"
)

var (
Expand All @@ -18,6 +20,12 @@ var (
randRange = rand.New(randSource)
)

type TestSuite struct {
suite.Suite
db *sql.DB
store ParcelStore
}

// getTestParcel возвращает тестовую посылку
func getTestParcel() Parcel {
return Parcel{
Expand All @@ -28,67 +36,101 @@ func getTestParcel() Parcel {
}
}

func (suite *TestSuite) SetupSuite() {
db, err := sql.Open("sqlite", "test.db")
if err != nil {
return
}
suite.db = db
suite.store = NewParcelStore(db)

}
func (suite *TestSuite) TearDownSuite() {
err := suite.db.Close()
require.NoError(suite.T(), err)
}

func TestTestSuite(t *testing.T) {
suite.Run(t, new(TestSuite))
}

// TestAddGetDelete проверяет добавление, получение и удаление посылки
func TestAddGetDelete(t *testing.T) {
func (suite *TestSuite) TestAddGetDelete() {
// prepare
db, err := // настройте подключение к БД
store := NewParcelStore(db)
parcel := getTestParcel()

// add
// добавьте новую посылку в БД, убедитесь в отсутствии ошибки и наличии идентификатора
num, err := suite.store.Add(parcel)
require.NoError(suite.T(), err)
require.NotEmpty(suite.T(), num)

// get
// получите только что добавленную посылку, убедитесь в отсутствии ошибки
// проверьте, что значения всех полей в полученном объекте совпадают со значениями полей в переменной parcel
p, err := suite.store.Get(num)
require.NoError(suite.T(), err)
// Adjust for parcel default number being 0
parcel.Number = num
require.Equal(suite.T(), parcel, p)

// delete
// удалите добавленную посылку, убедитесь в отсутствии ошибки
// проверьте, что посылку больше нельзя получить из БД
err = suite.store.Delete(num)
require.NoError(suite.T(), err)

// get check
_, err = suite.store.Get(num)
require.Error(suite.T(), err)

}

// TestSetAddress проверяет обновление адреса
func TestSetAddress(t *testing.T) {
func (suite *TestSuite) TestSetAddress() {
// prepare
db, err := // настройте подключение к БД
parcel := getTestParcel()

// add
// добавьте новую посылку в БД, убедитесь в отсутствии ошибки и наличии идентификатора
num, err := suite.store.Add(parcel)
require.NoError(suite.T(), err)
require.True(suite.T(), (num > 0))

// set address
// обновите адрес, убедитесь в отсутствии ошибки
newAddress := "new test address"
err = suite.store.SetAddress(num, newAddress)
require.NoError(suite.T(), err)

// check
// получите добавленную посылку и убедитесь, что адрес обновился
p, err := suite.store.Get(num)
assert.NoError(suite.T(), err)
assert.Equal(suite.T(), p.Address, newAddress)
}

// TestSetStatus проверяет обновление статуса
func TestSetStatus(t *testing.T) {
func (suite *TestSuite) TestSetStatus() {
// prepare
db, err := // настройте подключение к БД
parcel := getTestParcel()

// add
// добавьте новую посылку в БД, убедитесь в отсутствии ошибки и наличии идентификатора
num, err := suite.store.Add(parcel)
require.NoError(suite.T(), err)
require.True(suite.T(), (num > 0))

// set status
// обновите статус, убедитесь в отсутствии ошибки
err = suite.store.SetStatus(num, ParcelStatusSent)
require.NoError(suite.T(), err)

// check
// получите добавленную посылку и убедитесь, что статус обновился
p, err := suite.store.Get(num)
assert.NoError(suite.T(), err)
assert.Equal(suite.T(), ParcelStatusSent, p.Status)
}

// TestGetByClient проверяет получение посылок по идентификатору клиента
func TestGetByClient(t *testing.T) {
func (suite *TestSuite) TestGetByClient() {
// prepare
db, err := // настройте подключение к БД

parcels := []Parcel{
getTestParcel(),
getTestParcel(),
getTestParcel(),
}
parcelMap := map[int]Parcel{}

// задаём всем посылкам один и тот же идентификатор клиента
client := randRange.Intn(10_000_000)
Expand All @@ -98,24 +140,20 @@ func TestGetByClient(t *testing.T) {

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

id, err := suite.store.Add(parcels[i])
require.NoError(suite.T(), err)
require.True(suite.T(), (id > 0))
// обновляем идентификатор добавленной у посылки
parcels[i].Number = id

// сохраняем добавленную посылку в структуру map, чтобы её можно было легко достать по идентификатору посылки
parcelMap[id] = parcels[i]
}
// Не совсем понял комментарий, по поводу наличия в testify функции для работы с массивами которой можно заменить данный цикл :')
// Но так как это был precode, я полагаю что это был совет, а не требование, поэтому чтобы успеть закрыть проект до дедлайна отправляю так.

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

// check
for _, parcel := range storedParcels {
// в parcelMap лежат добавленные посылки, ключ - идентификатор посылки, значение - сама посылка
// убедитесь, что все посылки из storedParcels есть в parcelMap
// убедитесь, что значения полей полученных посылок заполнены верно
}
assert.ElementsMatch(suite.T(), parcels, storedParcels)
}
Binary file added test.db
Binary file not shown.
Binary file modified tracker.db
Binary file not shown.