1 # -*- coding: utf-8 -*-
3 # Copyright 2018 Whitestack, LLC
4 # *************************************************************
6 # This file is part of OSM Monitoring module
7 # All Rights Reserved to Whitestack, LLC
9 # Licensed under the Apache License, Version 2.0 (the "License"); you may
10 # not use this file except in compliance with the License. You may obtain
11 # a copy of the License at
13 # http://www.apache.org/licenses/LICENSE-2.0
15 # Unless required by applicable law or agreed to in writing, software
16 # distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
17 # WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
18 # License for the specific language governing permissions and limitations
20 # For those usages not covered by the Apache License, Version 2.0 please
21 # contact: bdiaz@whitestack.com or glavado@whitestack.com
23 from unittest
import TestCase
, mock
25 from osm_mon
.collector
.utils
.collector
import CollectorUtils
26 from osm_mon
.core
.config
import Config
27 from osm_mon
.core
.database
import VimCredentialsRepository
, VimCredentials
30 class CollectorServiceTest(TestCase
):
33 self
.config
= Config()
35 @mock.patch
.object(VimCredentialsRepository
, "get")
36 @mock.patch('osm_mon.core.database.db')
37 def test_get_vim_type(self
, db
, get_credentials
):
38 mock_creds
= VimCredentials()
39 mock_creds
.id = 'test_id'
40 mock_creds
.user
= 'user'
41 mock_creds
.url
= 'url'
42 mock_creds
.password
= 'password'
43 mock_creds
.tenant_name
= 'tenant_name'
44 mock_creds
.type = 'openstack'
45 mock_creds
.config
= '{}'
47 get_credentials
.return_value
= mock_creds
48 vim_type
= CollectorUtils
.get_vim_type('test_id')
49 self
.assertEqual(vim_type
, 'openstack')