Add get_package_source_version()

1 files changed, 23 insertions, 0 deletions

diff --git a/test/test_api.py b/test/test_api.py
index a6f74c1..405c0d6 100644
--- a/test/test_api.py
+++ b/test/test_api.py
@@ -364,6 +364,29 @@ def test_get_package_version(requests_mock, binarylist, binary_fileinfo):
     assert version == expected_version
 
 
+def test_get_package_source_version(requests_mock, package_directory, spec_file_1):
+    matcher = re.compile('/source/.*/.*')
+    requests_mock.get(matcher, text=package_directory)
+    matcher = re.compile('/source/.*/.*/.*')
+    requests_mock.get(matcher, content=spec_file_1)
+
+    specfile = spec_file_1.decode('utf-8').splitlines()
+    full_version = api.get_package_source_version(prj, 'package')
+    print(full_version)
+    version, release = full_version.split('-')
+    match_version = False
+    match_release = False
+    for line in specfile:
+        if line.strip().startswith('Version'):
+            assert version in line
+            match_version = True
+        if line.strip().startswith('Release'):
+            assert release in line
+            match_release = True
+
+    assert match_version and match_release
+
+
 def test_get_project_flags(requests_mock, project_meta):
     matcher = prj_meta_re
     requests_mock.get(matcher, text=project_meta)