Merge pull request #667 from fomars/develop

test pandora remote ammo
yandex · Nov 6, 2018 · bdb2d7f · bdb2d7f
2 parents 03055ca + 324b293
commit bdb2d7f
Showing 1 changed file with 27 additions and 0 deletions.
diff --git a/yandextank/plugins/Pandora/tests/test_pandora_plugin.py b/yandextank/plugins/Pandora/tests/test_pandora_plugin.py
@@ -0,0 +1,27 @@
+import pytest
+
+from yandextank.plugins.Pandora import Plugin
+
+
+@pytest.mark.parametrize('cfg, expected', [
+    (
+        {'pools': [
+            {
+                'ammo': {'uri-headers': '[User-Agent: Wget/1.13.4 (linux-gnu)] [Host: foo.ru] [Accept-Encoding: gzip,deflate,sdch]',
+                         'type': 'uri',
+                         'file': 'https://raw.githubusercontent.com/yandex/yandex-tank/develop/README.md'
+                         }
+            }]},
+        {'pools': [
+            {
+                'ammo': {'uri-headers': '[User-Agent: Wget/1.13.4 (linux-gnu)] [Host: foo.ru] [Accept-Encoding: gzip,deflate,sdch]',
+                         'type': 'uri',
+                         'file': 'some local file'}
+            }]}
+    )
+])
+def test_patch_config(cfg, expected):
+    plugin = Plugin(None, None)
+    # '/tmp/9b73d966bcbf27467d4c4190cfe58c2a.downloaded_resource'
+    filename = plugin.patch_config(cfg)['pools'][0]['ammo']['file']
+    assert filename.endswith('.downloaded_resource')