def test_kuviz_publisher_create_new_apikey(self, mocker):
token = '1234'
setup_mocks(mocker, self.credentials, is_public=False, token=token)
layers = [Layer('fake_table', credentials=self.credentials)]
kuviz_publisher = KuvizPublisher(None)
assert kuviz_publisher._create_maps_api_keys(layers) == token
def test_kuviz_publisher_set_layers(self, mocker):
setup_mocks(mocker, self.credentials)
source_1 = Source(build_geodataframe([-10, 0], [-10, 0]))
source_2 = Source(build_geodataframe([0, 10], [10, 0]))
layer_1 = Layer(source_1)
layer_2 = Layer(source_2)
vmap = Map([layer_1, layer_2])
kuviz_publisher = KuvizPublisher(None)
kuviz_publisher.set_layers(vmap.layers)
assert kuviz_publisher._layers != vmap.layers
assert len(kuviz_publisher._layers) == len(vmap.layers)
def test_kuviz_publisher_instantiation(self, mocker):
setup_mocks(mocker, self.credentials)
kuviz_publisher = KuvizPublisher(None)
assert isinstance(kuviz_publisher, KuvizPublisher)
assert kuviz_publisher._layers == []
def test_kuviz_publisher_publish(self, mocker):
setup_mocks(mocker, self.credentials)
kuviz = CartoKuvizMock('fake_kuviz')
mocker.patch('cartoframes.viz.kuviz._create_kuviz', return_value=kuviz)
vmap = Map(Layer('fake_table', credentials=self.credentials))
html = 'fake_html'
kuviz_name = 'fake_name'
kuviz_publisher = KuvizPublisher(None)
kuviz_publisher.set_layers(vmap.layers)
result = kuviz_publisher.publish(html, kuviz_name, None)
assert kuviz_publisher.kuviz == kuviz
assert result == kuviz_to_dict(kuviz)
def test_kuviz_publisher_use_custom_api_key(self, mocker):
setup_mocks(mocker, self.credentials)
vmap = Map(Layer('fake_table', credentials=self.credentials))
kuviz_publisher = KuvizPublisher(None)
kuviz_publisher.set_layers(vmap.layers, 'fake_api_key')
layers = kuviz_publisher.get_layers()
assert layers[0].source.credentials == self.credentials
assert layers[0].credentials == ({
'username':
self.username,
'api_key':
'fake_api_key',
'base_url':
'https://{}.carto.com'.format(self.username)
})
def test_kuviz_publisher_use_only_base_url(self, mocker):
credentials = Credentials(base_url='https://fakeuser.carto.com')
setup_mocks(mocker, credentials)
vmap = Map(Layer('fake_table', credentials=credentials))
kuviz_publisher = KuvizPublisher(None)
kuviz_publisher.set_layers(vmap.layers)
layers = kuviz_publisher.get_layers()
assert layers[0].source.credentials == credentials
assert layers[0].credentials == ({
'username':
'******', # Default VL username
'api_key':
DEFAULT_PUBLIC,
'base_url':
'https://fakeuser.carto.com'
})
def test_kuviz_delete_publication(self, mocker):
setup_mocks(mocker, self.credentials)
html = 'fake_html'
kuviz_name = 'fake_name'
kuviz = CartoKuvizMock(kuviz_name)
KuvizManagerNamedtuple = namedtuple('KuvizManager', ['all'])
kuviz_manager_namedtuple = KuvizManagerNamedtuple(lambda: [kuviz])
mocker.patch('cartoframes.viz.kuviz._create_kuviz', return_value=kuviz)
mocker.patch('cartoframes.viz.kuviz._get_kuviz_manager',
return_value=kuviz_manager_namedtuple)
mock = mocker.patch(
'tests.unit.mocks.kuviz_mock.CartoKuvizMock.delete')
vmap = Map(Layer('fake_table', credentials=self.credentials))
kuviz_publisher = KuvizPublisher(None)
kuviz_publisher.set_layers(vmap.layers)
kuviz_publisher.publish(html, kuviz_name, None)
delete_publication(kuviz_name, credentials=self.credentials)
mock.assert_called_once_with()
def test_kuviz_publisher_update_fail(self, mocker):
setup_mocks(mocker, self.credentials)
vmap = Map(Layer('fake_table', credentials=self.credentials))
html = 'fake_html'
kuviz_name = 'fake_name'
kuviz_publisher = KuvizPublisher(None)
kuviz_publisher.set_layers(vmap.layers)
with pytest.raises(Exception):
kuviz_publisher.update(html, kuviz_name, None)
def test_kuviz_all_publications(self, mocker):
setup_mocks(mocker, self.credentials)
html = 'fake_html'
kuviz_name = 'fake_name'
kuviz = CartoKuvizMock(kuviz_name)
KuvizManagerNamedtuple = namedtuple('KuvizManager', ['all'])
kuviz_manager_namedtuple = KuvizManagerNamedtuple(lambda: [kuviz])
mocker.patch('cartoframes.viz.kuviz._create_kuviz', return_value=kuviz)
mocker.patch('cartoframes.viz.kuviz._get_kuviz_manager',
return_value=kuviz_manager_namedtuple)
vmap = Map(Layer('fake_table', credentials=self.credentials))
kuviz_publisher = KuvizPublisher(None)
kuviz_publisher.set_layers(vmap.layers)
kuviz_publisher.publish(html, kuviz_name, None)
assert all_publications(
credentials=self.credentials)[0]['name'] == kuviz_name
