mirror of
https://github.com/amnezia-vpn/amneziawg-go.git
synced 2025-07-20 10:12:50 +02:00
37 lines
609 B
Go
37 lines
609 B
Go
package internal
|
|
|
|
type mockGenerator struct {
|
|
size int
|
|
}
|
|
|
|
func NewMockGenerator(size int) mockGenerator {
|
|
return mockGenerator{size: size}
|
|
}
|
|
|
|
func (m mockGenerator) Generate() []byte {
|
|
return make([]byte, m.size)
|
|
}
|
|
|
|
func (m mockGenerator) Size() int {
|
|
return m.size
|
|
}
|
|
|
|
func (m mockGenerator) Name() string {
|
|
return "mock"
|
|
}
|
|
|
|
type mockByteGenerator struct {
|
|
data []byte
|
|
}
|
|
|
|
func NewMockByteGenerator(data []byte) mockByteGenerator {
|
|
return mockByteGenerator{data: data}
|
|
}
|
|
|
|
func (bg mockByteGenerator) Generate() []byte {
|
|
return bg.data
|
|
}
|
|
|
|
func (bg mockByteGenerator) Size() int {
|
|
return len(bg.data)
|
|
}
|