Initial cloud-services repo - gateway service + pkg modules

2026-01-30 23:14:52 -05:00
commit fbb820d7b3
1037 changed files with 171318 additions and 0 deletions
--- a/pkg/db/queries/mocks/updatemanifests.go
+++ b/pkg/db/queries/mocks/updatemanifests.go
@@ -0,0 +1,137 @@
+package mocks
+
+import (
+	"encoding/json"
+	"time"
+
+	"fiskerinc.com/modules/common"
+	"fiskerinc.com/modules/db/queries"
+	"fiskerinc.com/modules/validator"
+
+	"github.com/go-pg/pg/v10/orm"
+	"github.com/pkg/errors"
+)
+
+type MockFingerprintParams struct {
+	Serial string
+	Time   time.Time
+}
+
+func (m *MockFingerprintParams) ManifestSerial() string {
+	return m.Serial
+}
+
+func (m *MockFingerprintParams) CurTime() time.Time {
+	return m.Time
+}
+
+type MockUpdateManifests struct {
+	queries.QueryBase
+	SelectResponse      []common.UpdateManifest
+	SelectByVINResponse []common.StatusManifest
+	LoadResponse        *common.UpdateManifest
+	ECUUpdatesMock      func(man *common.UpdateManifestECU, vin string) ([]*common.UpdateManifestECU, error)
+	FlashPackManifest   common.UpdateManifest
+	DBMockHelper
+}
+
+func (m *MockUpdateManifests) Count(filter common.UpdateManifest) (int, error) {
+	return len(m.SelectResponse), m.Error
+}
+
+func (m *MockUpdateManifests) Delete(manifest *common.UpdateManifest) (orm.Result, error) {
+	err := validator.ValidateIDField(manifest.ID)
+	if err != nil {
+		return nil, err
+	}
+
+	return m.ORMResponse, m.Error
+}
+
+func (m *MockUpdateManifests) Insert(manifest *common.UpdateManifest) (orm.Result, error) {
+	manifest.ID += 1
+	return m.ORMResponse, m.Error
+}
+
+func (m *MockUpdateManifests) ECUInsert(ecu *common.UpdateManifestECU) (orm.Result, error) {
+	ecu.ID = 1
+	return m.ORMResponse, m.Error
+}
+
+func (m *MockUpdateManifests) FileInsert(file *common.UpdateManifestFile) (orm.Result, error) {
+	return m.ORMResponse, m.Error
+}
+
+func (m *MockUpdateManifests) Select(filter *common.UpdateManifest, paging *queries.PageQueryOptions) ([]common.UpdateManifest, error) {
+	m.LastFilter = filter
+	m.LastPaging = paging
+
+	return m.SelectResponse, m.Error
+}
+
+func (m *MockUpdateManifests) Archive(ids []int64, active bool) (orm.Result, error) {
+	return m.ORMResponse, m.Error
+}
+
+func (m *MockUpdateManifests) SelectByVIN(vin string, paging *queries.PageQueryOptions) ([]common.StatusManifest, error) {
+	m.LastPaging = paging
+
+	return m.SelectByVINResponse, m.Error
+}
+
+func (m *MockUpdateManifests) Update(manifest *common.UpdateManifest) (orm.Result, error) {
+	m.LastFilter = manifest
+
+	return m.ORMResponse, m.Error
+}
+
+func (m *MockUpdateManifests) Load(manifest *common.UpdateManifest) error {
+	m.LastFilter = manifest
+
+	if m.LoadResponse != nil {
+		m.LoadResponse.ID = manifest.ID
+		data, err := json.Marshal(m.LoadResponse)
+		if err != nil {
+			return errors.WithStack(err)
+		}
+		err = json.Unmarshal(data, manifest)
+		if err != nil {
+			return errors.WithStack(err)
+		}
+	}
+
+	return m.Error
+}
+
+func (m *MockUpdateManifests) Search(filter common.UpdateManifestSearch, paging *queries.PageQueryOptions) ([]common.UpdateManifest, error) {
+	m.LastFilter = &filter
+	m.LastPaging = paging
+
+	return m.SelectResponse, m.Error
+}
+
+func (m *MockUpdateManifests) SearchCount(filter common.UpdateManifestSearch) (int, error) {
+	m.LastFilter = &filter
+
+	return len(m.SelectResponse), m.Error
+}
+
+func (m *MockUpdateManifests) SetListResp(list interface{}) {
+	if list != nil {
+		m.SelectResponse = list.([]common.UpdateManifest)
+	} else {
+		m.SelectResponse = nil
+	}
+}
+
+func (m *MockUpdateManifests) ECURollback(man *common.UpdateManifestECU, vin string) ([]*common.UpdateManifestECU, error) {
+	return m.ECUUpdatesMock(man, vin)
+}
+
+func (m *MockUpdateManifests) AddSUMSVersion(manifest *common.UpdateManifest) (orm.Result, error) {
+	return m.ORMResponse, m.Error
+}
+
+func (m *MockUpdateManifests) SelectFlashPackByVersion(versionNumber string) (manifest common.UpdateManifest, err error) {
+	return m.FlashPackManifest, nil
+}