forked from erp-dev/erp
62 lines
1.9 KiB
Python
62 lines
1.9 KiB
Python
from io import StringIO
|
|
|
|
from django.core.cache import cache
|
|
from django.core.management import call_command
|
|
from django.test import TestCase, override_settings
|
|
from rest_framework.test import APIClient
|
|
|
|
from flower.app_version import APP_VERSION_CACHE_KEY
|
|
|
|
|
|
class AppVersionAPITest(TestCase):
|
|
def setUp(self):
|
|
cache.delete(APP_VERSION_CACHE_KEY)
|
|
self.client = APIClient()
|
|
|
|
def tearDown(self):
|
|
cache.delete(APP_VERSION_CACHE_KEY)
|
|
|
|
@override_settings(
|
|
APP_LATEST_VERSION_MAJOR=1,
|
|
APP_LATEST_VERSION_MINOR=2,
|
|
APP_LATEST_VERSION_BUILD=345,
|
|
APP_DOWNLOAD_URL="https://example.com/app.apk",
|
|
)
|
|
def test_app_version_api_returns_settings_payload_without_auth(self):
|
|
response = self.client.get("/api/app-version/")
|
|
|
|
self.assertEqual(response.status_code, 200)
|
|
self.assertEqual(
|
|
response.data,
|
|
{
|
|
"latest_version": {
|
|
"major": 1,
|
|
"minor": 2,
|
|
"build": 345,
|
|
},
|
|
"download_url": "https://example.com/app.apk",
|
|
},
|
|
)
|
|
|
|
def test_set_app_version_command_updates_cached_api_payload(self):
|
|
output = StringIO()
|
|
call_command(
|
|
"set_app_version",
|
|
"--major",
|
|
"2",
|
|
"--minor",
|
|
"5",
|
|
"--build",
|
|
"1001",
|
|
"--download-url",
|
|
"https://example.com/latest.apk",
|
|
stdout=output,
|
|
)
|
|
|
|
response = self.client.get("/api/app-version/")
|
|
|
|
self.assertEqual(response.status_code, 200)
|
|
self.assertEqual(response.data["latest_version"], {"major": 2, "minor": 5, "build": 1001})
|
|
self.assertEqual(response.data["download_url"], "https://example.com/latest.apk")
|
|
self.assertIn("'build': 1001", output.getvalue())
|