mirror of
https://github.com/netbox-community/netbox.git
synced 2024-05-10 07:54:54 +00:00
d8784d4155
* WIP * WIP * Add git sync * Fix file hashing * Add last_synced to DataSource * Build out UI & API resources * Add status field to DataSource * Add UI control to sync data source * Add API endpoint to sync data sources * Fix display of DataSource job results * DataSource password should be write-only * General cleanup * Add data file UI view * Punt on HTTP, FTP support for now * Add DataSource URL validation * Add HTTP proxy support to git fetcher * Add management command to sync data sources * DataFile REST API endpoints should be read-only * Refactor fetch methods into backend classes * Replace auth & git branch fields with general-purpose parameters * Fix last_synced time * Render discrete form fields for backend parameters * Enable dynamic edit form for DataSource * Register DataBackend classes in application registry * Add search indexers for DataSource, DataFile * Add single & bulk delete views for DataFile * Add model documentation * Convert DataSource to a primary model * Introduce pre_sync & post_sync signals * Clean up migrations * Rename url to source_url * Clean up filtersets * Add API & filterset tests * Add view tests * Add initSelect() to HTMX refresh handler * Render DataSourceForm fieldsets dynamically * Update compiled static resources
94 lines
3.0 KiB
Python
94 lines
3.0 KiB
Python
from django.urls import reverse
|
|
from django.utils import timezone
|
|
|
|
from utilities.testing import APITestCase, APIViewTestCases
|
|
from ..choices import *
|
|
from ..models import *
|
|
|
|
|
|
class AppTest(APITestCase):
|
|
|
|
def test_root(self):
|
|
url = reverse('core-api:api-root')
|
|
response = self.client.get('{}?format=api'.format(url), **self.header)
|
|
|
|
self.assertEqual(response.status_code, 200)
|
|
|
|
|
|
class DataSourceTest(APIViewTestCases.APIViewTestCase):
|
|
model = DataSource
|
|
brief_fields = ['display', 'id', 'name', 'url']
|
|
bulk_update_data = {
|
|
'enabled': False,
|
|
'description': 'foo bar baz',
|
|
}
|
|
|
|
@classmethod
|
|
def setUpTestData(cls):
|
|
data_sources = (
|
|
DataSource(name='Data Source 1', type=DataSourceTypeChoices.LOCAL, source_url='file:///var/tmp/source1/'),
|
|
DataSource(name='Data Source 2', type=DataSourceTypeChoices.LOCAL, source_url='file:///var/tmp/source2/'),
|
|
DataSource(name='Data Source 3', type=DataSourceTypeChoices.LOCAL, source_url='file:///var/tmp/source3/'),
|
|
)
|
|
DataSource.objects.bulk_create(data_sources)
|
|
|
|
cls.create_data = [
|
|
{
|
|
'name': 'Data Source 4',
|
|
'type': DataSourceTypeChoices.GIT,
|
|
'source_url': 'https://example.com/git/source4'
|
|
},
|
|
{
|
|
'name': 'Data Source 5',
|
|
'type': DataSourceTypeChoices.GIT,
|
|
'source_url': 'https://example.com/git/source5'
|
|
},
|
|
{
|
|
'name': 'Data Source 6',
|
|
'type': DataSourceTypeChoices.GIT,
|
|
'source_url': 'https://example.com/git/source6'
|
|
},
|
|
]
|
|
|
|
|
|
class DataFileTest(
|
|
APIViewTestCases.GetObjectViewTestCase,
|
|
APIViewTestCases.ListObjectsViewTestCase,
|
|
APIViewTestCases.GraphQLTestCase
|
|
):
|
|
model = DataFile
|
|
brief_fields = ['display', 'id', 'path', 'url']
|
|
|
|
@classmethod
|
|
def setUpTestData(cls):
|
|
datasource = DataSource.objects.create(
|
|
name='Data Source 1',
|
|
type=DataSourceTypeChoices.LOCAL,
|
|
source_url='file:///var/tmp/source1/'
|
|
)
|
|
|
|
data_files = (
|
|
DataFile(
|
|
source=datasource,
|
|
path='dir1/file1.txt',
|
|
last_updated=timezone.now(),
|
|
size=1000,
|
|
hash='442da078f0111cbdf42f21903724f6597c692535f55bdfbbea758a1ae99ad9e1'
|
|
),
|
|
DataFile(
|
|
source=datasource,
|
|
path='dir1/file2.txt',
|
|
last_updated=timezone.now(),
|
|
size=2000,
|
|
hash='a78168c7c97115bafd96450ed03ea43acec495094c5caa28f0d02e20e3a76cc2'
|
|
),
|
|
DataFile(
|
|
source=datasource,
|
|
path='dir1/file3.txt',
|
|
last_updated=timezone.now(),
|
|
size=3000,
|
|
hash='12b8827a14c4d5a2f30b6c6e2b7983063988612391c6cbe8ee7493b59054827a'
|
|
),
|
|
)
|
|
DataFile.objects.bulk_create(data_files)
|