コード例 #1
0
func validateTotMemory(mem string) memory.MemSize {
	if mem == "" {
		fmt.Fprintf(os.Stderr, "-%s must be specified", totalFlag)
		os.Exit(1)
	}
	ms, err := memory.NewMemSizeFromString(mem)
	if err != nil {
		fmt.Fprintf(os.Stderr, "Error in -%s flag: %s", totalFlag, err)
		os.Exit(1)
	}
	if ms.LessThan(memory.MemSize(1024)) {
		fmt.Fprintf(os.Stderr, "Total memory (-%s flag) is less than 1K", totalFlag)
		os.Exit(1)
	}
	return ms
}
コード例 #2
0
	. "github.com/onsi/ginkgo"
	. "github.com/onsi/gomega"
)

var _ = Describe("MemorySize", func() {

	Context("basic constructors", func() {

		var (
			testItWorks func(string, int64)
			testItFails func(string)
		)

		BeforeEach(func() {
			testItWorks = func(memStr string, memVal int64) {
				ms, err := memory.NewMemSizeFromString(memStr)
				Ω(err).ShouldNot(HaveOccurred())
				Ω(ms.Bytes()).Should(Equal(memVal))
			}

			testItFails = func(memStr string) {
				ms, err := memory.NewMemSizeFromString(memStr)
				Ω(ms).Should(BeZero())
				Ω(err).Should(HaveOccurred())
			}
		})

		Context("succeeds", func() {
			It("accepts memory sizes in bytes, kilobytes, megabytes, or gigabytes", func() {
				testItWorks("1024b", kILO)
				testItWorks("1024B", kILO)