def setUp(self, mock_sync_node, mock_rabbit):
self.client = NasDaemon()
mock_rabbit.publish_message = MagicMock()
self.client.process_message = MagicMock()
self.client.SQLITE_DB = '/tmp/test_db_%s' % uuid.uuid4()
self.client.run()
with sqlite3.connect(self.client.SQLITE_DB) as con:
cur = con.cursor()
cur.execute('INSERT INTO zvols VALUES (?,?,?,?) ',
('vol1', None, None, None))
cur.execute('INSERT INTO zvols VALUES (?,?,?,?) ',
('vol2', 'my_tank', 'iqn.2001-04.com.nas-0-1-vol2',
'compute-0-3'))
cur.execute('INSERT INTO zvols VALUES (?,?,?,?) ',
('vol3_busy', 'my_tank',
'iqn.2001-04.com.nas-0-1-vol3_busy', 'compute-0-3'))
cur.execute('INSERT INTO zvol_calls VALUES (?,?,?)',
('vol3_busy', 'reply_to', time.time()))
cur.execute('INSERT INTO zvols VALUES (?,?,?,?) ',
('vol4_busy', 'my_tank',
'iqn.2001-04.com.nas-0-1-vol4_busy', 'compute-0-3'))
cur.execute('INSERT INTO zvol_calls VALUES (?,?,?)',
('vol4_busy', 'reply_to', time.time()))
con.commit()
def setUp(self, mock_rabbit_vm, mock_rabbit_sync):
self.nas_client = NasDaemon()
mock_rabbit_vm.publish_message = MagicMock()
mock_rabbit_sync.publish_message = MagicMock()
self.nas_client.process_message = MagicMock()
self.nas_client.is_sync_node = MagicMock(return_value = True)
self.nas_client.get_node_zpool = MagicMock(return_value = 'my_tank')
self.nas_client.SQLITE_DB = '/tmp/test_db_%s'%uuid.uuid4()
self.nas_client.run()
with sqlite3.connect(self.nas_client.SQLITE_DB) as con:
cur = con.cursor()
cur.execute('CREATE TABLE IF NOT EXISTS zvol_calls(zvol TEXT PRIMARY KEY NOT NULL, reply_to TEXT NOT NULL, time INT NOT NULL)')
cur.execute('CREATE TABLE IF NOT EXISTS zvols(zvol TEXT PRIMARY KEY NOT NULL, zpool TEXT NOT NULL, iscsi_target TEXT UNIQUE, remotehost TEXT)')
cur.execute('INSERT INTO zvols VALUES (?,?,?,?) ',('vol1', None, None, None))
cur.execute('INSERT INTO zvols VALUES (?,?,?,?) ',('vol2', 'my_tank', None, 'compute-0-1'))
cur.execute('INSERT INTO zvols VALUES (?,?,?,?) ',('vol3_busy', 'my_tank', None, 'compute-0-1'))
cur.execute('INSERT INTO zvol_calls VALUES (?,?,?)',('vol3_busy', 'reply_to', time.time()))
cur.execute('INSERT INTO zvols VALUES (?,?,?,?) ',('vol4_busy', 'my_tank', 'iqn.2001-04.com.nas-0-1-vol4_busy', 'compute-0-1'))
cur.execute('INSERT INTO zvol_calls VALUES (?,?,?)',('vol4_busy', 'reply_to', time.time()))
con.commit()
