testutils

Documentation

Index

• type DBMock
  • func (mo *DBMock) Connect(ctx context.Context, uri string) error
  • func (mo *DBMock) Disconnect(ctx context.Context) error
  • func (mo *DBMock) GetCPEByKeyword(ctx context.Context, keywords db.Keyword, opts ...db.QueryOptions) (*db.Result[schema.Cpe], error)
  • func (mo *DBMock) GetCPEByMatchString(ctx context.Context, cpeNameMatchString string, opts ...db.QueryOptions) (*db.Result[schema.Cpe], error)
  • func (mo *DBMock) GetCPEByName(ctx context.Context, cpeName string) (*schema.Cpe, error)
  • func (mo *DBMock) GetCVEByCPE(ctx context.Context, cpeName string, opts ...db.QueryOptions) (*db.Result[schema.Cve], error)
  • func (mo *DBMock) GetCVEByID(ctx context.Context, cveId string) (*schema.Cve, error)
  • func (mo *DBMock) GetCVEByKeyword(ctx context.Context, keywords db.Keyword, opts ...db.QueryOptions) (*db.Result[schema.Cve], error)
  • func (mo *DBMock) ID() string
  • func (mo *DBMock) Init(ctx context.Context) error
  • func (mo *DBMock) IsConnected(ctx context.Context) error
  • func (mo *DBMock) UpsertCPE(ctx context.Context, cve schema.Cpe) (bool, error)
  • func (mo *DBMock) UpsertCPEs(ctx context.Context, cpe []schema.Cpe) (int, error)
  • func (mo *DBMock) UpsertCVE(ctx context.Context, cve schema.Cve) (bool, error)
  • func (mo *DBMock) UpsertCVEs(ctx context.Context, cve []schema.Cve) (int, error)
• type Mock

Types

type DBMock

type DBMock struct {
	GetCVEByIDResult      *schema.Cve
	GetCVEByCPEResult     []schema.Cve
	GetCVEByKeywordResult []schema.Cve
	GetCPEByNameResult    *schema.Cpe
	GetCPEByMatchResult   []schema.Cpe
	GetCPEByKeywordResult []schema.Cpe
	GetNVDError           error
}

DBMock implememt db.DB interface to mock db client

func (*DBMock) Connect

func (mo *DBMock) Connect(ctx context.Context, uri string) error

func (*DBMock) Disconnect

func (mo *DBMock) Disconnect(ctx context.Context) error

func (*DBMock) GetCPEByKeyword

func (mo *DBMock) GetCPEByKeyword(ctx context.Context, keywords db.Keyword, opts ...db.QueryOptions) (*db.Result[schema.Cpe], error)

func (*DBMock) GetCPEByMatchString

func (mo *DBMock) GetCPEByMatchString(ctx context.Context, cpeNameMatchString string, opts ...db.QueryOptions) (*db.Result[schema.Cpe], error)

func (*DBMock) GetCPEByName

func (mo *DBMock) GetCPEByName(ctx context.Context, cpeName string) (*schema.Cpe, error)

func (*DBMock) GetCVEByCPE

func (mo *DBMock) GetCVEByCPE(ctx context.Context, cpeName string, opts ...db.QueryOptions) (*db.Result[schema.Cve], error)

func (*DBMock) GetCVEByID

func (mo *DBMock) GetCVEByID(ctx context.Context, cveId string) (*schema.Cve, error)

func (*DBMock) GetCVEByKeyword

func (mo *DBMock) GetCVEByKeyword(ctx context.Context, keywords db.Keyword, opts ...db.QueryOptions) (*db.Result[schema.Cve], error)

func (*DBMock) ID

func (mo *DBMock) ID() string

func (*DBMock) Init

func (mo *DBMock) Init(ctx context.Context) error

func (*DBMock) IsConnected

func (mo *DBMock) IsConnected(ctx context.Context) error

func (*DBMock) UpsertCPE

func (mo *DBMock) UpsertCPE(ctx context.Context, cve schema.Cpe) (bool, error)

func (*DBMock) UpsertCPEs

func (mo *DBMock) UpsertCPEs(ctx context.Context, cpe []schema.Cpe) (int, error)

func (*DBMock) UpsertCVE

func (mo *DBMock) UpsertCVE(ctx context.Context, cve schema.Cve) (bool, error)

func (*DBMock) UpsertCVEs

func (mo *DBMock) UpsertCVEs(ctx context.Context, cve []schema.Cve) (int, error)

type Mock

type Mock struct {
	DBMock
	ReqPath   string
	ReqBody   string
	Assertion func(code int, body string)
}

Mock mocks both DB and queue client for testing