1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
|
// Copyright Vespa.ai. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package jvm
import (
"testing"
"github.com/stretchr/testify/assert"
"github.com/vespa-engine/vespa/client/go/internal/admin/prog"
)
type dummyContainer struct{ containerBase }
func (*dummyContainer) ArgForMain() string { return "arg-for-main" }
func (*dummyContainer) exportExtraEnv(ps *prog.Spec) {}
func newDummyContainer() Container {
var dc dummyContainer
dc.serviceName = "foo"
dc.jvmOpts = NewOptions(&dc)
return &dc
}
func TestHeapSizeSimple(t *testing.T) {
var (
aa = MegaBytesOfMemory(123)
bb = MegaBytesOfMemory(234)
)
o := newDummyContainer().JvmOptions()
assert.Equal(t, aa, o.CurMinHeapSize(aa))
assert.Equal(t, bb, o.CurMaxHeapSize(bb))
assert.Equal(t, 2, len(o.jvmArgs))
assert.Equal(t, "-Xms123m", o.jvmArgs[0])
assert.Equal(t, "-Xmx234m", o.jvmArgs[1])
}
func TestHeapSizeMulti(t *testing.T) {
var (
aa = MegaBytesOfMemory(123)
bb = MegaBytesOfMemory(234)
cc = MegaBytesOfMemory(456)
dd = MegaBytesOfMemory(567)
)
o := newDummyContainer().JvmOptions()
assert.Equal(t, aa, o.CurMinHeapSize(aa))
assert.Equal(t, aa, o.CurMaxHeapSize(aa))
assert.Equal(t, 2, len(o.jvmArgs))
o.AppendOption("-Xms:234m")
o.AppendOption("-Xmx:456M")
assert.Equal(t, 4, len(o.jvmArgs))
assert.Equal(t, bb, o.CurMinHeapSize(aa))
assert.Equal(t, bb, o.CurMinHeapSize(dd))
assert.Equal(t, cc, o.CurMaxHeapSize(aa))
assert.Equal(t, cc, o.CurMaxHeapSize(dd))
o.AppendOption("-Xms1G")
o.AppendOption("-Xmx:2g")
assert.Equal(t, GigaBytesOfMemory(1), o.CurMinHeapSize(aa))
assert.Equal(t, GigaBytesOfMemory(2), o.CurMaxHeapSize(aa))
o.AppendOption("-Xms:16777216K")
o.AppendOption("-Xmx32505856k")
assert.Equal(t, KiloBytesOfMemory(16777216), o.CurMinHeapSize(aa))
assert.Equal(t, KiloBytesOfMemory(32505856), o.CurMaxHeapSize(aa))
}
func TestHeapSizeAdd(t *testing.T) {
var (
gg = MegaBytesOfMemory(12345)
hh = MegaBytesOfMemory(23456)
)
o := newDummyContainer().JvmOptions()
o.AddDefaultHeapSizeArgs(gg, hh)
assert.Equal(t, 3, len(o.jvmArgs))
assert.Equal(t, "-Xms12345m", o.jvmArgs[0])
assert.Equal(t, "-Xmx23456m", o.jvmArgs[1])
assert.Equal(t, "-XX:+UseTransparentHugePages", o.jvmArgs[2])
}
func TestHeapSizeNoAdd(t *testing.T) {
var (
bb = MegaBytesOfMemory(234)
cc = MegaBytesOfMemory(456)
)
o := newDummyContainer().JvmOptions()
o.AppendOption("-Xms128k")
o.AppendOption("-Xmx1280k")
o.AddDefaultHeapSizeArgs(bb, cc)
assert.Equal(t, 2, len(o.jvmArgs))
assert.Equal(t, "-Xms128k", o.jvmArgs[0])
assert.Equal(t, "-Xmx1280k", o.jvmArgs[1])
}
|