95 lines
3.8 KiB
Python
95 lines
3.8 KiB
Python
#!/usr/bin/python
|
|
# -*- coding: utf-8 -*-
|
|
#
|
|
# This Modularity Testing Framework helps you to write tests for modules
|
|
# Copyright (C) 2017 Red Hat, Inc.
|
|
#
|
|
# This program is free software; you can redistribute it and/or modify
|
|
# it under the terms of the GNU General Public License as published by
|
|
# he Free Software Foundation; either version 2 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License along
|
|
# with this program; if not, write to the Free Software Foundation, Inc.,
|
|
# 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
|
|
#
|
|
# Authors: Rado Pitonak <rpitonak@redhat.com>
|
|
#
|
|
|
|
from avocado import main
|
|
from avocado.core import exceptions
|
|
from moduleframework import module_framework
|
|
import time
|
|
|
|
|
|
class DBCheck(module_framework.AvocadoTest):
|
|
"""
|
|
:avocado: enable
|
|
"""
|
|
|
|
MONGO_CMD = 'mongo "$MONGODB_DATABASE" -u "$MONGODB_USER" -p "$MONGODB_PASSWORD"'
|
|
|
|
# authentication options for admin user
|
|
# https://docs.mongodb.com/manual/reference/program/mongo/#mongo-shell-authentication-options
|
|
MONGO_ADMIN_CMD = 'mongo "$MONGODB_DATABASE" -u "admin" -p "$MONGODB_ADMIN_PASSWORD" --authenticationDatabase admin --authenticationMechanism SCRAM-SHA-1'
|
|
|
|
def testConnectToDB(self):
|
|
"""
|
|
Test connection to database
|
|
"""
|
|
|
|
query = "db.getSiblingDB('test_database');"
|
|
|
|
self.start()
|
|
time.sleep(3)
|
|
|
|
self.run('{} --eval "{}"'.format(self.MONGO_CMD, query))
|
|
|
|
def testUserPrivileges(self):
|
|
"""
|
|
Testing user privileges.
|
|
"""
|
|
|
|
self.start()
|
|
time.sleep(3)
|
|
|
|
# mongo db queries stored in dictionary, written in order which they are called
|
|
queries = {"insert1": "db.testData.insert({ y : 1 });",
|
|
"insert2": "db.testData.insert({ z : 2 });",
|
|
"print_json": "db.testData.find().forEach(printjson);",
|
|
"count": "db.testData.count();",
|
|
"data_drop": "db.testData.drop();",
|
|
"db_drop": "db.dropDatabase();"}
|
|
|
|
self.run('{} --eval "{}"'.format(self.MONGO_CMD, queries["insert1"]))
|
|
self.run('{} --eval "{}"'.format(self.MONGO_CMD, queries["insert2"]))
|
|
self.run('{} --eval "{}"'.format(self.MONGO_CMD, queries["print_json"]))
|
|
self.run('{} --eval "{}"'.format(self.MONGO_CMD, queries["count"]))
|
|
self.run('{} --eval "{}"'.format(self.MONGO_CMD, queries["data_drop"]))
|
|
self.run('{} --eval "{}"'.format(self.MONGO_CMD, queries["db_drop"]))
|
|
|
|
def testAdminPrivileges(self):
|
|
"""
|
|
Testing privileges for admin user
|
|
"""
|
|
self.start()
|
|
time.sleep(3)
|
|
|
|
# mongo db queries stored in dictionary, written in order which they are called
|
|
queries = {"dropUser": "db=db.getSiblingDB('${MONGODB_DATABASE}');db.dropUser('${MONGODB_USER}');",
|
|
"createUser1": "db=db.getSiblingDB('${MONGODB_DATABASE}'); db.createUser({user:'${MONGODB_USER}',pwd:'${MONGODB_PASSWORD}',roles:['readWrite','userAdmin','dbAdmin']});",
|
|
"insert": "db=db.getSiblingDB('${MONGODB_DATABASE}');db.testData.insert({x:0});",
|
|
"createUser2": "db.createUser({user:'test_user2',pwd:'test_password2',roles:['readWrite']});"}
|
|
|
|
self.run('{} --eval "{}"'.format(self.MONGO_ADMIN_CMD, queries["dropUser"]))
|
|
self.run('{} --eval "{}"'.format(self.MONGO_ADMIN_CMD, queries["createUser1"]))
|
|
self.run('{} --eval "{}"'.format(self.MONGO_ADMIN_CMD, queries["insert"]))
|
|
self.run('{} --eval "{}"'.format(self.MONGO_CMD, queries["createUser2"]))
|
|
|
|
if __name__ == '__main__':
|
|
main()
|