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
21 # For those usages not covered by the Apache License, Version 2.0 please
22 # contact: bdiaz@whitestack.com or glavado@whitestack.com
25 """Test an end to end Openstack vim_account requests."""
31 from peewee
import SqliteDatabase
33 from osm_mon
.core
import database
34 from osm_mon
.core
.auth
import AuthManager
35 from osm_mon
.core
.database
import DatabaseManager
, VimCredentials
, Alarm
37 log
= logging
.getLogger(__name__
)
39 test_db
= SqliteDatabase(':memory:')
41 MODELS
= [VimCredentials
, Alarm
]
44 class VimAccountTest(unittest
.TestCase
):
50 test_db
.drop_tables(MODELS
)
51 test_db
.create_tables(MODELS
)
52 self
.auth_manager
= AuthManager()
53 self
.database_manager
= DatabaseManager()
54 self
.database_manager
.create_tables()
60 def test_create_edit_delete_vim_account(self
):
61 """Test vim_account creation message from KafkaProducer."""
62 # Set-up message, producer and consumer for tests
66 "vim_type": "openstack",
67 "vim_url": "auth_url",
69 "vim_password": "password",
70 "vim_tenant_name": "tenant",
76 self
.auth_manager
.store_auth_credentials(create_payload
)
78 creds
= self
.auth_manager
.get_credentials('test_id')
80 self
.assertIsNotNone(creds
)
81 self
.assertEqual(creds
.name
, create_payload
['name'])
82 self
.assertEqual(json
.loads(creds
.config
), create_payload
['config'])
84 # Set-up message, producer and consumer for tests
87 "name": "test_name_edited",
88 "vim_type": "openstack",
89 "vim_url": "auth_url",
91 "vim_password": "password",
92 "vim_tenant_name": "tenant",
95 "foo_edited": "bar_edited"
99 self
.auth_manager
.store_auth_credentials(edit_payload
)
101 creds
= self
.auth_manager
.get_credentials('test_id')
103 self
.assertEqual(creds
.name
, edit_payload
['name'])
104 self
.assertEqual(json
.loads(creds
.config
), edit_payload
['config'])
110 self
.auth_manager
.delete_auth_credentials(delete_payload
)
112 creds
= self
.auth_manager
.get_credentials('test_id')
113 self
.assertIsNone(creds
)