Compare commits

...

2 Commits

Author SHA1 Message Date
6c95d944a1 0.1.2-prod 2025-08-21 10:42:56 +03:00
173d4c670c 0.1.2-prod 2025-08-21 10:42:35 +03:00
7 changed files with 529 additions and 107 deletions

2
.gitignore vendored
View File

@@ -1,2 +1,4 @@
*.exe
*.json
*.log
*.zip

193
README.md Normal file
View File

@@ -0,0 +1,193 @@
# Go-библиотека и утилита для ККТ "Штрих-М" (shtrih-kkt)
[![License: MIT](https://img.shields.io/badge/License-MIT-yellow.svg)](https://opensource.org/licenses/MIT)
Go-библиотека и консольная утилита для взаимодействия с фискальными регистраторами (ККТ) "Штрих-М" через официальный 32-битный COM-драйвер. Проект разработан с фокусом на безопасный сбор данных в режиме "только чтение" и предоставляет гибкие режимы работы для различных сценариев использования.
**Важнейшее ограничение:** Приложения, использующие эту библиотеку, **должны быть скомпилированы для 32-битной архитектуры (`GOARCH=386`)** из-за зависимости от 32-битного COM-драйвера.
## Ключевые возможности
* **Надежная обертка над COM-драйвером:** Предоставляет безопасный и удобный Go-интерфейс для драйвера "Штрих-М".
* **Комплексный сбор данных:** Агрегирует полную информацию о ККТ, включая регистрационные данные, статус ФН, версии ПО, лицензии и атрибуты торговли.
* **Умный автопоиск устройств:**
* **COM-порты:** Автоматически сканирует все системные COM-порты на двух самых распространенных скоростях (`115200` и `4800`), предотвращая зависания на "портах-призраках".
* **TCP/IP (RNDIS):** Параллельно сканирует стандартные для RNDIS-устройств IP-подсети (`192.168.137.0/24`, `192.168.138.0/24`).
* **Два режима работы утилиты:**
1. **Режим автопоиска:** При первом запуске или отсутствии конфигурации выполняет полный поиск устройств, собирает с них данные и **сохраняет найденные конфигурации** в `connect.json` для последующих быстрых запусков.
2. **Стационарный режим:** При наличии файла `connect.json` использует заданные в нем параметры для быстрого опроса конкретных ККТ, пропуская этап сканирования.
* **Гибкое управление данными:**
* Сохраняет информацию о каждом ККТ в отдельный JSON-файл (`/date/{ЗН_ККТ}.json`).
* "Обогащает" данные ККТ информацией о рабочей станции (hostname, TeamViewer ID и т.д.), заимствуя ее из существующих JSON-файлов в папке `/date`.
* Автоматически обновляет временные метки (`current_time`, `v_time`) в существующих файлах при повторных опросах.
* **Файловое логирование с ротацией:** В стационарном режиме ведет подробный лог в папке `/logs`, настройки которого (срок хранения) задаются в файле `service.json`.
* **Тестируемость:** Включает в себя `mockDriver` для написания unit-тестов без необходимости подключения реального оборудования.
## Архитектура
Библиотека построена на простом интерфейсе `Driver`, что позволяет легко подменять реализации:
* `comDriver`: Основная реализация для работы с реальным COM-драйвером.
* `mockDriver`: Имитационная реализация для unit-тестирования.
Это обеспечивает слабую связанность и позволяет тестировать логику приложений, использующих библиотеку, в изолированной среде.
## Начало работы
### Требования
1. **Go:** Версия 1.18 или выше.
2. **ОС:** Windows (x86 или x64).
3. **32-битный (x86) тулчейн Go:** Даже на 64-битной системе для компиляции требуется 32-битный набор инструментов.
4. **Драйвер "Штрих-М":** На целевой машине должен быть установлен и зарегистрирован официальный драйвер от "Штрих-М" (например, `DrvFR_4.15_882.exe`).
### Использование библиотеки в вашем проекте
1. **Добавьте библиотеку в ваш проект:**
```bash
go get github.com/your-username/shtrih-kkt/pkg/shtrih
```
2. **Пример использования:**
```go
package main
import (
"fmt"
"log"
"shtrih-kkt/pkg/shtrih"
)
func main() {
// Конфигурация для подключения по COM-порту
config := shtrih.Config{
ConnectionType: 0,
ComName: "COM3",
ComNumber: 3,
BaudRate: 6, // Индекс для 115200
Password: 30,
}
// Создаем новый драйвер
driver := shtrih.New(config)
// Подключаемся
if err := driver.Connect(); err != nil {
log.Fatalf("Ошибка подключения: %v", err)
}
// Гарантируем отключение в конце
defer driver.Disconnect()
// Получаем информацию
info, err := driver.GetFiscalInfo()
if err != nil {
log.Fatalf("Ошибка получения информации: %v", err)
}
fmt.Printf("Успешно получены данные для ККТ модели: %s\n", info.ModelName)
fmt.Printf("Серийный номер: %s\n", info.SerialNumber)
}
```
### Использование готовой утилиты `shtrih-scanner.exe`
Утилита предназначена для запуска "рядом" с другими служебными файлами.
1. **Режим автопоиска (первый запуск):**
* Просто запустите `shtrih-scanner.exe`.
* Программа выполнит полный поиск устройств.
* В папке `/date` будут созданы JSON-файлы с данными для каждой найденной ККТ.
* Будет создан или перезаписан файл `connect.json` с параметрами найденных устройств.
2. **Стационарный режим (последующие запуски):**
* Убедитесь, что рядом с `shtrih-scanner.exe` лежит `connect.json`.
* Для настройки логирования создайте файл `service.json`.
* Запустите `shtrih-scanner.exe`.
* Программа быстро опросит устройства из `connect.json` и обновит временные метки в файлах в папке `/date`.
#### Конфигурационные файлы
* `connect.json` (генерируется автоматически):
```json
{
"shtrih": [
{
"type_connect": 0,
"com_port": "COM1",
"com_baudrate": "115200"
},
{
"type_connect": 6,
"ip": "192.168.137.111",
"ip_port": "7778"
}
]
}
```
* `service.json` (создается вручную для настройки логов):
```json
{
"service": {
"log_level": "info",
"log_days": 7
}
}
```
## Сборка проекта
**Важно:** Сборка должна производиться для архитектуры `386`.
Откройте терминал в корневой папке проекта.
**Для PowerShell:**
```powershell
$env:GOARCH="386"; $env:GOOS="windows"; go build -o shtrih-scanner.exe
```
**Для CMD:**
```cmd
set GOARCH=386
set GOOS=windows
go build -o shtrih-scanner.exe
```
**Сборка без консольного окна (для фоновой работы):**
```powershell
$env:GOARCH="386"; $env:GOOS="windows"; go build -ldflags="-H=windowsgui" -o shtrih-scanner.exe
```
## Тестирование
Для запуска unit-тестов, использующих `mockDriver`, выполните команду в корне проекта:
```bash
go test ./...
```
## Структура проекта
```
shtrih-kkt/
├── go.mod # Файл модуля Go
├── main.go # Исходный код утилиты shtrih-scanner.exe
├── README.md # Этот файл
└── pkg/
└── shtrih/
├── driver.go # Основная логика библиотеки и реализация comDriver
├── mock_driver.go # Реализация mockDriver для тестов
└── driver_test.go # Unit-тесты для библиотеки
---
# Файлы, создаваемые во время работы:
shtrih-scanner.exe
connect.json
service.json
date/
│ └── 0012345678901234.json
logs/
└── shtrih-scanner.log
```
## Лицензия
Этот проект распространяется под лицензией MIT. См. файл `LICENSE` для получения дополнительной информации.

1
go.mod
View File

@@ -5,6 +5,7 @@ go 1.23.4
require (
github.com/go-ole/go-ole v1.3.0
go.bug.st/serial v1.6.4
gopkg.in/natefinch/lumberjack.v2 v2.2.1
)
require (

2
go.sum
View File

@@ -13,5 +13,7 @@ go.bug.st/serial v1.6.4/go.mod h1:nofMJxTeNVny/m6+KaafC6vJGj3miwQZ6vW4BZUGJPI=
golang.org/x/sys v0.1.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
golang.org/x/sys v0.19.0 h1:q5f1RH2jigJ1MoAWp2KTp3gm5zAGFUTarQZ5U386+4o=
golang.org/x/sys v0.19.0/go.mod h1:/VUhepiaJMQUp4+oa/7Zr1D23ma6VTLIYjOOTFZPUcA=
gopkg.in/natefinch/lumberjack.v2 v2.2.1 h1:bBRl1b0OH9s/DuPhuXpNl+VtCaJXFZ5/uEFST95x9zc=
gopkg.in/natefinch/lumberjack.v2 v2.2.1/go.mod h1:YD8tP3GAjkrDg1eZH7EGmyESg/lsYskCTPBJVb9jqSc=
gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA=
gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=

223
main.go
View File

@@ -3,6 +3,7 @@ package main
import (
"encoding/json"
"fmt"
"io"
"log"
"os"
"path/filepath"
@@ -10,15 +11,21 @@ import (
"time"
"shtrih-kkt/pkg/shtrih"
"gopkg.in/natefinch/lumberjack.v2"
)
const (
configFileName = "connect.json"
serviceConfigName = "service.json"
outputDir = "date"
comSearchTimeout = 220 * time.Millisecond
tcpSearchTimeout = 300 * time.Millisecond
logsDir = "logs"
comSearchTimeout = 200 * time.Millisecond // Уменьшенный таймаут
tcpSearchTimeout = 150 * time.Millisecond
)
// --- СТРУКТУРЫ ДЛЯ ПАРСИНГА КОНФИГУРАЦИОННЫХ ФАЙЛОВ ---
// ConfigFile соответствует структуре файла connect.json
type ConfigFile struct {
Timeout int `json:"timeout_to_ip_port"`
@@ -35,6 +42,17 @@ type ConnectionSettings struct {
IPPort string `json:"ip_port"`
}
// ServiceFile используется для чтения настроек логирования из service.json
type ServiceFile struct {
Service ServiceConfig `json:"service"`
}
// ServiceConfig содержит параметры логирования
type ServiceConfig struct {
LogLevel string `json:"log_level"`
LogDays int `json:"log_days"`
}
// PolledDevice связывает конфигурацию, использованную для подключения,
// с фискальной информацией, полученной от устройства.
type PolledDevice struct {
@@ -42,13 +60,15 @@ type PolledDevice struct {
Info *shtrih.FiscalInfo
}
// --- ОСНОВНАЯ ЛОГИКА ПРИЛОЖЕНИЯ ---
func main() {
log.Println("Запуск приложения для сбора данных с ККТ Штрих-М...")
configData, err := os.ReadFile(configFileName)
if err != nil {
if os.IsNotExist(err) {
log.Printf("Файл конфигурации '%s' не найден. Запускаю режим автопоиска устройств...", configFileName)
log.Printf("Файл конфигурации '%s' не найден. Запускаю режим автопоиска...", configFileName)
runDiscoveryMode()
} else {
log.Fatalf("Ошибка чтения файла конфигурации '%s': %v", configFileName, err)
@@ -61,11 +81,56 @@ func main() {
log.Println("Работа приложения завершена.")
}
// setupLogger настраивает запись логов в файл для стационарного режима.
func setupLogger() {
data, err := os.ReadFile(serviceConfigName)
if err != nil {
log.Printf("Предупреждение: файл настроек '%s' не найден. Логирование продолжится в консоль.", serviceConfigName)
return
}
var serviceFile ServiceFile
if err := json.Unmarshal(data, &serviceFile); err != nil {
log.Printf("Предупреждение: не удалось прочитать настройки из '%s' (%v). Логирование продолжится в консоль.", serviceConfigName, err)
return
}
// Устанавливаем значения по умолчанию, если в файле их нет
logDays := serviceFile.Service.LogDays
if logDays <= 0 {
logDays = 7
}
// Создаем папку для логов, если ее нет
if err := os.MkdirAll(logsDir, 0755); err != nil {
log.Printf("Ошибка создания директории для логов '%s': %v. Логирование продолжится в консоль.", logsDir, err)
return
}
logFilePath := filepath.Join(logsDir, "shtrih-scanner.log")
// Настраиваем ротацию логов
lumberjackLogger := &lumberjack.Logger{
Filename: logFilePath,
MaxSize: 5, // мегабайты
MaxBackups: 10,
MaxAge: logDays, // дни
Compress: true, // сжимать старые файлы
}
// Устанавливаем вывод логов и в файл, и в консоль
log.SetOutput(io.MultiWriter(os.Stdout, lumberjackLogger))
log.Printf("Логирование настроено. Уровень: %s, ротация: %d дней. Файл: %s", serviceFile.Service.LogLevel, logDays, logFilePath)
}
func runConfigMode(data []byte) {
// Первым делом настраиваем логирование для стационарного режима!
setupLogger()
var configFile ConfigFile
if err := json.Unmarshal(data, &configFile); err != nil {
log.Printf("Ошибка парсинга JSON из '%s': %v. Переключаюсь на режим автопоиска.", configFileName, err)
runDiscoveryMode()
runDiscoveryMode() // В случае ошибки автопоиск будет логировать только в консоль
return
}
@@ -99,16 +164,13 @@ func runDiscoveryMode() {
log.Printf("Найдено %d устройств. Начинаю сбор информации...", len(configs))
polledDevices := processDevices(configs)
// Если были успешно опрошены какие-либо устройства, сохраняем их конфигурацию
if len(polledDevices) > 0 {
saveConfiguration(polledDevices)
}
}
// processDevices - основная функция, реализующая "умную" логику обновления и создания файлов.
// Теперь она возвращает срез успешно опрошенных устройств.
func processDevices(configs []shtrih.Config) []PolledDevice {
// Шаг 1: Сначала собираем информацию со всех найденных устройств.
var polledDevices []PolledDevice // Было: var freshKKTData []*shtrih.FiscalInfo
var polledDevices []PolledDevice
for _, config := range configs {
log.Printf("--- Опрашиваю устройство: %+v ---", config)
driver := shtrih.New(config)
@@ -119,7 +181,7 @@ func processDevices(configs []shtrih.Config) []PolledDevice {
}
info, err := driver.GetFiscalInfo()
driver.Disconnect() // Отключаемся сразу после получения данных
driver.Disconnect()
if err != nil {
log.Printf("Ошибка при получении фискальной информации: %v", err)
@@ -129,30 +191,26 @@ func processDevices(configs []shtrih.Config) []PolledDevice {
log.Println("Получена пустая информация или отсутствует серийный номер, данные проигнорированы.")
continue
}
// Сохраняем и конфигурацию, и результат
polledDevices = append(polledDevices, PolledDevice{Config: config, Info: info})
}
if len(polledDevices) == 0 {
log.Println("--- Не удалось собрать данные ни с одного устройства. Завершение. ---")
return nil // Возвращаем nil, если ничего не найдено
return nil
}
log.Printf("--- Всего собрано данных с %d устройств. Начинаю обработку файлов. ---", len(polledDevices))
// Шаг 2: Ищем "донора" данных о рабочей станции в папке /date.
sourceWSDataMap := findSourceWorkstationData()
// Шаг 3: Обрабатываем каждого "свежего" ККТ в соответствии с новой логикой.
var successCount int
for _, pd := range polledDevices { // Итерируемся по новой структуре
kktInfo := pd.Info // Получаем доступ к данным ККТ
for _, pd := range polledDevices {
kktInfo := pd.Info
fileName := fmt.Sprintf("%s.json", kktInfo.SerialNumber)
filePath := filepath.Join(outputDir, fileName)
// ... (остальная часть цикла остается без изменений) ...
if _, err := os.Stat(filePath); err == nil {
log.Printf("Файл для ККТ %s уже существует. Обновляю временную метку...", kktInfo.SerialNumber)
log.Printf("Файл для ККТ %s уже существует. Обновляю временные метки...", kktInfo.SerialNumber)
if err := updateTimestampInFile(filePath); err != nil {
log.Printf("Не удалось обновить файл %s: %v", filePath, err)
} else {
@@ -169,7 +227,7 @@ func processDevices(configs []shtrih.Config) []PolledDevice {
hostname, _ := os.Hostname()
wsDataToUse = map[string]interface{}{"hostname": hostname}
}
wsDataToUse["current_time"] = time.Now().Format("2006-01-02 15:04:05")
if err := saveNewMergedInfo(kktInfo, wsDataToUse, filePath); err != nil {
log.Printf("Не удалось создать файл для ККТ %s: %v", kktInfo.SerialNumber, err)
} else {
@@ -179,10 +237,11 @@ func processDevices(configs []shtrih.Config) []PolledDevice {
}
log.Printf("--- Обработка файлов завершена. Успешно создано/обновлено: %d файлов. ---", successCount)
return polledDevices // Возвращаем результат
return polledDevices
}
// findSourceWorkstationData ищет в папке /date любой .json файл и извлекает из него
// все данные как `map[string]interface{}`.
// --- ФУНКЦИИ ДЛЯ РАБОТЫ С ФАЙЛАМИ ---
func findSourceWorkstationData() map[string]interface{} {
files, err := os.ReadDir(outputDir)
if err != nil {
@@ -204,13 +263,11 @@ func findSourceWorkstationData() map[string]interface{} {
continue
}
// Проверяем, что это не файл от нашего ККТ (у него не должно быть поля modelName)
// и что у него есть hostname. Это делает выбор донора более надежным.
_, hasModelName := content["modelName"]
_, hasHostname := content["hostname"]
if !hasModelName && hasHostname {
log.Printf("Найден файл-донор с данными о рабочей станции: %s", filePath)
return content // Возвращаем все содержимое файла как карту.
return content
}
}
}
@@ -219,7 +276,6 @@ func findSourceWorkstationData() map[string]interface{} {
return nil
}
// updateTimestampInFile читает JSON-файл, обновляет в нем поле current_time и перезаписывает его.
func updateTimestampInFile(filePath string) error {
data, err := os.ReadFile(filePath)
if err != nil {
@@ -231,7 +287,10 @@ func updateTimestampInFile(filePath string) error {
return fmt.Errorf("ошибка парсинга JSON: %w", err)
}
content["current_time"] = time.Now().Format("2006-01-02 15:04:05")
// Обновляем оба поля времени
currentTime := time.Now().Format("2006-01-02 15:04:05")
content["current_time"] = currentTime
content["v_time"] = currentTime
updatedData, err := json.MarshalIndent(content, "", " ")
if err != nil {
@@ -241,22 +300,21 @@ func updateTimestampInFile(filePath string) error {
return os.WriteFile(filePath, updatedData, 0644)
}
// saveNewMergedInfo объединяет данные ККТ и данные рабочей станции (в виде map) и сохраняет в новый JSON-файл.
func saveNewMergedInfo(kktInfo *shtrih.FiscalInfo, wsData map[string]interface{}, filePath string) error {
var kktMap map[string]interface{}
kktJSON, _ := json.Marshal(kktInfo)
json.Unmarshal(kktJSON, &kktMap)
// Сливаем карты. Ключи из wsData перезапишут любые совпадения в kktMap.
// Добавляем актуальные временные метки в данные рабочей станции
currentTime := time.Now().Format("2006-01-02 15:04:05")
wsData["current_time"] = currentTime
wsData["v_time"] = currentTime
for key, value := range wsData {
kktMap[key] = value
}
// Удаляем поля, специфичные для ККТ, из данных донора, если они случайно туда попали.
// Это предотвратит запись, например, "serialNumber" от АТОЛ в файл Штриха.
delete(kktMap, "serialNumber")
// Возвращаем серийный номер нашего ККТ, который мы сохранили в структуре kktInfo.
kktMap["serialNumber"] = kktInfo.SerialNumber
if err := os.MkdirAll(filepath.Dir(filePath), 0755); err != nil {
@@ -276,20 +334,50 @@ func saveNewMergedInfo(kktInfo *shtrih.FiscalInfo, wsData map[string]interface{}
return nil
}
// convertSettingsToConfigs преобразует настройки из файла в формат, понятный библиотеке.
func saveConfiguration(polledDevices []PolledDevice) {
log.Printf("Сохранение %d найденных конфигураций в файл '%s'...", len(polledDevices), configFileName)
var configFile ConfigFile
data, err := os.ReadFile(configFileName)
if err == nil {
if err := json.Unmarshal(data, &configFile); err != nil {
log.Printf("Предупреждение: файл '%s' поврежден (%v). Он будет перезаписан.", configFileName, err)
configFile = ConfigFile{}
}
}
var newShtrihSettings []ConnectionSettings
for _, pd := range polledDevices {
newShtrihSettings = append(newShtrihSettings, convertConfigToSettings(pd.Config))
}
configFile.Shtrih = newShtrihSettings
updatedData, err := json.MarshalIndent(configFile, "", " ")
if err != nil {
log.Printf("Ошибка: не удалось преобразовать конфигурацию в JSON: %v", err)
return
}
if err := os.WriteFile(configFileName, updatedData, 0644); err != nil {
log.Printf("Ошибка: не удалось записать конфигурацию в файл '%s': %v", configFileName, err)
return
}
log.Printf("Конфигурация успешно сохранена в '%s'.", configFileName)
}
// --- ВСПОМОГАТЕЛЬНЫЕ ФУНКЦИИ ---
func convertSettingsToConfigs(settings []ConnectionSettings) []shtrih.Config {
var configs []shtrih.Config
baudRateMap := map[string]int32{
"115200": 6, "57600": 5, "38400": 4, "19200": 3, "9600": 2,
"115200": 6, "57600": 5, "38400": 4, "19200": 3, "9600": 2, "4800": 1,
}
for _, s := range settings {
config := shtrih.Config{
ConnectionType: s.TypeConnect,
Password: 30,
}
config := shtrih.Config{ConnectionType: s.TypeConnect, Password: 30}
switch s.TypeConnect {
case 0: // COM-порт
case 0:
comNum, err := strconv.Atoi(s.ComPort[3:])
if err != nil {
log.Printf("Некорректное имя COM-порта '%s' в конфигурации, пропуск.", s.ComPort)
@@ -303,7 +391,7 @@ func convertSettingsToConfigs(settings []ConnectionSettings) []shtrih.Config {
config.ComName = s.ComPort
config.ComNumber = int32(comNum)
config.BaudRate = baudRate
case 6: // TCP/IP
case 6:
port, err := strconv.Atoi(s.IPPort)
if err != nil {
log.Printf("Некорректный TCP-порт '%s' для IP '%s', пропуск.", s.IPPort, s.IP)
@@ -320,63 +408,18 @@ func convertSettingsToConfigs(settings []ConnectionSettings) []shtrih.Config {
return configs
}
// saveConfiguration обновляет файл connect.json, записывая в него
// конфигурации успешно найденных и опрошенных устройств.
func saveConfiguration(polledDevices []PolledDevice) {
log.Printf("Сохранение %d найденных конфигураций в файл '%s'...", len(polledDevices), configFileName)
// Шаг 1: Читаем существующий файл или создаем новую структуру.
var configFile ConfigFile
data, err := os.ReadFile(configFileName)
if err == nil {
// Файл есть, парсим его, чтобы не потерять другие секции (например, "atol")
if err := json.Unmarshal(data, &configFile); err != nil {
log.Printf("Предупреждение: файл '%s' поврежден (%v). Он будет перезаписан.", configFileName, err)
configFile = ConfigFile{} // Создаем пустую структуру в случае ошибки
}
}
// Шаг 2: Формируем новый срез настроек для "shtrih".
var newShtrihSettings []ConnectionSettings
for _, pd := range polledDevices {
newShtrihSettings = append(newShtrihSettings, convertConfigToSettings(pd.Config))
}
// Шаг 3: Полностью заменяем секцию "shtrih" новыми данными.
configFile.Shtrih = newShtrihSettings
// Шаг 4: Записываем обновленную структуру обратно в файл.
updatedData, err := json.MarshalIndent(configFile, "", " ")
if err != nil {
log.Printf("Ошибка: не удалось преобразовать конфигурацию в JSON: %v", err)
return
}
if err := os.WriteFile(configFileName, updatedData, 0644); err != nil {
log.Printf("Ошибка: не удалось записать конфигурацию в файл '%s': %v", configFileName, err)
return
}
log.Printf("Конфигурация успешно сохранена в '%s'.", configFileName)
}
// convertConfigToSettings преобразует внутренний формат shtrih.Config
// в формат ConnectionSettings для записи в connect.json.
func convertConfigToSettings(config shtrih.Config) ConnectionSettings {
// Карта для обратного преобразования индекса скорости в строку
baudRateReverseMap := map[int32]string{
6: "115200", 5: "57600", 4: "38400", 3: "19200", 2: "9600",
6: "115200", 5: "57600", 4: "38400", 3: "19200", 2: "9600", 1: "4800",
}
settings := ConnectionSettings{
TypeConnect: config.ConnectionType,
}
settings := ConnectionSettings{TypeConnect: config.ConnectionType}
switch config.ConnectionType {
case 0: // COM-порт
case 0:
settings.ComPort = config.ComName
settings.ComBaudrate = baudRateReverseMap[config.BaudRate]
case 6: // TCP/IP
case 6:
settings.IP = config.IPAddress
settings.IPPort = strconv.Itoa(int(config.TCPPort))
}

129
pkg/shtrih/driver_test.go Normal file
View File

@@ -0,0 +1,129 @@
// Тесты для пакета shtrih
package shtrih
import (
"fmt"
"reflect"
"testing"
)
// getSampleFiscalInfo создает пример структуры FiscalInfo для использования в тестах.
func getSampleFiscalInfo() *FiscalInfo {
return &FiscalInfo{
ModelName: "ШТРИХ-М-01Ф",
SerialNumber: "0012345678901234",
RNM: "0009876543210987",
OrganizationName: "ООО Ромашка",
Inn: "7701234567",
FnSerial: "9960440300112233",
FfdVersion: "120",
}
}
// TestMockDriver_SuccessfulPath проверяет стандартный успешный сценарий:
// подключение -> получение данных -> отключение.
func TestMockDriver_SuccessfulPath(t *testing.T) {
// Arrange: Готовим данные и создаем мок-драйвер.
sampleData := getSampleFiscalInfo()
driver := NewMockDriver(sampleData, nil, nil)
// Act 1: Подключаемся.
err := driver.Connect()
if err != nil {
t.Fatalf("Connect() вернул неожиданную ошибку: %v", err)
}
// Act 2: Получаем фискальную информацию.
info, err := driver.GetFiscalInfo()
if err != nil {
t.Fatalf("GetFiscalInfo() вернул неожиданную ошибку: %v", err)
}
// Act 3: Отключаемся.
err = driver.Disconnect()
if err != nil {
t.Fatalf("Disconnect() вернул неожиданную ошибку: %v", err)
}
// Assert: Проверяем, что полученные данные соответствуют ожидаемым.
if !reflect.DeepEqual(info, sampleData) {
t.Errorf("Полученные данные не совпадают с мок-данными.\nПолучено: %+v\nОжидалось: %+v", info, sampleData)
}
// Assert: Проверяем, что все методы были вызваны.
// Для этого нам нужно преобразовать интерфейс обратно в конкретный тип мок-драйвера.
mock, ok := driver.(*mockDriver)
if !ok {
t.Fatal("Не удалось преобразовать драйвер в *mockDriver для проверки вызовов.")
}
if !mock.ConnectCalled {
t.Error("Ожидалось, что Connect() будет вызван, но этого не произошло.")
}
if !mock.GetFiscalInfoCalled {
t.Error("Ожидалось, что GetFiscalInfo() будет вызван, но этого не произошло.")
}
if !mock.DisconnectCalled {
t.Error("Ожидалось, что Disconnect() будет вызван, но этого не произошло.")
}
}
// TestMockDriver_ConnectError проверяет, что драйвер корректно обрабатывает
// ошибку, возвращаемую при подключении.
func TestMockDriver_ConnectError(t *testing.T) {
// Arrange: Создаем симулируемую ошибку.
simulatedError := fmt.Errorf("порт COM5 занят")
driver := NewMockDriver(nil, simulatedError, nil)
// Act: Пытаемся подключиться.
err := driver.Connect()
// Assert: Проверяем, что была возвращена именно наша ошибка.
if err == nil {
t.Fatal("Connect() не вернул ошибку, хотя ожидалось.")
}
if err != simulatedError {
t.Errorf("Connect() вернул неверную ошибку. Получено: %v, Ожидалось: %v", err, simulatedError)
}
}
// TestMockDriver_GetInfoWhileDisconnected проверяет, что попытка получить
// данные без предварительного подключения вернет ошибку.
func TestMockDriver_GetInfoWhileDisconnected(t *testing.T) {
// Arrange: Создаем стандартный мок-драйвер.
driver := NewMockDriver(getSampleFiscalInfo(), nil, nil)
// Act: Сразу пытаемся получить данные.
_, err := driver.GetFiscalInfo()
// Assert: Проверяем, что получили ошибку.
if err == nil {
t.Fatal("GetFiscalInfo() не вернул ошибку при вызове без подключения.")
}
}
// TestMockDriver_GetInfoError проверяет, что драйвер корректно обрабатывает
// ошибку, возвращаемую при получении данных.
func TestMockDriver_GetInfoError(t *testing.T) {
// Arrange: Создаем симулируемую ошибку.
simulatedError := fmt.Errorf("ошибка чтения ФН")
driver := NewMockDriver(nil, nil, simulatedError)
// Act
err := driver.Connect()
if err != nil {
t.Fatalf("Connect() неожиданно вернул ошибку: %v", err)
}
info, err := driver.GetFiscalInfo()
// Assert
if err == nil {
t.Fatal("GetFiscalInfo() не вернул ошибку, хотя ожидалось.")
}
if err != simulatedError {
t.Errorf("GetFiscalInfo() вернул неверную ошибку. Получено: %v, Ожидалось: %v", err, simulatedError)
}
if info != nil {
t.Error("GetFiscalInfo() вернул данные вместе с ошибкой, хотя должен был вернуть nil.")
}
}

View File

@@ -1,33 +1,85 @@
// Package shtrih (продолжение)
package shtrih
// mockDriver — это имитация драйвера для тестирования.
import (
"fmt"
"log"
)
// mockDriver представляет собой имитацию реального драйвера для целей тестирования.
// Он реализует интерфейс Driver.
type mockDriver struct {
FiscalInfoToReturn *FiscalInfo
ErrorToReturn error
// MockData - это структура с фискальными данными, которую вернет GetFiscalInfo.
MockData *FiscalInfo
// ConnectErr - ошибка, которую вернет метод Connect, если она задана.
ConnectErr error
// GetFiscalInfoErr - ошибка, которую вернет метод GetFiscalInfo, если она задана.
GetFiscalInfoErr error
// Внутренние флаги для проверки вызовов в тестах.
connected bool
ConnectCalled bool
DisconnectCalled bool
GetFiscalInfoCalled bool
}
// NewMock создает новый мок-драйвер.
func NewMock(info *FiscalInfo, err error) Driver {
// NewMockDriver является конструктором для создания нового мок-драйвера.
// Позволяет заранее определить, какие данные и ошибки будут возвращаться.
func NewMockDriver(data *FiscalInfo, connectErr, getInfoErr error) Driver {
return &mockDriver{
FiscalInfoToReturn: info,
ErrorToReturn: err,
MockData: data,
ConnectErr: connectErr,
GetFiscalInfoErr: getInfoErr,
}
}
// Connect имитирует подключение к ККТ.
func (m *mockDriver) Connect() error {
if m.ErrorToReturn != nil {
return m.ErrorToReturn
m.ConnectCalled = true
log.Println("Mock Driver: Connect() вызван.")
// Если была задана ошибка подключения, возвращаем ее.
if m.ConnectErr != nil {
return m.ConnectErr
}
// Если уже "подключены", ничего не делаем.
if m.connected {
return nil
}
m.connected = true
return nil
}
// Disconnect имитирует отключение от ККТ.
func (m *mockDriver) Disconnect() error {
m.DisconnectCalled = true
log.Println("Mock Driver: Disconnect() вызван.")
// Если не были "подключены", ничего не делаем.
if !m.connected {
return nil
}
m.connected = false
return nil
}
// GetFiscalInfo имитирует получение фискальных данных.
func (m *mockDriver) GetFiscalInfo() (*FiscalInfo, error) {
if m.ErrorToReturn != nil {
return nil, m.ErrorToReturn
m.GetFiscalInfoCalled = true
log.Println("Mock Driver: GetFiscalInfo() вызван.")
// Проверяем, было ли установлено "соединение".
if !m.connected {
return nil, fmt.Errorf("мок-драйвер: не подключен")
}
return m.FiscalInfoToReturn, nil
// Если была задана ошибка получения данных, возвращаем ее.
if m.GetFiscalInfoErr != nil {
return nil, m.GetFiscalInfoErr
}
// Если не были предоставлены мок-данные, возвращаем ошибку.
if m.MockData == nil {
return nil, fmt.Errorf("мок-драйвер: данные для имитации не предоставлены")
}
return m.MockData, nil
}