123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161 |
- /*
- * ECAL
- *
- * Copyright 2020 Matthias Ladkau. All rights reserved.
- *
- * This Source Code Form is subject to the terms of the MIT
- * License, If a copy of the MIT License was not distributed with this
- * file, You can obtain one at https://opensource.org/licenses/MIT.
- */
- package interpreter
- import (
- "flag"
- "fmt"
- "os"
- "testing"
- "devt.de/krotik/common/datautil"
- "devt.de/krotik/common/timeutil"
- "devt.de/krotik/ecal/engine"
- "devt.de/krotik/ecal/parser"
- "devt.de/krotik/ecal/scope"
- "devt.de/krotik/ecal/util"
- )
- // Main function for all tests in this package
- func TestMain(m *testing.M) {
- flag.Parse()
- // Run the tests
- res := m.Run()
- // Check if all nodes have been tested
- for n := range providerMap {
- if _, ok := usedNodes[n]; !ok {
- fmt.Println("Not tested node: ", n)
- }
- }
- os.Exit(res)
- }
- // Used nodes map which is filled during unit testing. Prefilled only with nodes
- // which should not be encountered in ASTs.
- //
- var usedNodes = map[string]bool{
- parser.NodeEOF: true,
- }
- // Last used logger
- //
- var testlogger *util.MemoryLogger
- // Last used cron
- //
- var testcron *timeutil.Cron
- // Last used processor
- //
- var testprocessor engine.Processor
- func UnitTestEval(input string, vs parser.Scope) (interface{}, error) {
- return UnitTestEvalAndAST(input, vs, "")
- }
- func UnitTestEvalAndAST(input string, vs parser.Scope, expectedAST string) (interface{}, error) {
- return UnitTestEvalAndASTAndImport(input, vs, expectedAST, nil)
- }
- func UnitTestEvalAndASTAndImport(input string, vs parser.Scope, expectedAST string, importLocator util.ECALImportLocator) (interface{}, error) {
- var traverseAST func(n *parser.ASTNode)
- traverseAST = func(n *parser.ASTNode) {
- if n.Name == "" {
- panic(fmt.Sprintf("Node found with empty string name: %s", n))
- }
- usedNodes[n.Name] = true
- for _, cn := range n.Children {
- traverseAST(cn)
- }
- }
- // Parse the input
- erp := NewECALRuntimeProvider("ECALTestRuntime", importLocator, nil)
- testlogger = erp.Logger.(*util.MemoryLogger)
- // For testing we change the cron object to be a testing cron which goes
- // quickly through a day when started. To test cron functionality a test
- // needs to first specify a setCronTrigger and the sinks. Once this has
- // been done the testcron object needs to be started. It will go through
- // a day instantly and add a deterministic number of events (according to
- // the cronspec given to setCronTrigger for one day).
- erp.Cron.Stop()
- erp.Cron = timeutil.NewTestingCronDay()
- testcron = erp.Cron
- testprocessor = erp.Processor
- ast, err := parser.ParseWithRuntime("ECALEvalTest", input, erp)
- if err != nil {
- return nil, err
- }
- traverseAST(ast)
- if expectedAST != "" && ast.String() != expectedAST {
- return nil, fmt.Errorf("Unexpected AST result:\n%v", ast.String())
- }
- // Validate input
- if err := ast.Runtime.Validate(); err != nil {
- return nil, err
- }
- if vs == nil {
- vs = scope.NewScope(scope.GlobalScope)
- }
- return ast.Runtime.Eval(vs, make(map[string]interface{}))
- }
- /*
- addLogFunction adds a simple log function to a given Scope.
- */
- func addLogFunction(vs parser.Scope) *datautil.RingBuffer {
- buf := datautil.NewRingBuffer(20)
- vs.SetValue("testlog", &TestLogger{buf})
- return buf
- }
- /*
- TestLogger is a simple logger function which can be added to tess.
- */
- type TestLogger struct {
- buf *datautil.RingBuffer
- }
- func (tl *TestLogger) Run(instanceID string, vs parser.Scope, is map[string]interface{}, args []interface{}) (interface{}, error) {
- tl.buf.Add(fmt.Sprint(args...))
- return nil, nil
- }
- func (tl *TestLogger) DocString() (string, error) {
- return "testlogger docstring", nil
- }
- func (tl *TestLogger) String() string {
- return "TestLogger"
- }
- func (tl *TestLogger) MarshalJSON() ([]byte, error) {
- return []byte(tl.String()), nil
- }
|