1 files changed, 111 insertions, 0 deletions
diff --git a/vendor/github.com/playwright-community/playwright-go/clock.go b/vendor/github.com/playwright-community/playwright-go/clock.go
new file mode 100644
index 0000000..8bab037
--- /dev/null
+++ b/vendor/github.com/playwright-community/playwright-go/clock.go
@@ -0,0 +1,111 @@
+package playwright
+
+import (
+	"errors"
+	"time"
+)
+
+type clockImpl struct {
+	browserCtx *browserContextImpl
+}
+
+func newClock(bCtx *browserContextImpl) Clock {
+	return &clockImpl{
+		browserCtx: bCtx,
+	}
+}
+
+func (c *clockImpl) FastForward(ticks interface{}) error {
+	params, err := parseTicks(ticks)
+	if err != nil {
+		return err
+	}
+
+	_, err = c.browserCtx.channel.Send("clockFastForward", params)
+	return err
+}
+
+func (c *clockImpl) Install(options ...ClockInstallOptions) (err error) {
+	params := map[string]any{}
+	if len(options) == 1 {
+		if options[0].Time != nil {
+			params, err = parseTime(options[0].Time)
+			if err != nil {
+				return err
+			}
+		}
+	}
+
+	_, err = c.browserCtx.channel.Send("clockInstall", params)
+
+	return err
+}
+
+func (c *clockImpl) PauseAt(time interface{}) error {
+	params, err := parseTime(time)
+	if err != nil {
+		return err
+	}
+
+	_, err = c.browserCtx.channel.Send("clockPauseAt", params)
+	return err
+}
+
+func (c *clockImpl) Resume() error {
+	_, err := c.browserCtx.channel.Send("clockResume")
+	return err
+}
+
+func (c *clockImpl) RunFor(ticks interface{}) error {
+	params, err := parseTicks(ticks)
+	if err != nil {
+		return err
+	}
+
+	_, err = c.browserCtx.channel.Send("clockRunFor", params)
+	return err
+}
+
+func (c *clockImpl) SetFixedTime(time interface{}) error {
+	params, err := parseTime(time)
+	if err != nil {
+		return err
+	}
+
+	_, err = c.browserCtx.channel.Send("clockSetFixedTime", params)
+	return err
+}
+
+func (c *clockImpl) SetSystemTime(time interface{}) error {
+	params, err := parseTime(time)
+	if err != nil {
+		return err
+	}
+
+	_, err = c.browserCtx.channel.Send("clockSetSystemTime", params)
+	return err
+}
+
+func parseTime(t interface{}) (map[string]any, error) {
+	switch v := t.(type) {
+	case int, int64:
+		return map[string]any{"timeNumber": v}, nil
+	case string:
+		return map[string]any{"timeString": v}, nil
+	case time.Time:
+		return map[string]any{"timeNumber": v.UnixMilli()}, nil
+	default:
+		return nil, errors.New("time should be one of: int, int64, string, time.Time")
+	}
+}
+
+func parseTicks(ticks interface{}) (map[string]any, error) {
+	switch v := ticks.(type) {
+	case int, int64:
+		return map[string]any{"ticksNumber": v}, nil
+	case string:
+		return map[string]any{"ticksString": v}, nil
+	default:
+		return nil, errors.New("ticks should be one of: int, int64, string")
+	}
+}