Test plugin root_dir

Author: Garulf

tests/test_plugin.py

@@ -1,3 +1,4 @@
+import pytest
 from pyflowlauncher.plugin import Plugin
 
 
@@ -28,4 +29,25 @@ def test_add_methods():
 def test_settings():
     plugin = Plugin()
     plugin._client.recieve = lambda: {'method': 'settings', 'parameters': [], 'settings': {'test': 'test'}}
-    assert plugin.settings == {'test': 'test'}
+    assert plugin.settings == {'test': 'test'}
+
+
+def test_run_dir(tmp_path, monkeypatch):
+    monkeypatch.setattr('sys.argv', [tmp_path / 'plugin.py'])
+    plugin = Plugin()
+    assert plugin.plugin_run_dir() == tmp_path
+
+
+def test_root_dir(tmp_path, monkeypatch):
+    monkeypatch.setattr('sys.argv', [tmp_path / 'plugin.py'])
+    monkeypatch.setattr('pyflowlauncher.plugin.Path.exists', lambda _: True)
+    plugin = Plugin()
+    assert plugin.plugin_root_dir() == tmp_path
+
+
+def test_root_dir_not_found(tmp_path, monkeypatch):
+    monkeypatch.setattr('sys.argv', [tmp_path / 'plugin.py'])
+    monkeypatch.setattr('pyflowlauncher.plugin.Path.exists', lambda _: False)
+    plugin = Plugin()
+    with pytest.raises(FileNotFoundError):
+        assert plugin.plugin_root_dir() == tmp_path