diff options
Diffstat (limited to 'snaps/domain')
-rw-r--r-- | snaps/domain/keypair.py | 34 | ||||
-rw-r--r-- | snaps/domain/test/keypair_tests.py | 35 |
2 files changed, 69 insertions, 0 deletions
diff --git a/snaps/domain/keypair.py b/snaps/domain/keypair.py new file mode 100644 index 0000000..2865125 --- /dev/null +++ b/snaps/domain/keypair.py @@ -0,0 +1,34 @@ +# Copyright (c) 2017 Cable Television Laboratories, Inc. ("CableLabs") +# and others. All rights reserved. +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at: +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. + + +class Keypair: + """ + SNAPS domain object for Keypairs. Should contain attributes that + are shared amongst cloud providers + """ + def __init__(self, name, id, public_key): + """ + Constructor + :param name: the keypair's name + :param id: the keypair's id + """ + self.name = name + self.id = id + self.public_key = public_key + + def __eq__(self, other): + return (self.name == other.name and self.id == other.id and + self.public_key == other.public_key) diff --git a/snaps/domain/test/keypair_tests.py b/snaps/domain/test/keypair_tests.py new file mode 100644 index 0000000..93f99ff --- /dev/null +++ b/snaps/domain/test/keypair_tests.py @@ -0,0 +1,35 @@ +# Copyright (c) 2017 Cable Television Laboratories, Inc. ("CableLabs") +# and others. All rights reserved. +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at: +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. + +import unittest +from snaps.domain.keypair import Keypair + + +class KeypairDomainObjectTests(unittest.TestCase): + """ + Tests the construction of the snaps.domain.test.Keypair class + """ + + def test_construction_positional(self): + keypair = Keypair('foo', '123-456', 'foo-bar') + self.assertEqual('foo', keypair.name) + self.assertEqual('123-456', keypair.id) + self.assertEqual('foo-bar', keypair.public_key) + + def test_construction_named(self): + keypair = Keypair(public_key='foo-bar', id='123-456', name='foo') + self.assertEqual('foo', keypair.name) + self.assertEqual('123-456', keypair.id) + self.assertEqual('foo-bar', keypair.public_key) |