gwen.web.eval.action.unit.BindElementLocator.scala Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of gwen-web Show documentation
Web automation engine for Gwen
The newest version!
/*
 * Copyright 2021-2024 Branko Juric, Brady Wood
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package gwen.web.eval.action.unit

import gwen.web.eval.WebContext
import gwen.web.eval.binding.RelativeSelectorType
import gwen.web.eval.binding.SelectorType
import gwen.web.eval.binding.LocatorKey

import gwen.core.behavior.BehaviorType
import gwen.core.eval.action.UnitStepAction
import gwen.core.node.GwenNode
import gwen.core.node.gherkin.Step

import scala.util.chaining._

class BindElementLocator(name: String, selectorType: SelectorType, expression: String, relative: Option[(RelativeSelectorType, String, Option[Int])], timeoutSecs: Option[Long], index: Option[Int]) extends UnitStepAction[WebContext] {

  override def apply(parent: GwenNode, step: Step, ctx: WebContext): Step = {

    step tap { _ =>

      checkStepRules(step, BehaviorType.Context, ctx)
      relative foreach { (rSelector, rElement, rWithinPixels) =>
        ctx.getLocatorBinding(rElement)
      }

      ctx.topScope.set(LocatorKey.baseKey(name), selectorType.toString)
      ctx.topScope.set(LocatorKey.selectorKey(name, selectorType), expression)

      if (relative.isEmpty) {
        RelativeSelectorType.values foreach { rSelectorType =>
          val rKey = LocatorKey.relativeKey(name, selectorType, rSelectorType)
          ctx.topScope.getOpt(rKey) foreach { _ =>
            ctx.topScope.set(rKey, null)
          }
          if (rSelectorType == RelativeSelectorType.near) {
            val rKeyWithinPixels = LocatorKey.relativeKeyWithinPixels(name, selectorType, rSelectorType)
            ctx.topScope.getOpt(rKeyWithinPixels) foreach { _ =>
              ctx.topScope.set(rKeyWithinPixels, null)
            }
          }
        }
      } else {
        relative foreach { (rSelectorType, rElement, rPixels) =>
          val rKey = LocatorKey.relativeKey(name, selectorType, rSelectorType)
          ctx.topScope.set(rKey, rElement)
          rPixels foreach { withinPixels =>
            val rKeyWithinPixels = LocatorKey.relativeKeyWithinPixels(name, selectorType, rSelectorType)
            ctx.topScope.set(rKeyWithinPixels, withinPixels.toString)
          }
        }
      }

      val iKey = LocatorKey.indexKey(name, selectorType)
      if (index.isEmpty) {
        ctx.topScope.getOpt(iKey) foreach { _ =>
          ctx.topScope.set(iKey, null)
        }
      } else {
        index foreach { idx =>
          ctx.topScope.set(iKey, idx.toString)
        }
      }

      val tKey = LocatorKey.timeoutSecsKey(name, selectorType)
      if (timeoutSecs.isEmpty) {
        ctx.topScope.getOpt(tKey) foreach { _ =>
          ctx.topScope.set(tKey, null)
        }
      } else {
        timeoutSecs foreach { secs =>
          ctx.topScope.set(tKey, secs.toString)
        }
      }

    }

  }

}