From f8eaa9d23f36e532fa6af678afa9609f3c87a430 Mon Sep 17 00:00:00 2001
From: Ivan Krutov <vania-pooh@vania-pooh.com>
Date: Thu, 30 Nov 2017 09:02:57 +0300
Subject: [PATCH] Added ability to override video frame rate (fixes #287)

---
 docs/special-capabilities.adoc | 9 ++++++++-
 service/docker.go              | 4 ++++
 service_test.go                | 2 ++
 session/session.go             | 1 +
 4 files changed, 15 insertions(+), 1 deletion(-)

diff --git a/docs/special-capabilities.adoc b/docs/special-capabilities.adoc
index 036ea217..a779dd47 100644
--- a/docs/special-capabilities.adoc
+++ b/docs/special-capabilities.adoc
@@ -37,7 +37,7 @@ You should manually resize window to desired width and height.
 You need to use `setSize` method instead.
 ====
 
-=== Video Recording: enableVideo, videoName, videoScreenSize
+=== Video Recording: enableVideo, videoName, videoScreenSize, videoFrameRate
 
 NOTE: This feature requires some preparation. Please refer <<Video Recording>> section for details.
 
@@ -69,6 +69,13 @@ resolution is less than actual, screen on video will be trimmed starting from to
 videoScreenSize: "1024x768"
 ----
 
+* Default video frame rate is `12` frames per second. Specifying `videoFrameRate` capability changes this value:
+
+.Type: int
+----
+videoFrameRate: 24
+----
+
 === Custom Test Name: name
 
 For debugging purposes it is often useful to give a distinct name to every test case.
diff --git a/service/docker.go b/service/docker.go
index f1625e21..bddd837d 100644
--- a/service/docker.go
+++ b/service/docker.go
@@ -280,6 +280,10 @@ func startVideoContainer(ctx context.Context, cl *client.Client, requestId uint6
 	if videoScreenSize != "" {
 		env = append(env, fmt.Sprintf("VIDEO_SIZE=%s", videoScreenSize))
 	}
+	videoFrameRate := caps.VideoFrameRate
+	if videoFrameRate > 0 {
+		env = append(env, fmt.Sprintf("FRAME_RATE=%d", videoFrameRate))
+	}
 	log.Printf("[%d] [CREATING_VIDEO_CONTAINER] [%s]\n", requestId, videoContainerImage)
 	videoContainer, err := cl.ContainerCreate(ctx,
 		&ctr.Config{
diff --git a/service_test.go b/service_test.go
index 59183014..1475ce7b 100644
--- a/service_test.go
+++ b/service_test.go
@@ -233,6 +233,8 @@ func createDockerStarter(t *testing.T, env *service.Environment, cfg *config.Con
 		ScreenResolution:      "1024x768",
 		VNC:                   true,
 		Video:                 true,
+		VideoScreenSize:       "1024x768",
+		VideoFrameRate:        25,
 		HostsEntries:          "example.com:192.168.0.1,test.com:192.168.0.2",
 		ApplicationContainers: "one,two",
 		TimeZone:              "Europe/Moscow",
diff --git a/session/session.go b/session/session.go
index 3f3ec4ca..ba20dc2f 100644
--- a/session/session.go
+++ b/session/session.go
@@ -14,6 +14,7 @@ type Caps struct {
 	Video                 bool   `json:"enableVideo"`
 	VideoName             string `json:"videoName"`
 	VideoScreenSize       string `json:"videoScreenSize"`
+	VideoFrameRate        uint16 `json:"videoFrameRate"`
 	TestName              string `json:"name"`
 	TimeZone              string `json:"timeZone"`
 	ContainerHostname     string `json:"containerHostname"`