import python
import semmle.python.dataflow.DataFlow

class TestConfiguration extends DataFlow::Configuration {
  TestConfiguration() { this = "Test configuration" }

  override predicate isSource(ControlFlowNode source) { source.(NameNode).getId() = "SOURCE" }

  override predicate isSink(ControlFlowNode sink) {
    exists(CallNode call |
      call.getFunction().(NameNode).getId() = "SINK" and
      sink = call.getAnArg()
    )
  }
}