From bdc424e39de6b614579e83f62537f672cef98f55 Mon Sep 17 00:00:00 2001 From: Julien Pivotto Date: Wed, 6 Feb 2019 11:20:35 +0100 Subject: [PATCH] lang: Add to_lower and to_upper functions Signed-off-by: Julien Pivotto --- lang/funcs/core/corestrings/to_lower_func.go | 39 ++++++++++++++++ .../core/corestrings/to_lower_func_test.go | 44 +++++++++++++++++++ lang/funcs/core/corestrings/to_upper_func.go | 39 ++++++++++++++++ .../core/corestrings/to_upper_func_test.go | 44 +++++++++++++++++++ 4 files changed, 166 insertions(+) create mode 100644 lang/funcs/core/corestrings/to_lower_func.go create mode 100644 lang/funcs/core/corestrings/to_lower_func_test.go create mode 100644 lang/funcs/core/corestrings/to_upper_func.go create mode 100644 lang/funcs/core/corestrings/to_upper_func_test.go diff --git a/lang/funcs/core/corestrings/to_lower_func.go b/lang/funcs/core/corestrings/to_lower_func.go new file mode 100644 index 00000000..102efb3e --- /dev/null +++ b/lang/funcs/core/corestrings/to_lower_func.go @@ -0,0 +1,39 @@ +// Mgmt +// Copyright (C) 2013-2018+ James Shubin and the project contributors +// Written by James Shubin and the project contributors +// +// This program is free software: you can redistribute it and/or modify +// it under the terms of the GNU General Public License as published by +// the Free Software Foundation, either version 3 of the License, or +// (at your option) any later version. +// +// This program is distributed in the hope that it will be useful, +// but WITHOUT ANY WARRANTY; without even the implied warranty of +// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +// GNU General Public License for more details. +// +// You should have received a copy of the GNU General Public License +// along with this program. If not, see . + +package corestrings + +import ( + "strings" + + "github.com/purpleidea/mgmt/lang/funcs/simple" + "github.com/purpleidea/mgmt/lang/types" +) + +func init() { + simple.ModuleRegister(moduleName, "to_lower", &types.FuncValue{ + T: types.NewType("func(a str) str"), + V: ToLower, + }) +} + +// ToLower turns a string to lowercase. +func ToLower(input []types.Value) (types.Value, error) { + return &types.StrValue{ + V: strings.ToLower(input[0].Str()), + }, nil +} diff --git a/lang/funcs/core/corestrings/to_lower_func_test.go b/lang/funcs/core/corestrings/to_lower_func_test.go new file mode 100644 index 00000000..11dcae95 --- /dev/null +++ b/lang/funcs/core/corestrings/to_lower_func_test.go @@ -0,0 +1,44 @@ +// Mgmt +// Copyright (C) 2013-2018+ James Shubin and the project contributors +// Written by James Shubin and the project contributors +// +// This program is free software: you can redistribute it and/or modify +// it under the terms of the GNU General Public License as published by +// the Free Software Foundation, either version 3 of the License, or +// (at your option) any later version. +// +// This program is distributed in the hope that it will be useful, +// but WITHOUT ANY WARRANTY; without even the implied warranty of +// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +// GNU General Public License for more details. +// +// You should have received a copy of the GNU General Public License +// along with this program. If not, see . + +package corestrings + +import ( + "testing" + + "github.com/purpleidea/mgmt/lang/types" +) + +func testToLower(t *testing.T, input, expected string) { + inputStr := &types.StrValue{V: input} + value, err := ToLower([]types.Value{inputStr}) + if err != nil { + t.Error(err) + return + } + if value.Str() != expected { + t.Errorf("Invalid output, expected %s, got %s", expected, value.Str()) + } +} + +func TestToLowerSimple(t *testing.T) { + testToLower(t, "Hello", "hello") +} + +func TestToLowerSameString(t *testing.T) { + testToLower(t, "HELLO 22", "hello 22") +} diff --git a/lang/funcs/core/corestrings/to_upper_func.go b/lang/funcs/core/corestrings/to_upper_func.go new file mode 100644 index 00000000..06c149c5 --- /dev/null +++ b/lang/funcs/core/corestrings/to_upper_func.go @@ -0,0 +1,39 @@ +// Mgmt +// Copyright (C) 2013-2018+ James Shubin and the project contributors +// Written by James Shubin and the project contributors +// +// This program is free software: you can redistribute it and/or modify +// it under the terms of the GNU General Public License as published by +// the Free Software Foundation, either version 3 of the License, or +// (at your option) any later version. +// +// This program is distributed in the hope that it will be useful, +// but WITHOUT ANY WARRANTY; without even the implied warranty of +// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +// GNU General Public License for more details. +// +// You should have received a copy of the GNU General Public License +// along with this program. If not, see . + +package corestrings + +import ( + "strings" + + "github.com/purpleidea/mgmt/lang/funcs/simple" + "github.com/purpleidea/mgmt/lang/types" +) + +func init() { + simple.ModuleRegister(moduleName, "to_upper", &types.FuncValue{ + T: types.NewType("func(a str) str"), + V: ToUpper, + }) +} + +// ToUpper turns a string to uppercase. +func ToUpper(input []types.Value) (types.Value, error) { + return &types.StrValue{ + V: strings.ToUpper(input[0].Str()), + }, nil +} diff --git a/lang/funcs/core/corestrings/to_upper_func_test.go b/lang/funcs/core/corestrings/to_upper_func_test.go new file mode 100644 index 00000000..77a1c7a6 --- /dev/null +++ b/lang/funcs/core/corestrings/to_upper_func_test.go @@ -0,0 +1,44 @@ +// Mgmt +// Copyright (C) 2013-2018+ James Shubin and the project contributors +// Written by James Shubin and the project contributors +// +// This program is free software: you can redistribute it and/or modify +// it under the terms of the GNU General Public License as published by +// the Free Software Foundation, either version 3 of the License, or +// (at your option) any later version. +// +// This program is distributed in the hope that it will be useful, +// but WITHOUT ANY WARRANTY; without even the implied warranty of +// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +// GNU General Public License for more details. +// +// You should have received a copy of the GNU General Public License +// along with this program. If not, see . + +package corestrings + +import ( + "testing" + + "github.com/purpleidea/mgmt/lang/types" +) + +func testToUpper(t *testing.T, input, expected string) { + inputStr := &types.StrValue{V: input} + value, err := ToUpper([]types.Value{inputStr}) + if err != nil { + t.Error(err) + return + } + if value.Str() != expected { + t.Errorf("Invalid output, expected %s, got %s", expected, value.Str()) + } +} + +func TestToUpperSimple(t *testing.T) { + testToUpper(t, "Hello", "HELLO") +} + +func TestToUpperSameString(t *testing.T) { + testToUpper(t, "HELLO 22", "HELLO 22") +}