def test_token_model_null_user(self): dt = isotime.add_utc_tz(datetime.datetime.utcnow()) tk = TokenAPI(user=None, token=uuid.uuid4().hex, expiry=isotime.format(dt)) self.assertRaises(ValueError, Token.add_or_update, TokenAPI.to_model(tk))
def test_token_model(self): dt = isotime.add_utc_tz(datetime.datetime.utcnow()) tk1 = TokenAPI(user='******', token=uuid.uuid4().hex, expiry=isotime.format(dt, offset=False)) tkdb1 = TokenAPI.to_model(tk1) self.assertIsNotNone(tkdb1) self.assertIsInstance(tkdb1, TokenDB) self.assertEqual(tkdb1.user, tk1.user) self.assertEqual(tkdb1.token, tk1.token) self.assertEqual(tkdb1.expiry, isotime.parse(tk1.expiry)) tkdb2 = Token.add_or_update(tkdb1) self.assertEqual(tkdb1, tkdb2) self.assertIsNotNone(tkdb2.id) tk2 = TokenAPI.from_model(tkdb2) self.assertEqual(tk2.user, tk1.user) self.assertEqual(tk2.token, tk1.token) self.assertEqual(tk2.expiry, tk1.expiry)
def test_token_model_null_expiry(self): tk = TokenAPI(user='******', token=uuid.uuid4().hex, expiry=None) self.assertRaises(ValueError, Token.add_or_update, TokenAPI.to_model(tk))
WF2_YAML = yaml.safe_dump(WF2_SPEC, default_flow_style=False) WF2_NAME = '%s.%s' % (PACK, WF2_YAML_FILE_NAME.replace('.yaml', '')) WF2 = workflows.Workflow(None, {'name': WF2_NAME, 'definition': WF2_YAML}) WF2_EXEC = copy.deepcopy(MISTRAL_EXECUTION) WF2_EXEC['workflow_name'] = WF2_NAME # Action executions requirements ACTION_CONTEXT = {'user': '******'} ACTION_PARAMS = {'friend': 'Rocky'} CHAMPION = worker.Worker(None) # Token for auth test cases TOKEN_API = TokenAPI( user=ACTION_CONTEXT['user'], token=uuid.uuid4().hex, expiry=isotime.format(isotime.add_utc_tz(datetime.datetime.utcnow()), offset=False)) TOKEN_DB = TokenAPI.to_model(TOKEN_API) NON_EMPTY_RESULT = 'non-empty' def process_create(payload): if isinstance(payload, LiveActionDB): CHAMPION.execute_action(payload) @mock.patch.object(LocalShellRunner, 'run', mock. MagicMock(return_value=(LIVEACTION_STATUS_SUCCEEDED, NON_EMPTY_RESULT, None))) @mock.patch.object(CUDPublisher, 'publish_create', mock.MagicMock(side_effect=process_create)) @mock.patch.object(CUDPublisher, 'publish_update', mock.MagicMock(return_value=None)) class TestMistralRunner(DbTestCase):