コード例 #1
0
ファイル: test_util.py プロジェクト: hfeeki/tokenserver
    def test_secret_append(self):

        fd, filename = mkstemp()
        os.close(fd)
        try:
            generate_secret(filename, 'node')
            time.sleep(1.1)
            generate_secret(filename, 'node')
            secrets = Secrets(filename)
            self.assertEqual(len(secrets.get('node')), 2)
        finally:
            os.remove(filename)
コード例 #2
0
ファイル: util.py プロジェクト: jrgm/tokenserver
def generate_secret(filename, node):
    """Generates a new secret for the given node and saves it to a secrets
    file.

    :param filename: the complete path to the filename we want to put the
                     secret into.
    :param node: the node we want to generate the secret for.
    """
    if os.path.exists(filename):
        secrets = Secrets(filename)
    else:
        secrets = Secrets()

    secrets.add(node)
    secrets.save(filename)
    return node, secrets.get(node)[0]
コード例 #3
0
    def test_read_write(self):
        secrets = Secrets()

        secrets.add('phx23456')
        secrets.add('phx23456')
        secrets.add('phx23')

        phx23456_secrets = secrets.get('phx23456')
        self.assertEqual(len(secrets.get('phx23456')), 2)
        self.assertEqual(len(secrets.get('phx23')), 1)

        path = self.tempfile()

        secrets.save(path)

        secrets2 = Secrets(path)
        self.assertEqual(len(secrets2.get('phx23456')), 2)
        self.assertEqual(len(secrets2.get('phx23')), 1)
        self.assertEquals(secrets2.get('phx23456'), phx23456_secrets)
コード例 #4
0
    def test_multiple_files(self):
        # creating two distinct files
        secrets = Secrets()
        secrets.add('phx23456')
        one = self.tempfile()
        secrets.save(one)

        secrets = Secrets()
        secrets.add('phx123')
        two = self.tempfile()
        secrets.save(two)

        # loading the two files
        files = one, two
        secrets = Secrets(files)
        keys = secrets.keys()
        keys.sort()
        self.assertEqual(keys, ['phx123', 'phx23456'])
コード例 #5
0
    def test_read_write(self):
        secrets = Secrets()

        # We can only add one secret per second to the file, since
        # they are timestamped to 1s resolution.  Fake it.
        real_time = time.time
        time.time = itertools.count(int(real_time())).next
        try:
            secrets.add('phx23456')
            secrets.add('phx23456')
            secrets.add('phx23')
        finally:
            time.time = real_time

        phx23456_secrets = secrets.get('phx23456')
        self.assertEqual(len(secrets.get('phx23456')), 2)
        self.assertEqual(len(secrets.get('phx23')), 1)

        path = self.tempfile()

        secrets.save(path)

        secrets2 = Secrets(path)
        self.assertEqual(len(secrets2.get('phx23456')), 2)
        self.assertEqual(len(secrets2.get('phx23')), 1)
        self.assertEquals(secrets2.get('phx23456'), phx23456_secrets)
コード例 #6
0
    def test_multiple_files(self):
        # creating two distinct files
        secrets = Secrets()
        secrets.add("phx23456")
        one = self.tempfile()
        secrets.save(one)

        secrets = Secrets()
        secrets.add("phx123")
        two = self.tempfile()
        secrets.save(two)

        # loading the two files
        files = one, two
        secrets = Secrets(files)
        keys = secrets.keys()
        keys.sort()
        self.assertEqual(keys, ["phx123", "phx23456"])
コード例 #7
0
    def test_read_write(self):
        secrets = Secrets()

        # We can only add one secret per second to the file, since
        # they are timestamped to 1s resolution.  Fake it.
        real_time = time.time
        time.time = itertools.count(int(real_time())).next
        try:
            secrets.add("phx23456")
            secrets.add("phx23456")
            secrets.add("phx23")
        finally:
            time.time = real_time

        phx23456_secrets = secrets.get("phx23456")
        self.assertEqual(len(secrets.get("phx23456")), 2)
        self.assertEqual(len(secrets.get("phx23")), 1)

        path = self.tempfile()

        secrets.save(path)

        secrets2 = Secrets(path)
        self.assertEqual(len(secrets2.get("phx23456")), 2)
        self.assertEqual(len(secrets2.get("phx23")), 1)
        self.assertEquals(secrets2.get("phx23456"), phx23456_secrets)